From 3eaf6f0faf069cf412f1d282039706dea4544e31 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Thu, 16 May 2024 14:45:23 -0400 Subject: [PATCH] untrack tracker grad --- src/concrete_solve.jl | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/concrete_solve.jl b/src/concrete_solve.jl index 707c7f95d..20b0d0e88 100644 --- 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)