Skip to content

Commit 2341a8b

Browse files
committed
directly use unsafe_reinterpret
1 parent 48bf9b7 commit 2341a8b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/blas.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,9 @@ end
4242
Base.scale!(s::Real, X::GPUArray) = scale!(X, s)
4343
function Base.scale!(X::GPUArray{T}, s::Real) where T <: BLAS.BlasComplex
4444
R = typeof(real(zero(T)))
45-
buff = reinterpret(R, vec(X))
46-
BLAS.scal!(2*length(X), R(s), buff, 1)
45+
N = 2*length(X)
46+
buff = unsafe_reinterpret(R, X, (N,))
47+
BLAS.scal!(N, R(s), buff, 1)
4748
X
4849
end
4950
function Base.scale!(X::GPUArray{T}, s::Real) where T <: Union{Float32, Float64}

0 commit comments

Comments
 (0)