From 50c252325e958613a82b58710c339b96c60d0f71 Mon Sep 17 00:00:00 2001 From: DhairyaLGandhi Date: Thu, 2 May 2024 18:30:05 +0530 Subject: [PATCH 1/2] chore: rm mentions of SDESolution --- ext/SciMLBaseChainRulesCoreExt.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ext/SciMLBaseChainRulesCoreExt.jl b/ext/SciMLBaseChainRulesCoreExt.jl index 0e84ccbc3..4e9412063 100644 --- a/ext/SciMLBaseChainRulesCoreExt.jl +++ b/ext/SciMLBaseChainRulesCoreExt.jl @@ -111,15 +111,15 @@ end function ChainRulesCore.rrule( ::Type{ - <:ODESolution{uType, tType, isinplace, P, NP, F, G, K, + <:RODESolution{uType, tType, isinplace, P, NP, F, G, K, ND }}, u, args...) where {uType, tType, isinplace, P, NP, F, G, K, ND} - function SDESolutionAdjoint(ȳ) + function RODESolutionAdjoint(ȳ) (NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...) end - SDESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDESolutionAdjoint + RODESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), RODESolutionAdjoint end function ChainRulesCore.rrule(::SciMLBase.EnsembleSolution, sim, time, converged) From 3e64e258083e7527ee1c8e4e0817e2fee9ac9351 Mon Sep 17 00:00:00 2001 From: DhairyaLGandhi Date: Thu, 2 May 2024 18:31:21 +0530 Subject: [PATCH 2/2] chore: rm mentions of SDESolution in ZygoteExt --- ext/SciMLBaseZygoteExt.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl index c7ba1827e..a89bb3a93 100644 --- a/ext/SciMLBaseZygoteExt.jl +++ b/ext/SciMLBaseZygoteExt.jl @@ -135,11 +135,11 @@ end @adjoint function SDEProblem{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...) where {uType, tType, isinplace, P, NP, F, G, K, ND} - function SDESolutionAdjoint(ȳ) + function SDEProblemAdjoint(ȳ) (ȳ, ntuple(_ -> nothing, length(args))...) end - SDESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDESolutionAdjoint + SDEProblem{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDEProblemAdjoint end @adjoint function NonlinearSolution{T, N, uType, R, P, A, O, uType2}(u,