Skip to content

Commit b97a7d9

Browse files
a few more
1 parent 0c56bec commit b97a7d9

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

Diff for: src/derivative_wrappers.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -680,7 +680,7 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
680680
ytmp = _tmp5
681681
end
682682

683-
tmp1 .= 0 # should be removed for dλ
683+
Enzyme.make_zero!(tmp1) # should be removed for dλ
684684
ytmp .= y
685685

686686
#if dgrad !== nothing
@@ -698,7 +698,7 @@ function _vecjacobian!(dλ, y, λ, p, t, S::TS, isautojacvec::EnzymeVJP, dgrad,
698698
#if dy !== nothing
699699
# tmp3 = dy
700700
#else
701-
tmp3 .= 0
701+
Enzyme.make_zero!(tmp3)
702702
#end
703703

704704
vec(tmp4) .= vec(λ)

Diff for: src/gauss_adjoint.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -497,7 +497,7 @@ function vec_pjac!(out, λ, y, t, S::GaussIntegrand)
497497
tmp3, tmp4, tmp6 = paramjac_config
498498
vtmp4 = vec(tmp4)
499499
vtmp4 .= λ
500-
out .= 0
500+
Enzyme.make_zero!(out)
501501
Enzyme.make_zero!(tmp6)
502502
Enzyme.autodiff(
503503
Enzyme.Reverse, Enzyme.Duplicated(pf, tmp6), Enzyme.Const,

Diff for: src/quadrature_adjoint.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ function vec_pjac!(out, λ, y, t, S::AdjointSensitivityIntegrand)
293293
elseif sensealg.autojacvec isa EnzymeVJP
294294
tmp3, tmp4, tmp6 = paramjac_config
295295
tmp4 .= λ
296-
out .= 0
296+
Enzyme.make_zero!(out)
297297
Enzyme.make_zero!(tmp6)
298298
Enzyme.autodiff(
299299
Enzyme.Reverse, Enzyme.Duplicated(pf, tmp6), Enzyme.Const,

0 commit comments

Comments
 (0)