actually add remake_zero

Author: ChrisRackauckas

src/derivative_wrappers.jl

@@ -720,11 +720,7 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
     isautojacvec = get_jacvec(sensealg)
 
     if inplace_sensitivity(S)
-
-        # Correctness over speed
-        # TODO: Get a fix for `remake_zero!` to allow reusing zero'd memory
-        # https://github.com/EnzymeAD/Enzyme.jl/issues/2400
-        _tmp6 = Enzyme.make_zero(f)
+        _tmp6 = Enzyme.remake_zero!(f)
 
         if W === nothing
             Enzyme.autodiff(Enzyme.Reverse, Enzyme.Duplicated(f, _tmp6),

src/gauss_adjoint.jl

@@ -508,10 +508,7 @@ function vec_pjac!(out, λ, y, t, S::GaussIntegrand)
         Enzyme.remake_zero!(out)
 
         if SciMLBase.isinplace(sol.prob.f)
-            # Correctness over speed
-            # TODO: Get a fix for `remake_zero!` to allow reusing zero'd memory
-            # https://github.com/EnzymeAD/Enzyme.jl/issues/2400
-            tmp6 = Enzyme.make_zero(tmp6)
+            tmp6 = Enzyme.remake_zero!(tmp6)
 
             Enzyme.autodiff(
                 Enzyme.Reverse, Enzyme.Duplicated(pf, tmp6), Enzyme.Const,

src/quadrature_adjoint.jl

@@ -314,10 +314,7 @@ function vec_pjac!(out, λ, y, t, S::AdjointSensitivityIntegrand)
         end
 
         if SciMLBase.isinplace(sol.prob.f)
-            # Correctness over speed
-            # TODO: Get a fix for `remake_zero!` to allow reusing zero'd memory
-            # https://github.com/EnzymeAD/Enzyme.jl/issues/2400
-            tmp6 = Enzyme.make_zero(f)
+            tmp6 = Enzyme.remake_zero!(f)
             Enzyme.autodiff(
                 Enzyme.Reverse, Enzyme.Duplicated(f, tmp6), Enzyme.Const,
                 Enzyme.Duplicated(tmp3, tmp4),