Skip to content

Commit 0b63bbd

Browse files
a few more
1 parent 1408f61 commit 0b63bbd

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

src/derivative_wrappers.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -668,14 +668,14 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
668668
ytmp = _tmp5
669669
end
670670

671-
tmp1 .= 0 # should be removed for dλ
671+
Enzyme.make_zero!(tmp1) # should be removed for dλ
672672
ytmp .= y
673673

674674
#if dgrad !== nothing
675675
# tmp2 = dgrad
676676
#else
677677
dup = if !(tmp2 isa DiffEqBase.NullParameters)
678-
tmp2 .= 0
678+
Enzyme.make_zero!(tmp2)
679679
Enzyme.Duplicated(p, tmp2)
680680
else
681681
Enzyme.Const(p)
@@ -685,7 +685,7 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
685685
#if dy !== nothing
686686
# tmp3 = dy
687687
#else
688-
tmp3 .= 0
688+
Enzyme.make_zero!(tmp3)
689689
#end
690690

691691
vec(tmp4) .= vec(λ)

src/gauss_adjoint.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ function vec_pjac!(out, λ, y, t, S::GaussIntegrand)
476476
tmp3, tmp4, tmp6 = paramjac_config
477477
vtmp4 = vec(tmp4)
478478
vtmp4 .= λ
479-
out .= 0
479+
Enzyme.make_zero!(out)
480480
Enzyme.make_zero!(tmp6)
481481
Enzyme.autodiff(
482482
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
@@ -290,7 +290,7 @@ function vec_pjac!(out, λ, y, t, S::AdjointSensitivityIntegrand)
290290
elseif sensealg.autojacvec isa EnzymeVJP
291291
tmp3, tmp4, tmp6 = paramjac_config
292292
tmp4 .= λ
293-
out .= 0
293+
Enzyme.make_zero!(out)
294294
Enzyme.make_zero!(tmp6)
295295
Enzyme.autodiff(
296296
Enzyme.Reverse, Enzyme.Duplicated(pf, tmp6), Enzyme.Const,

0 commit comments

Comments
 (0)