Skip to content

Commit bc1620d

Browse files
Merge pull request #1170 from SciML/ChrisRackauckas-patch-2
vec tmp4 in quadrature adjoint
2 parents f8e7f6a + 6a4a022 commit bc1620d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/quadrature_adjoint.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,8 @@ function vec_pjac!(out, λ, y, t, S::AdjointSensitivityIntegrand)
298298
out .= p_grad
299299
elseif sensealg.autojacvec isa EnzymeVJP
300300
tmp3, tmp4, tmp6 = paramjac_config
301-
tmp4 .= λ
301+
vtmp4 = vec(tmp4)
302+
vtmp4 .= λ
302303
out .= 0
303304
Enzyme.autodiff(
304305
Enzyme.Reverse, Enzyme.Duplicated(pf, tmp6), Enzyme.Const,

0 commit comments

Comments
 (0)