Update derivative_wrappers.jl

Author: ChrisRackauckas

src/derivative_wrappers.jl

@@ -720,7 +720,15 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
     isautojacvec = get_jacvec(sensealg)
 
     if inplace_sensitivity(S)
-        Enzyme.remake_zero!(_tmp6)
+        if S isa CallbackSensitivityFunction
+            # 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)
+        else
+            Enzyme.remake_zero!(_tmp6)
+        end
+        
         if W === nothing
             Enzyme.autodiff(Enzyme.Reverse, Enzyme.Duplicated(f, _tmp6),
                 Enzyme.Const, Enzyme.Duplicated(tmp3, tmp4),