Merge pull request #1220 from SciML/remake_zero

Use remake_zero!

Author: ChrisRackauckas

Project.toml

@@ -65,7 +65,7 @@ DiffEqCallbacks = "4"
 DiffEqNoiseProcess = "5.19"
 Distributed = "1"
 Distributions = "0.25"
-Enzyme = "0.13"
+Enzyme = "0.13.50"
 FastBroadcast = "0.3.5"
 FiniteDiff = "2"
 ForwardDiff = "0.10, 1"

src/derivative_wrappers.jl

@@ -695,15 +695,15 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
         ytmp = _tmp5
     end
 
-    Enzyme.make_zero!(tmp1) # should be removed for dλ
+    Enzyme.remake_zero!(tmp1) # should be removed for dλ
     vec(ytmp) .= vec(y)
 
     #if dgrad !== nothing
     #  tmp2 = dgrad
     #else
     dup = if !(tmp2 isa SciMLBase.NullParameters)
         # tmp2 .= 0
-        Enzyme.make_zero!(tmp2)
+        Enzyme.remake_zero!(tmp2)
         Enzyme.Duplicated(p, repack(tmp2))
     else
         Enzyme.Const(p)
@@ -713,7 +713,7 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
     #if dy !== nothing
     #      tmp3 = dy
     #else
-    Enzyme.make_zero!(tmp3)
+    Enzyme.remake_zero!(tmp3)
     #end
 
     vec(tmp4) .= vec(λ)
@@ -722,7 +722,7 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
     if inplace_sensitivity(S)
 
         # Correctness over speed
-        # TODO: Get a fix for `make_zero!` to allow reusing zero'd memory
+        # 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)
 

src/gauss_adjoint.jl

@@ -504,12 +504,12 @@ function vec_pjac!(out, λ, y, t, S::GaussIntegrand)
         tmp3, tmp4, tmp6 = paramjac_config
         vtmp4 = vec(tmp4)
         vtmp4 .= λ
-        Enzyme.make_zero!(tmp3)
-        Enzyme.make_zero!(out)
+        Enzyme.remake_zero!(tmp3)
+        Enzyme.remake_zero!(out)
 
         if SciMLBase.isinplace(sol.prob.f)
             # Correctness over speed
-            # TODO: Get a fix for `make_zero!` to allow reusing zero'd memory
+            # 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)
 

src/quadrature_adjoint.jl

@@ -302,8 +302,8 @@ function vec_pjac!(out, λ, y, t, S::AdjointSensitivityIntegrand)
         tmp3, tmp4, tmp6 = paramjac_config
         vtmp4 = vec(tmp4)
 
-        Enzyme.make_zero!(out)
-        Enzyme.make_zero!(tmp3)
+        Enzyme.remake_zero!(out)
+        Enzyme.remake_zero!(tmp3)
         vtmp4 .= λ
 
         if !(p isa AbstractArray)
@@ -315,7 +315,7 @@ function vec_pjac!(out, λ, y, t, S::AdjointSensitivityIntegrand)
 
         if SciMLBase.isinplace(sol.prob.f)
             # Correctness over speed
-            # TODO: Get a fix for `make_zero!` to allow reusing zero'd memory
+            # 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)
             Enzyme.autodiff(