Merge pull request #947 from SciML/ChrisRackauckas-patch-6

Fix ReverseDiff capture overload ambiguities

Author: ChrisRackauckas

ext/DiffEqBaseReverseDiffExt.jl

@@ -121,6 +121,15 @@ function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
     DiffEqBase.solve_up(prob, sensealg, u0, reduce(vcat, p), args...; kwargs...)
 end
 
+function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
+    sensealg::Union{
+        SciMLBase.AbstractOverloadingSensitivityAlgorithm,
+        Nothing}, u0::ReverseDiff.TrackedArray,
+    p::AbstractArray{<:ReverseDiff.TrackedReal},
+    args...; kwargs...)
+    DiffEqBase.solve_up(prob, sensealg, u0, reduce(vcat, p), args...; kwargs...)
+end
+
 function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
     sensealg::Union{
         SciMLBase.AbstractOverloadingSensitivityAlgorithm,
@@ -130,6 +139,15 @@ function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
     DiffEqBase.solve_up(prob, sensealg, reduce(vcat, u0), p, args...; kwargs...)
 end
 
+function DiffEqBase.solve_up(prob::DiffEqBase.DEProblem,
+    sensealg::Union{
+        SciMLBase.AbstractOverloadingSensitivityAlgorithm,
+        Nothing},
+    u0::AbstractArray{<:ReverseDiff.TrackedReal}, p::ReverseDiff.TrackedArray,
+    args...; kwargs...)
+    DiffEqBase.solve_up(prob, sensealg, reduce(vcat, u0), p, args...; kwargs...)
+end
+
 # Required becase ReverseDiff.@grad function DiffEqBase.solve_up is not supported!
 import DiffEqBase: solve_up
 ReverseDiff.@grad function solve_up(prob, sensealg, u0, p, args...; kwargs...)