178
178
end
179
179
end
180
180
181
- function Base.:\ (D:: Diagonal{<:Any, <:AnyGPUArray } , B:: AnyGPUVecOrMat )
181
+ function Base.:\ (D:: Diagonal{<:Any, <:AbstractGPUArray } , B:: AbstractGPUVecOrMat )
182
182
z = D. diag .== 0
183
183
if any (z)
184
184
i = findfirst (collect (z))
@@ -189,7 +189,7 @@ function Base.:\(D::Diagonal{<:Any, <:AnyGPUArray}, B::AnyGPUVecOrMat)
189
189
end
190
190
191
191
if VERSION < v " 1.8-"
192
- function LinearAlgebra. ldiv! (D:: Diagonal{<:Any, <:AnyGPUArray } ,
192
+ function LinearAlgebra. ldiv! (D:: Diagonal{<:Any, <:AbstractGPUArray } ,
193
193
B:: StridedVecOrMat )
194
194
m, n = size (B, 1 ), size (B, 2 )
195
195
if m != length (D. diag)
322
322
LinearAlgebra. mul! (C:: AnyGPUVecOrMat , A:: AnyGPUVecOrMat , B:: AnyGPUVecOrMat , a:: Number , b:: Number ) = generic_matmatmul! (C, A, B, a, b)
323
323
324
324
325
- function generic_rmul! (X:: AnyArray , s:: Number )
325
+ function generic_rmul! (X:: AbstractArray , s:: Number )
326
326
gpu_call (X, s; name= " rmul!" ) do ctx, X, s
327
327
i = @linearidx X
328
328
@inbounds X[i] *= s
@@ -331,9 +331,9 @@ function generic_rmul!(X::AnyArray, s::Number)
331
331
return X
332
332
end
333
333
334
- LinearAlgebra. rmul! (A:: AnyGPUArray , b:: Number ) = generic_rmul! (A, b)
334
+ LinearAlgebra. rmul! (A:: AbstractGPUArray , b:: Number ) = generic_rmul! (A, b)
335
335
336
- function generic_lmul! (s:: Number , X:: AnyArray )
336
+ function generic_lmul! (s:: Number , X:: AbstractArray )
337
337
gpu_call (X, s; name= " lmul!" ) do ctx, X, s
338
338
i = @linearidx X
339
339
@inbounds X[i] = s* X[i]
@@ -342,7 +342,7 @@ function generic_lmul!(s::Number, X::AnyArray)
342
342
return X
343
343
end
344
344
345
- LinearAlgebra. lmul! (a:: Number , B:: AnyGPUArray ) = generic_lmul! (a, B)
345
+ LinearAlgebra. lmul! (a:: Number , B:: AbstractGPUArray ) = generic_lmul! (a, B)
346
346
347
347
348
348
# # permutedims
0 commit comments