We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents b5a5286 + 69e1dec commit 683087eCopy full SHA for 683087e
src/blas.jl
@@ -39,15 +39,15 @@ for elty in (Float64, Float32)
39
end
40
41
42
-Base.scale!(s::Real, X::GPUArray) = scale!(X, s)
43
-function Base.scale!(X::GPUArray{T}, s::Real) where T <: BLAS.BlasComplex
+Base.scale!(s::Number, X::GPUArray) = scale!(X, s)
+function Base.scale!(X::GPUArray{T}, s::Number) where T <: BLAS.BlasComplex
44
R = typeof(real(zero(T)))
45
N = 2*length(X)
46
buff = unsafe_reinterpret(R, X, (N,))
47
BLAS.scal!(N, R(s), buff, 1)
48
X
49
50
-function Base.scale!(X::GPUArray{T}, s::Real) where T <: Union{Float32, Float64}
+function Base.scale!(X::GPUArray{T}, s::Number) where T <: Union{Float32, Float64}
51
BLAS.scal!(length(X), T(s), X, 1)
52
53
0 commit comments