Skip to content

Commit d9c3e74

Browse files
Merge pull request #945 from prbzrg/have-algext
Simplify `:alg` extraction
2 parents 301dd29 + 319fd62 commit d9c3e74

File tree

4 files changed

+92
-117
lines changed

4 files changed

+92
-117
lines changed

ext/DiffEqBaseReverseDiffExt.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -76,23 +76,23 @@ end
7676
end
7777

7878
# `ReverseDiff.TrackedArray`
79-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
79+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
8080
sensealg::Union{
8181
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
8282
Nothing}, u0::ReverseDiff.TrackedArray,
8383
p::ReverseDiff.TrackedArray, args...; kwargs...)
8484
ReverseDiff.track(DiffEqBase.solve_up, prob, sensealg, u0, p, args...; kwargs...)
8585
end
8686

87-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
87+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
8888
sensealg::Union{
8989
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
9090
Nothing}, u0, p::ReverseDiff.TrackedArray,
9191
args...; kwargs...)
9292
ReverseDiff.track(DiffEqBase.solve_up, prob, sensealg, u0, p, args...; kwargs...)
9393
end
9494

95-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
95+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
9696
sensealg::Union{
9797
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
9898
Nothing}, u0::ReverseDiff.TrackedArray, p,
@@ -101,7 +101,7 @@ function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
101101
end
102102

103103
# `AbstractArray{<:ReverseDiff.TrackedReal}`
104-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
104+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
105105
sensealg::Union{
106106
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
107107
Nothing},
@@ -112,7 +112,7 @@ function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
112112
kwargs...)
113113
end
114114

115-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
115+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
116116
sensealg::Union{
117117
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
118118
Nothing}, u0,
@@ -121,7 +121,7 @@ function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
121121
DiffEqBase.solve_up(prob, sensealg, u0, reduce(vcat, p), args...; kwargs...)
122122
end
123123

124-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
124+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
125125
sensealg::Union{
126126
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
127127
Nothing}, u0::ReverseDiff.TrackedArray,

ext/DiffEqBaseTrackerExt.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,23 +67,23 @@ end
6767
end
6868
@inline DiffEqBase.ODE_DEFAULT_NORM(u::Tracker.TrackedReal, t::Tracker.TrackedReal) = abs(u)
6969

70-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
70+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
7171
sensealg::Union{
7272
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
7373
Nothing}, u0::Tracker.TrackedArray,
7474
p::Tracker.TrackedArray, args...; kwargs...)
7575
Tracker.track(DiffEqBase.solve_up, prob, sensealg, u0, p, args...; kwargs...)
7676
end
7777

78-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
78+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
7979
sensealg::Union{
8080
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
8181
Nothing}, u0::Tracker.TrackedArray, p, args...;
8282
kwargs...)
8383
Tracker.track(DiffEqBase.solve_up, prob, sensealg, u0, p, args...; kwargs...)
8484
end
8585

86-
function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
86+
function DiffEqBase.solve_up(prob::DiffEqBase.AbstractDEProblem,
8787
sensealg::Union{
8888
SciMLBase.AbstractOverloadingSensitivityAlgorithm,
8989
Nothing}, u0, p::Tracker.TrackedArray, args...;

src/DiffEqBase.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,19 +48,19 @@ PrecompileTools.@recompile_invalidations begin
4848
import PreallocationTools
4949

5050
import FunctionWrappersWrappers
51-
51+
5252
using SciMLBase
5353

5454
using SciMLOperators: AbstractSciMLOperator, AbstractSciMLScalarOperator
5555

56-
using SciMLBase: @def, DEIntegrator, DEProblem,
56+
using SciMLBase: @def, DEIntegrator, AbstractDEProblem,
5757
AbstractDiffEqInterpolation,
5858
DECallback, AbstractDEOptions, DECache, AbstractContinuousCallback,
5959
AbstractDiscreteCallback, AbstractLinearProblem, AbstractNonlinearProblem,
6060
AbstractOptimizationProblem, AbstractSteadyStateProblem,
6161
AbstractJumpProblem,
6262
AbstractNoiseProblem, AbstractEnsembleProblem, AbstractDynamicalODEProblem,
63-
DEAlgorithm, StandardODEProblem, AbstractIntegralProblem,
63+
AbstractDEAlgorithm, StandardODEProblem, AbstractIntegralProblem,
6464
AbstractSensitivityAlgorithm, AbstractODEAlgorithm,
6565
AbstractSDEAlgorithm, AbstractDDEAlgorithm, AbstractDAEAlgorithm,
6666
AbstractSDDEAlgorithm, AbstractRODEAlgorithm, DAEInitializationAlgorithm,

0 commit comments

Comments
 (0)