Merge pull request #418 from oscardssmith/os/parallel-precompile

run precompile workload in parallel
SciML · May 3, 2024 · db1032c · db1032c
2 parents 71b3f87 + c834033
commit db1032c
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/NonlinearSolve.jl b/src/NonlinearSolve.jl
@@ -135,11 +135,19 @@ include("default.jl")
         TrustRegion(; linsolve = LUFactorization()), nothing)
 
     @compile_workload begin
+        @sync begin
+        for T in (Float32, Float64), (fn, u0) in nlfuncs
+            Threads.@spawn NonlinearProblem(fn, T.(u0), T(2))
+        end
+        for (fn, u0) in nlfuncs
+            Threads.@spawn NonlinearLeastSquaresProblem(fn, u0, 2.0)
+        end
         for prob in probs_nls, alg in nls_algs
-            solve(prob, alg; abstol = 1e-2, verbose = false)
+            Threads.@spawn solve(prob, alg; abstol = 1e-2, verbose = false)
         end
         for prob in probs_nlls, alg in nlls_algs
-            solve(prob, alg; abstol = 1e-2, verbose = false)
+            Threads.@spawn solve(prob, alg; abstol = 1e-2, verbose = false)
+        end
         end
     end
 end