use sparse for lu_instance

SciML · Nov 27, 2024 · 07e9ffd · 07e9ffd
1 parent 0acbe41
commit 07e9ffd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/OrdinaryDiffEqDifferentiation/src/OrdinaryDiffEqDifferentiation.jl b/lib/OrdinaryDiffEqDifferentiation/src/OrdinaryDiffEqDifferentiation.jl
@@ -20,7 +20,7 @@ using DiffEqBase
 import LinearAlgebra
 import LinearAlgebra: Diagonal, I, UniformScaling, diagind, mul!, lmul!, axpby!, opnorm, lu
 import LinearAlgebra: LowerTriangular, UpperTriangular
-import SparseArrays: SparseMatrixCSC, AbstractSparseMatrix, nonzeros
+import SparseArrays: SparseMatrixCSC, AbstractSparseMatrix, nonzeros, sparse
 import ArrayInterface
 
 import StaticArrayInterface

diff --git a/lib/OrdinaryDiffEqDifferentiation/src/derivative_utils.jl b/lib/OrdinaryDiffEqDifferentiation/src/derivative_utils.jl
@@ -774,7 +774,11 @@ function build_J_W(alg, u, uprev, p, t, dt, f::F, ::Type{uEltypeNoUnits},
         elseif J isa StaticMatrix
             StaticWOperator(J, false)
         else
-            ArrayInterface.lu_instance(J)
+            if alg.autodiff isa AutoSparse
+                ArrayInterface.lu_instance(sparse(J))
+            else
+                ArrayInterface.lu_instance(J)
+            end
         end
     end
     return J, W