Skip to content

Commit b89abad

Browse files
Fix method override in CUDA
1 parent 99c54ec commit b89abad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ext/LinearSolveCUDAExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ function SciMLBase.solve!(cache::LinearSolve.LinearCache, alg::CudaOffloadLUFact
5252
SciMLBase.build_linear_solution(alg, y, nothing, cache)
5353
end
5454

55-
function LinearSolve.init_cacheval(alg::CudaOffloadLUFactorization, A, b, u, Pl, Pr,
55+
function LinearSolve.init_cacheval(alg::CudaOffloadLUFactorization, A::AbstractArray, b, u, Pl, Pr,
5656
maxiters::Int, abstol, reltol, verbose::LinearVerbosity,
5757
assumptions::OperatorAssumptions)
5858
T = eltype(A)
@@ -94,7 +94,7 @@ function SciMLBase.solve!(cache::LinearSolve.LinearCache, alg::CudaOffloadFactor
9494
SciMLBase.build_linear_solution(alg, y, nothing, cache)
9595
end
9696

97-
function LinearSolve.init_cacheval(alg::CudaOffloadFactorization, A, b, u, Pl, Pr,
97+
function LinearSolve.init_cacheval(alg::CudaOffloadFactorization, A::AbstractArray, b, u, Pl, Pr,
9898
maxiters::Int, abstol, reltol, verbose::LinearVerbosity,
9999
assumptions::OperatorAssumptions)
100100
qr(CUDA.CuArray(A))

0 commit comments

Comments
 (0)