diff --git a/src/derivative_wrappers.jl b/src/derivative_wrappers.jl index 07c449318..51953572c 100644 --- a/src/derivative_wrappers.jl +++ b/src/derivative_wrappers.jl @@ -680,7 +680,8 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad, ytmp = _tmp5 end - Enzyme.make_zero!(tmp1) # should be removed for dλ + tmp1 .= 0 + #Enzyme.make_zero!(tmp1) # should be removed for dλ ytmp .= y #if dgrad !== nothing @@ -698,7 +699,8 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad, #if dy !== nothing # tmp3 = dy #else - Enzyme.make_zero!(tmp3) + tmp3 .= 0 + #Enzyme.make_zero!(tmp3) #end vec(tmp4) .= vec(λ)