diff --git a/ext/SciMLBaseChainRulesCoreExt.jl b/ext/SciMLBaseChainRulesCoreExt.jl index 5cfd0fbee..b0e517502 100644 --- a/ext/SciMLBaseChainRulesCoreExt.jl +++ b/ext/SciMLBaseChainRulesCoreExt.jl @@ -43,7 +43,7 @@ function ChainRulesCore.rrule(config::ChainRulesCore.RuleConfig{ typeof(VA.alg), typeof(VA.interp), typeof(VA.alg_choice), - typeof(VA.destats), + typeof(VA.stats), }(du, nothing, nothing, @@ -54,7 +54,7 @@ function ChainRulesCore.rrule(config::ChainRulesCore.RuleConfig{ VA.interp, VA.dense, 0, - VA.destats, + VA.stats, VA.alg_choice, VA.retcode) (NoTangent(), Δ′, NoTangent(), NoTangent()) diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl index 1f09d55c3..77e3ea492 100644 --- a/ext/SciMLBaseZygoteExt.jl +++ b/ext/SciMLBaseZygoteExt.jl @@ -24,8 +24,8 @@ using RecursiveArrayTools N = length(VA.prob.p) Δ′ = ODESolution{T, N, typeof(du), Nothing, Nothing, typeof(VA.t), typeof(VA.k), typeof(dprob), typeof(VA.alg), typeof(VA.interp), - typeof(VA.destats), typeof(VA.alg_choice)}(du, nothing, nothing, - VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.destats, + typeof(VA.stats), typeof(VA.alg_choice)}(du, nothing, nothing, + VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.stats, VA.alg_choice, VA.retcode) (Δ′, nothing, nothing) end @@ -53,8 +53,8 @@ end N = length(VA.prob.p) Δ′ = ODESolution{T, N, typeof(du), Nothing, Nothing, typeof(VA.t), typeof(VA.k), typeof(dprob), typeof(VA.alg), typeof(VA.interp), - typeof(VA.destats), typeof(VA.alg_choice)}(du, nothing, nothing, - VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.destats, + typeof(VA.stats), typeof(VA.alg_choice)}(du, nothing, nothing, + VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.stats, VA.alg_choice, VA.retcode) (Δ′, nothing, nothing) end