address comments and add compat for FastBroadcast

SciML · Oct 31, 2023 · abaf747 · abaf747
1 parent 7390f71
commit abaf747
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/Project.toml b/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"

diff --git a/src/NonlinearSolve.jl b/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

diff --git a/src/pseudotransient.jl b/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