fix make_zero!

SciML · Jun 9, 2024 · 549bdfc · 549bdfc
1 parent 27381c0
commit 549bdfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/DiffEqBaseEnzymeExt.jl b/ext/DiffEqBaseEnzymeExt.jl
@@ -49,7 +49,7 @@ function Enzyme.EnzymeRules.reverse(config::Enzyme.EnzymeRules.ConfigWidth{1},
         end
         ptr.dval .+= darg
     end
-    Enzyme.make_zero!(dres)
+    Enzyme.make_zero!(dres.u)
     return ntuple(_ -> nothing, Val(length(args) + 4))
 end