handle stochastic cases

SciML · Aug 25, 2022 · 462402d · 462402d
1 parent 6b945e1
commit 462402d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/backsolve_adjoint.jl b/src/backsolve_adjoint.jl
@@ -272,13 +272,13 @@ end
  else
  transformed_function = StochasticTransformedFunction(sol, sol.prob.f, sol.prob.g,
  corfunc_analytical)
- drift_function = ODEFunction(transformed_function)
+ drift_function = ODEFunction{false,true}(transformed_function)
  sense_drift = ODEBacksolveSensitivityFunction(g, sensealg, discrete, sol,
  dgdu_continuous, dgdp_continuous,
  drift_function, alg)
  end
 
- diffusion_function = ODEFunction(sol.prob.g, jac = diffusion_jac,
+ diffusion_function = ODEFunction{isinplace(sol.prob),true}(sol.prob.g, jac = diffusion_jac,
  paramjac = diffusion_paramjac)
  sense_diffusion = ODEBacksolveSensitivityFunction(g, sensealg, discrete, sol,
  dgdu_continuous, dgdp_continuous,

diff --git a/src/interpolating_adjoint.jl b/src/interpolating_adjoint.jl
@@ -431,7 +431,7 @@ end
  abstol = abstol),
  tspan = tspan)
 
- diffusion_function = ODEFunction(sol.prob.g, jac = diffusion_jac,
+ diffusion_function = ODEFunction{isinplace(sol.prob),true}(sol.prob.g, jac = diffusion_jac,
  paramjac = diffusion_paramjac)
  sense_diffusion = ODEInterpolatingAdjointSensitivityFunction(g, sensealg, discrete, sol,
  dgdu_continuous,