untrack tracker grad

SciML · May 16, 2024 · 3eaf6f0 · 3eaf6f0
1 parent d75fefa
commit 3eaf6f0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/concrete_solve.jl b/src/concrete_solve.jl
@@ -1519,6 +1519,9 @@ function DiffEqBase._concrete_solve_adjoint(
     end
 
     function steadystatebackpass(Δ)
+	if originator isa SciMLBase.TrackerOriginator
+            Δ = Tracker.data(Δ)
+	end
         # Δ = dg/dx or diffcache.dg_val
         # del g/del p = 0
         function df(_out, u, p, t, i)