fix calc_t derivative

SciML · Nov 29, 2024 · 30e9ab6 · 30e9ab6
1 parent f9002b2
commit 30e9ab6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/OrdinaryDiffEqDifferentiation/src/derivative_utils.jl b/lib/OrdinaryDiffEqDifferentiation/src/derivative_utils.jl
@@ -50,7 +50,7 @@ function calc_tderivative!(integrator, cache, dtd1, repeat_step)
                     autodiff_alg
                 end
 
-                autodiff_alg = ADTypes.dense_ad(alg_autodiff(alg))
+                autodiff_alg = alg_autodiff(alg) isa AutoSparse ? ADTypes.dense_ad(alg_autodiff(alg)) : alg_autodiff(alg)
                 DI.derivative!(tf, linsolve_tmp, dT, cache.grad_config, autodiff_alg, t)
             end
         end