fix: parallel precompile

Author: avik-pal

lib/BracketingNonlinearSolve/src/BracketingNonlinearSolve.jl

@@ -35,8 +35,8 @@ end
         algs = (Alefeld(), Bisection(), Brent(), Falsi(), ITP(), Ridder())
 
         @compile_workload begin
-            for alg in algs
-                CommonSolve.solve(prob_brack, alg; abstol = 1e-6)
+            @sync for alg in algs
+                Threads.@spawn CommonSolve.solve(prob_brack, alg; abstol = 1e-6)
             end
         end
     end

lib/NonlinearSolveFirstOrder/src/NonlinearSolveFirstOrder.jl

@@ -35,23 +35,21 @@ include("pseudo_transient.jl")
 include("solve.jl")
 
 @setup_workload begin
-    include(joinpath(
-        @__DIR__, "..", "..", "..", "common", "nonlinear_problem_workloads.jl"
-    ))
-    include(joinpath(
-        @__DIR__, "..", "..", "..", "common", "nlls_problem_workloads.jl"
-    ))
+    include("../../../common/nonlinear_problem_workloads.jl")
+    include("../../../common/nlls_problem_workloads.jl")
 
     nlp_algs = [NewtonRaphson(), TrustRegion(), LevenbergMarquardt()]
     nlls_algs = [GaussNewton(), TrustRegion(), LevenbergMarquardt()]
 
     @compile_workload begin
-        for prob in nonlinear_problems, alg in nlp_algs
-            CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
-        end
+        @sync begin
+            for prob in nonlinear_problems, alg in nlp_algs
+                Threads.@spawn CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
+            end
 
-        for prob in nlls_problems, alg in nlls_algs
-            CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
+            for prob in nlls_problems, alg in nlls_algs
+                Threads.@spawn CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
+            end
         end
     end
 end

lib/NonlinearSolveQuasiNewton/src/NonlinearSolveQuasiNewton.jl

@@ -33,15 +33,13 @@ include("klement.jl")
 include("solve.jl")
 
 @setup_workload begin
-    include(joinpath(
-        @__DIR__, "..", "..", "..", "common", "nonlinear_problem_workloads.jl"
-    ))
+    include("../../../common/nonlinear_problem_workloads.jl")
 
     algs = [Broyden(), Klement()]
 
     @compile_workload begin
-        for prob in nonlinear_problems, alg in algs
-            CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
+        @sync for prob in nonlinear_problems, alg in algs
+            Threads.@spawn CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
         end
     end
 end

lib/NonlinearSolveSpectralMethods/src/NonlinearSolveSpectralMethods.jl

@@ -23,8 +23,8 @@ include("solve.jl")
     algs = [DFSane()]
 
     @compile_workload begin
-        for prob in nonlinear_problems, alg in algs
-            CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
+        @sync for prob in nonlinear_problems, alg in algs
+            Threads.@spawn CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
         end
     end
 end

lib/SimpleNonlinearSolve/src/SimpleNonlinearSolve.jl

@@ -148,8 +148,8 @@ function solve_adjoint_internal end
         #!format: on
 
         @compile_workload begin
-            for prob in (prob_scalar, prob_iip, prob_oop), alg in algs
-                CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
+            @sync for prob in (prob_scalar, prob_iip, prob_oop), alg in algs
+                Threads.@spawn CommonSolve.solve(prob, alg; abstol = 1e-2, verbose = false)
             end
         end
     end

src/NonlinearSolve.jl

@@ -58,12 +58,18 @@ include("default.jl")
     include(joinpath(@__DIR__, "..", "common", "nlls_problem_workloads.jl"))
 
     @compile_workload begin
-        for prob in nonlinear_problems
-            CommonSolve.solve(prob, nothing; abstol = 1e-2, verbose = false)
-        end
-
-        for prob in nlls_problems
-            CommonSolve.solve(prob, nothing; abstol = 1e-2, verbose = false)
+        @sync begin
+            for prob in nonlinear_problems
+                Threads.@spawn CommonSolve.solve(
+                    prob, nothing; abstol = 1e-2, verbose = false
+                )
+            end
+
+            for prob in nlls_problems
+                Threads.@spawn CommonSolve.solve(
+                    prob, nothing; abstol = 1e-2, verbose = false
+                )
+            end
         end
     end
 end