Skip to content

Commit 6371246

Browse files
committed
fix pushforward! tuple
1 parent ec1b9f4 commit 6371246

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lib/OrdinaryDiffEqDifferentiation/src/operators.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ function prepare_jvp(f::DiffEqBase.AbstractDiffEqFunction, du, u, p, t, autodiff
6767
autodiff = autodiff isa AutoSparse ? ADTypes.dense_ad(autodiff) : autodiff
6868
@assert DI.check_inplace(autodiff) "AD backend $(autodiff) doesn't support in-place problems."
6969
di_prep = DI.prepare_pushforward(f, du, autodiff, u, (u,), DI.ConstantOrCache(p), DI.Constant(t))
70-
return (Jv, v, u, p, t) -> DI.pushforward!(f, du, (reshape(Jv, size(du)),), di_prep, autodiff, u, (reshape(v,size(u)), DI.ConstantOrCache(p), DI.Constant(t)))
70+
return (Jv, v, u, p, t) -> DI.pushforward!(f, du, (reshape(Jv, size(du)),), di_prep, autodiff, u, (reshape(v,size(u)),), DI.ConstantOrCache(p), DI.Constant(t))
7171
end
7272

7373
function SciMLOperators.update_coefficients!(J::JVPCache, u, p, t)

0 commit comments

Comments
 (0)