Remove extra chainrulescore

Author: ChrisRackauckas

ext/DiffEqBaseChainRulesCoreExt.jl

@@ -7,5 +7,22 @@ import ChainRulesCore
 import ChainRulesCore: NoTangent
 
 ChainRulesCore.rrule(::typeof(numargs), f) = (numargs(f), df -> (NoTangent(), NoTangent()))
+ChainRulesCore.@non_differentiable checkkwargs(kwargshandle)
+
+function ChainRulesCore.frule(::typeof(solve_up), prob,
+    sensealg::Union{Nothing, AbstractSensitivityAlgorithm},
+    u0, p, args...;
+    kwargs...)
+    _solve_forward(prob, sensealg, u0, p, SciMLBase.ChainRulesOriginator(), args...;
+        kwargs...)
+end
+
+function ChainRulesCore.rrule(::typeof(solve_up), prob::AbstractDEProblem,
+    sensealg::Union{Nothing, AbstractSensitivityAlgorithm},
+    u0, p, args...;
+    kwargs...)
+    _solve_adjoint(prob, sensealg, u0, p, SciMLBase.ChainRulesOriginator(), args...;
+        kwargs...)
+end
 
 end

src/solve.jl

@@ -1051,8 +1051,6 @@ function checkkwargs(kwargshandle; kwargs...)
     end
 end
 
-@non_differentiable checkkwargs(kwargshandle)
-
 function get_concrete_problem(prob::AbstractJumpProblem, isadapt; kwargs...)
     prob
 end
@@ -1396,22 +1394,6 @@ discrete sensitivity algorithms.
 """
 struct SensitivityADPassThrough <: AbstractDEAlgorithm end
 
-function ChainRulesCore.frule(::typeof(solve_up), prob,
-    sensealg::Union{Nothing, AbstractSensitivityAlgorithm},
-    u0, p, args...;
-    kwargs...)
-    _solve_forward(prob, sensealg, u0, p, SciMLBase.ChainRulesOriginator(), args...;
-        kwargs...)
-end
-
-function ChainRulesCore.rrule(::typeof(solve_up), prob::AbstractDEProblem,
-    sensealg::Union{Nothing, AbstractSensitivityAlgorithm},
-    u0, p, args...;
-    kwargs...)
-    _solve_adjoint(prob, sensealg, u0, p, SciMLBase.ChainRulesOriginator(), args...;
-        kwargs...)
-end
-
 ###
 ### Legacy Dispatches to be Non-Breaking
 ###