Merge pull request #1055 from SciML/revert-1054-mb/use-make-zero

Revert "use Enzyme make_zero"
SciML · Jun 6, 2024 · 5fa2b7c · 5fa2b7c
2 parents bc50a2b + fd51d10
commit 5fa2b7c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/ext/DiffEqBaseEnzymeExt.jl b/ext/DiffEqBaseEnzymeExt.jl
@@ -28,7 +28,10 @@ function Enzyme.EnzymeRules.augmented_primal(config::Enzyme.EnzymeRules.ConfigWi
         SciMLBase.EnzymeOriginator(), ntuple(arg_copy, Val(length(args)))...;
         kwargs...)
 
-    dres = Enzyme.make_zero(res[1])::RT
+    dres = deepcopy(res[1])::RT
+    for v in dres.u
+        v .= 0
+    end
     tup = (dres, res[2])
     return Enzyme.EnzymeRules.AugmentedReturn{RT, RT, Any}(res[1], dres, tup::Any)
 end
@@ -49,7 +52,9 @@ function Enzyme.EnzymeRules.reverse(config::Enzyme.EnzymeRules.ConfigWidth{1},
         end
         ptr.dval .+= darg
     end
-    dres = Enzyme.make_zero(dres)
+    for v in dres.u
+        v .= 0
+    end
     return ntuple(_ -> nothing, Val(length(args) + 4))
 end