178178 end
179179end
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)
189189end
190190
191191if 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)
322322LinearAlgebra. 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
332332end
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
343343end
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