From e036d21c43e754818a0a0e7781bfd378522015e3 Mon Sep 17 00:00:00 2001
From: Matt Bossart <bossart.matt@gmail.com>
Date: Thu, 6 Jun 2024 12:31:14 -0400
Subject: [PATCH] use Enzyme make_zero

---
 ext/DiffEqBaseEnzymeExt.jl | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/ext/DiffEqBaseEnzymeExt.jl b/ext/DiffEqBaseEnzymeExt.jl
index aa5a1a0e0..f69d5233b 100644
--- a/ext/DiffEqBaseEnzymeExt.jl
+++ b/ext/DiffEqBaseEnzymeExt.jl
@@ -28,10 +28,7 @@ function Enzyme.EnzymeRules.augmented_primal(config::Enzyme.EnzymeRules.ConfigWi
         SciMLBase.EnzymeOriginator(), ntuple(arg_copy, Val(length(args)))...;
         kwargs...)
 
-    dres = deepcopy(res[1])::RT
-    for v in dres.u
-        v .= 0
-    end
+    dres = Enzyme.make_zero(res[1])::RT
     tup = (dres, res[2])
     return Enzyme.EnzymeRules.AugmentedReturn{RT, RT, Any}(res[1], dres, tup::Any)
 end
@@ -52,9 +49,7 @@ function Enzyme.EnzymeRules.reverse(config::Enzyme.EnzymeRules.ConfigWidth{1},
         end
         ptr.dval .+= darg
     end
-    for v in dres.u
-        v .= 0
-    end
+    dres = Enzyme.make_zero(dres)
     return ntuple(_ -> nothing, Val(length(args) + 4))
 end