address comments and add compat for FastBroadcast

Author: yonatanwesen

Project.toml

@@ -43,6 +43,7 @@ ConcreteStructs = "0.2"
 DiffEqBase = "6.130"
 EnumX = "1"
 Enzyme = "0.11"
+FastBroadcast = "0.1.9, 0.2"
 FastLevenbergMarquardt = "0.1"
 FiniteDiff = "2"
 ForwardDiff = "0.10.3"

src/NonlinearSolve.jl

@@ -9,7 +9,7 @@ import PrecompileTools
 
 PrecompileTools.@recompile_invalidations begin
     using DiffEqBase, LinearAlgebra, LinearSolve, SparseArrays, SparseDiffTools
-    using FastBroadcast: @.., True, False
+    using FastBroadcast: @..
     import ArrayInterface: restructure
 
     import ADTypes: AbstractFiniteDifferencesMode

src/pseudotransient.jl

@@ -167,9 +167,7 @@ function perform_step!(cache::PseudoTransientCache{false})
     if linsolve === nothing
         cache.du = fu1 / cache.J
     else
-        linres = dolinsolve(alg.precs, linsolve; A = cache.J,
-            b = _vec(fu1),
-            linu = _vec(cache.du), p, reltol = cache.abstol)
+        linres = dolinsolve(alg.precs, linsolve; A = cache.J,b = _vec(fu1),linu = _vec(cache.du), p, reltol = cache.abstol)
         cache.linsolve = linres.cache
     end
     cache.u = @. u - cache.du  # `u` might not support mutation