Skip to content

Commit 6471619

Browse files
Namespace
1 parent b3c38a3 commit 6471619

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

lib/LinearSolveCUDA/src/LinearSolveCUDA.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@ using CUDA, LinearAlgebra, LinearSolve, SciMLBase
44

55
struct CudaOffloadFactorization <: LinearSolve.AbstractFactorization end
66

7-
function SciMLBase.solve(cache::LinearCache, alg::CudaOffloadFactorization; kwargs...)
7+
function SciMLBase.solve(cache::LinearSolve.LinearCache, alg::CudaOffloadFactorization; kwargs...)
88
if cache.isfresh
9-
fact = do_factorization(alg, CUDA.CuArray(cache.A), cache.b, cache.u)
10-
cache = set_cacheval(cache, fact)
9+
fact = LinearSolve.do_factorization(alg, CUDA.CuArray(cache.A), cache.b, cache.u)
10+
cache = LinearSolve.set_cacheval(cache, fact)
1111
end
1212

1313
copyto!(cache.u, cache.b)

lib/LinearSolvePardiso/src/LinearSolvePardiso.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ function LinearSolve.init_cacheval(alg::PardisoJL, A, b, u, Pl, Pr, maxiters, ab
9393
return solver
9494
end
9595

96-
function SciMLBase.solve(cache::LinearCache, alg::PardisoJL; kwargs...)
96+
function SciMLBase.solve(cache::LinearSolve.LinearCache, alg::PardisoJL; kwargs...)
9797
@unpack A, b, u = cache
9898
A = convert(AbstractMatrix, A)
9999

0 commit comments

Comments
 (0)