Skip to content

Commit 683087e

Browse files
authored
Merge pull request #83 from JuliaGPU/sd/scale
use number, not real
2 parents b5a5286 + 69e1dec commit 683087e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/blas.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,15 +39,15 @@ for elty in (Float64, Float32)
3939
end
4040
end
4141

42-
Base.scale!(s::Real, X::GPUArray) = scale!(X, s)
43-
function Base.scale!(X::GPUArray{T}, s::Real) where T <: BLAS.BlasComplex
42+
Base.scale!(s::Number, X::GPUArray) = scale!(X, s)
43+
function Base.scale!(X::GPUArray{T}, s::Number) where T <: BLAS.BlasComplex
4444
R = typeof(real(zero(T)))
4545
N = 2*length(X)
4646
buff = unsafe_reinterpret(R, X, (N,))
4747
BLAS.scal!(N, R(s), buff, 1)
4848
X
4949
end
50-
function Base.scale!(X::GPUArray{T}, s::Real) where T <: Union{Float32, Float64}
50+
function Base.scale!(X::GPUArray{T}, s::Number) where T <: Union{Float32, Float64}
5151
BLAS.scal!(length(X), T(s), X, 1)
5252
X
5353
end

0 commit comments

Comments
 (0)