Skip to content

Commit 61d7a29

Browse files
Fix use of remake_zero! with callback affect differentiation
1 parent b87d92b commit 61d7a29

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

src/derivative_wrappers.jl

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -720,14 +720,7 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
720720
isautojacvec = get_jacvec(sensealg)
721721

722722
if inplace_sensitivity(S)
723-
if S isa CallbackSensitivityFunction
724-
# Correctness over speed
725-
# TODO: Get a fix for `remake_zero!` to allow reusing zero'd memory
726-
# https://github.com/EnzymeAD/Enzyme.jl/issues/2400
727-
_tmp6 = Enzyme.make_zero(f)
728-
else
729-
Enzyme.remake_zero!(_tmp6)
730-
end
723+
Enzyme.remake_zero!(_tmp6)
731724

732725
if W === nothing
733726
Enzyme.autodiff(Enzyme.Reverse, Enzyme.Duplicated(f, _tmp6),

0 commit comments

Comments
 (0)