Skip to content

Commit ee9cc9c

Browse files
Removing Any for rmul and lmul
1 parent abf4e07 commit ee9cc9c

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/host/linalg.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ else
178178
end
179179
end
180180

181-
function Base.:\(D::Diagonal{<:Any, <:AnyGPUArray}, B::AnyGPUVecOrMat)
181+
function Base.:\(D::Diagonal{<:Any, <:AbstractGPUArray}, B::AbstractGPUVecOrMat)
182182
z = D.diag .== 0
183183
if any(z)
184184
i = findfirst(collect(z))
@@ -189,7 +189,7 @@ function Base.:\(D::Diagonal{<:Any, <:AnyGPUArray}, B::AnyGPUVecOrMat)
189189
end
190190

191191
if VERSION < v"1.8-"
192-
function LinearAlgebra.ldiv!(D::Diagonal{<:Any, <:AnyGPUArray},
192+
function LinearAlgebra.ldiv!(D::Diagonal{<:Any, <:AbstractGPUArray},
193193
B::StridedVecOrMat)
194194
m, n = size(B, 1), size(B, 2)
195195
if m != length(D.diag)
@@ -322,7 +322,7 @@ end
322322
LinearAlgebra.mul!(C::AnyGPUVecOrMat, A::AnyGPUVecOrMat, B::AnyGPUVecOrMat, a::Number, b::Number) = generic_matmatmul!(C, A, B, a, b)
323323

324324

325-
function generic_rmul!(X::AnyArray, s::Number)
325+
function generic_rmul!(X::AbstractArray, s::Number)
326326
gpu_call(X, s; name="rmul!") do ctx, X, s
327327
i = @linearidx X
328328
@inbounds X[i] *= s
@@ -331,9 +331,9 @@ function generic_rmul!(X::AnyArray, s::Number)
331331
return X
332332
end
333333

334-
LinearAlgebra.rmul!(A::AnyGPUArray, b::Number) = generic_rmul!(A, b)
334+
LinearAlgebra.rmul!(A::AbstractGPUArray, b::Number) = generic_rmul!(A, b)
335335

336-
function generic_lmul!(s::Number, X::AnyArray)
336+
function generic_lmul!(s::Number, X::AbstractArray)
337337
gpu_call(X, s; name="lmul!") do ctx, X, s
338338
i = @linearidx X
339339
@inbounds X[i] = s*X[i]
@@ -342,7 +342,7 @@ function generic_lmul!(s::Number, X::AnyArray)
342342
return X
343343
end
344344

345-
LinearAlgebra.lmul!(a::Number, B::AnyGPUArray) = generic_lmul!(a, B)
345+
LinearAlgebra.lmul!(a::Number, B::AbstractGPUArray) = generic_lmul!(a, B)
346346

347347

348348
## permutedims

0 commit comments

Comments
 (0)