another calc_tderivative

SciML · Nov 27, 2024 · 8d5f6a2 · 8d5f6a2
1 parent a0ddbe4
commit 8d5f6a2
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/lib/OrdinaryDiffEqDifferentiation/src/derivative_utils.jl b/lib/OrdinaryDiffEqDifferentiation/src/derivative_utils.jl
@@ -42,7 +42,7 @@ function calc_tderivative!(integrator, cache, dtd1, repeat_step)
                 alg = unwrap_alg(integrator, true)
                 #derivative!(dT, tf, t, du2, integrator, cache.grad_config)
                 autodiff_alg = alg_autodiff(alg)
-                
+
                 autodiff_alg = if autodiff_alg isa AutoSparse
                     ADTypes.dense_ad(autodiff_alg)
                 else
@@ -68,7 +68,15 @@ function calc_tderivative(integrator, cache)
         tf = cache.tf
         tf.u = uprev
         tf.p = p
-        dT = DI.derivative(tf, alg_autodiff(alg), t)
+
+        autodiff_alg = alg_autodiff(alg)
+        autodiff_alg = if autodiff_alg isa AutoSparse
+            autodiff_alg = ADTypes.dense_ad(autodiff_alg)
+        else
+            autodiff_alg
+        end
+
+        dT = DI.derivative(tf, autodiff_alg, t)
     end
     dT
 end