diff --git a/lib/NonlinearSolveBase/src/solve.jl b/lib/NonlinearSolveBase/src/solve.jl index 76d765d7b..04b25026f 100644 --- a/lib/NonlinearSolveBase/src/solve.jl +++ b/lib/NonlinearSolveBase/src/solve.jl @@ -103,7 +103,12 @@ end return Expr(:block, calls...) end -@generated function SciMLBase.__solve( +function SciMLBase.__solve(prob::AbstractNonlinearProblem, alg::NonlinearSolvePolyAlgorithm, + args...; kwargs...) + __generated_polysolve(prob, alg, args...; kwargs...) +end + +@generated function __generated_polysolve( prob::AbstractNonlinearProblem, alg::NonlinearSolvePolyAlgorithm{Val{N}}, args...; stats = NLStats(0, 0, 0, 0, 0), alias_u0 = false, verbose = true, kwargs... ) where {N} diff --git a/test/forward_ad_tests.jl b/test/forward_ad_tests.jl index f3cf74bae..163349267 100644 --- a/test/forward_ad_tests.jl +++ b/test/forward_ad_tests.jl @@ -71,6 +71,7 @@ end Broyden(), Klement(), DFSane(), + FastShortcutNonlinearPolyalg(), nothing, NLsolveJL(), CMINPACK(),