Incorrect tag being passed

SciML · Oct 29, 2023 · b110e05 · b110e05
1 parent 756ad8c
commit b110e05
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/utils.jl b/src/utils.jl
@@ -21,6 +21,8 @@ function __findmin(f, x)
     end
 end
 
+struct NonlinearSolveTag end
+
 """
     default_adargs_to_adtype(; chunk_size = Val{0}(), autodiff = Val{true}(),
         standardtag = Val{true}(), diff_type = Val{:forward})
@@ -54,6 +56,7 @@ function default_adargs_to_adtype(; chunk_size = missing, autodiff = nothing,
 
     ad = _unwrap_val(autodiff)
     tag = _unwrap_val(standardtag)
+    tag isa Bool && tag && (tag = NonlinearSolveTag())
     ad && return AutoForwardDiff{_unwrap_val(chunk_size), typeof(tag)}(tag)
     return AutoFiniteDiff(; fdtype = diff_type)
 end
@@ -206,8 +209,6 @@ 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