Skip to content

Commit f8dcb17

Browse files
Merge pull request #1002 from SciML/adjoint
Update some solution adjoint reverse mode rules
2 parents f9d7427 + 9467757 commit f8dcb17

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name = "SciMLBase"
22
uuid = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
33
authors = ["Chris Rackauckas <accounts@chrisrackauckas.com> and contributors"]
4-
version = "2.86.0"
4+
version = "2.86.1"
55

66
[deps]
77
ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

ext/SciMLBaseChainRulesCoreExt.jl

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -75,29 +75,31 @@ end
7575
function ChainRulesCore.rrule(
7676
::Type{
7777
<:ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10,
78-
T11, T12, T13, T14
78+
T11, T12, T13, T14, T15, T16
7979
}}, u,
8080
args...) where {T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11,
81-
T12, T13, T14}
81+
T12, T13, T14, T15, T16}
8282
function ODESolutionAdjoint(ȳ)
8383
(NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
8484
end
8585

86-
ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14}(u, args...),
86+
ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14, T15, T16}(u, args...),
8787
ODESolutionAdjoint
8888
end
8989

9090
function ChainRulesCore.rrule(
9191
::Type{
92-
<:RODESolution{uType, tType, isinplace, P, NP, F, G, K,
93-
ND
92+
<:RODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10,
93+
T11, T12, T13, T14
9494
}}, u,
95-
args...) where {uType, tType, isinplace, P, NP, F, G, K, ND}
95+
args...) where {T1, T2, T3, T4, T5, T6, T7, T8, T9, T10,
96+
T11, T12, T13, T14}
9697
function RODESolutionAdjoint(ȳ)
9798
(NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
9899
end
99100

100-
RODESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...),
101+
RODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10,
102+
T11, T12, T13, T14}(u, args...),
101103
RODESolutionAdjoint
102104
end
103105

0 commit comments

Comments
 (0)