We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ec1b9f4 commit 6371246Copy full SHA for 6371246
lib/OrdinaryDiffEqDifferentiation/src/operators.jl
@@ -67,7 +67,7 @@ function prepare_jvp(f::DiffEqBase.AbstractDiffEqFunction, du, u, p, t, autodiff
67
autodiff = autodiff isa AutoSparse ? ADTypes.dense_ad(autodiff) : autodiff
68
@assert DI.check_inplace(autodiff) "AD backend $(autodiff) doesn't support in-place problems."
69
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)))
+ 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))
71
end
72
73
function SciMLOperators.update_coefficients!(J::JVPCache, u, p, t)
0 commit comments