Skip to content

Commit affc130

Browse files
fix GPU test
1 parent 6debe9c commit affc130

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
@@ -7,7 +7,7 @@ function SciMLBase.solve!(cache::LinearSolve.LinearCache, alg::CudaOffloadFactor
77
kwargs...)
88
if cache.isfresh
99
fact = LinearSolve.do_factorization(alg, CUDA.CuArray(cache.A), cache.b, cache.u)
10-
cache = LinearSolve.set_cacheval(cache, fact)
10+
cache.cacheval = fact
1111
cache.isfresh = false
1212
end
1313

@@ -24,7 +24,7 @@ function LinearSolve.do_factorization(alg::CudaOffloadFactorization, A, b, u)
2424
A = A.A
2525
end
2626

27-
fact = qr(CUDA.CuArray(A))
27+
fact = lu(CUDA.CuArray(A))
2828
return fact
2929
end
3030

0 commit comments

Comments
 (0)