Use a default NonlinearSolve Tag

SciML · Oct 27, 2023 · 8de2b6f · 8de2b6f
1 parent d8d1c95
commit 8de2b6f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/utils.jl b/src/utils.jl
@@ -206,14 +206,17 @@ function get_concrete_algorithm(alg, prob)
     return __get_concrete_algorithm(alg, prob)
 end
 
+struct NonlinearSolveTag end
+
 function __get_concrete_algorithm(alg, prob)
     @unpack sparsity, jac_prototype = prob.f
     use_sparse_ad = sparsity !== nothing || jac_prototype !== nothing
     ad = if eltype(prob.u0) <: Complex
         # Use Finite Differencing
         use_sparse_ad ? AutoSparseFiniteDiff() : AutoFiniteDiff()
     else
-        use_sparse_ad ? AutoSparseForwardDiff() : AutoForwardDiff{nothing, Nothing}(nothing)
+        tag = NonlinearSolveTag()
+        use_sparse_ad ? AutoSparseForwardDiff(; tag) : AutoForwardDiff(; tag)
     end
     return set_ad(alg, ad)
 end