update DAEResidualJacobianWrapper constructor

SciML · Nov 29, 2024 · 96a1487 · 96a1487
1 parent aed21fd
commit 96a1487
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/OrdinaryDiffEqNonlinearSolve/src/utils.jl b/lib/OrdinaryDiffEqNonlinearSolve/src/utils.jl
@@ -72,7 +72,8 @@ mutable struct DAEResidualJacobianWrapper{isAD, F, pType, duType, uType, alphaTy
     uprev::uprevType
     t::tType
     function DAEResidualJacobianWrapper(alg, f, p, α, invγdt, tmp, uprev, t)
-        isautodiff = alg_autodiff(alg) isa AutoForwardDiff
+        autodiff_alg = alg_autodiff(alg) isa AutoSparse ? dense_ad(alg_autodiff(alg)) : alg_autodiff(alg)
+        isautodiff = autodiff_alg isa AutoForwardDiff
         if isautodiff
             tmp_du = PreallocationTools.dualcache(uprev)
             tmp_u = PreallocationTools.dualcache(uprev)