Skip to content

Commit 1ef9286

Browse files
Merge pull request #639 from SciML/ChrisRackauckas-patch-1
Fix LinearSolveCUDAExt.jl
2 parents 40f9f16 + ace17c7 commit 1ef9286

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

ext/LinearSolveCUDAExt.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@ module LinearSolveCUDAExt
33
using CUDA
44
using LinearSolve: LinearSolve, is_cusparse, defaultalg, cudss_loaded, DefaultLinearSolver,
55
DefaultAlgorithmChoice, ALREADY_WARNED_CUDSS, LinearCache, needs_concrete_A,
6-
error_no_cudss_lu, CUDSS_LOADED, init_cacheval
6+
error_no_cudss_lu, init_cacheval, OperatorAssumptions, CudaOffloadFactorization,
7+
SparspakFactorization, KLUFactorization, UMFPACKFactorization
78
using LinearSolve.LinearAlgebra, LinearSolve.SciMLBase, LinearSolve.ArrayInterface
89
using SciMLBase: AbstractSciMLOperator
910

@@ -25,7 +26,7 @@ function LinearSolve.defaultalg(A::CUDA.CUSPARSE.CuSparseMatrixCSR{Tv, Ti}, b,
2526
end
2627

2728
function LinearSolve.error_no_cudss_lu(A::CUDA.CUSPARSE.CuSparseMatrixCSR)
28-
if !LinearSolve.CUDSS_LOADED[]
29+
if !LinearSolve.cudss_loaded(A)
2930
error("CUDSS.jl is required for LU Factorizations on CuSparseMatrixCSR. Please load this library.")
3031
end
3132
nothing

0 commit comments

Comments
 (0)