Skip to content

Commit 04f0a7b

Browse files
only inplace
1 parent f86b92f commit 04f0a7b

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

src/derivative_wrappers.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -720,8 +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-
_tmp6 = Enzyme.remake_zero!(f)
724-
723+
Enzyme.remake_zero!(_tmp6)
725724
if W === nothing
726725
Enzyme.autodiff(Enzyme.Reverse, Enzyme.Duplicated(f, _tmp6),
727726
Enzyme.Const, Enzyme.Duplicated(tmp3, tmp4),

src/gauss_adjoint.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -508,7 +508,7 @@ function vec_pjac!(out, λ, y, t, S::GaussIntegrand)
508508
Enzyme.remake_zero!(out)
509509

510510
if SciMLBase.isinplace(sol.prob.f)
511-
tmp6 = Enzyme.remake_zero!(tmp6)
511+
Enzyme.remake_zero!(tmp6)
512512

513513
Enzyme.autodiff(
514514
Enzyme.Reverse, Enzyme.Duplicated(pf, tmp6), Enzyme.Const,

src/quadrature_adjoint.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ function vec_pjac!(out, λ, y, t, S::AdjointSensitivityIntegrand)
314314
end
315315

316316
if SciMLBase.isinplace(sol.prob.f)
317-
tmp6 = Enzyme.remake_zero!(f)
317+
Enzyme.remake_zero!(tmp6)
318318
Enzyme.autodiff(
319319
Enzyme.Reverse, Enzyme.Duplicated(f, tmp6), Enzyme.Const,
320320
Enzyme.Duplicated(tmp3, tmp4),

0 commit comments

Comments
 (0)