Skip to content

Commit 9467757

Browse files
Update some solution adjoint reverse mode rules
These ones are only hit in the DiffEqGPU interfaces right now. Although we will shortly switch to Tangent projections soon, this will at least get tests working there again for now.
1 parent f9d7427 commit 9467757

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)