Skip to content

Commit 6a4a022

Browse files
vec tmp4 in quadrature adjoint
1 parent f8e7f6a commit 6a4a022

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)