bring back compilation workload

Author: jClugstor

lib/NonlinearSolveQuasiNewton/src/NonlinearSolveQuasiNewton.jl

@@ -34,19 +34,19 @@ include("klement.jl")
 
 include("solve.jl")
 
-# @setup_workload begin
-#     nonlinear_functions = (
-#         (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), 0.1),
-#         (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), [0.1]),
-#         (NonlinearFunction{true, NoSpecialize}((du, u, p) -> du .= u .* u .- p), [0.1])
-#     )
-
-#     nonlinear_problems = NonlinearProblem[]
-#     for (fn, u0) in nonlinear_functions
-#         push!(nonlinear_problems, NonlinearProblem(fn, u0, 2.0))
-#     end
-
-#     algs = [Broyden(), Klement()]
+@setup_workload begin
+    nonlinear_functions = (
+        (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), 0.1),
+        (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), [0.1]),
+        (NonlinearFunction{true, NoSpecialize}((du, u, p) -> du .= u .* u .- p), [0.1])
+    )
+
+    nonlinear_problems = NonlinearProblem[]
+    for (fn, u0) in nonlinear_functions
+        push!(nonlinear_problems, NonlinearProblem(fn, u0, 2.0))
+    end
+
+    algs = [Broyden(), Klement()]
 
     @compile_workload begin
         @sync for prob in nonlinear_problems, alg in algs

src/NonlinearSolve.jl

@@ -52,17 +52,17 @@ include("default.jl")
 
 include("forward_diff.jl")
 
-# @setup_workload begin
-#     nonlinear_functions = (
-#         (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), 0.1),
-#         (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), [0.1]),
-#         (NonlinearFunction{true, NoSpecialize}((du, u, p) -> du .= u .* u .- p), [0.1])
-#     )
-
-#     nonlinear_problems = NonlinearProblem[]
-#     for (fn, u0) in nonlinear_functions
-#         push!(nonlinear_problems, NonlinearProblem(fn, u0, 2.0))
-#     end
+@setup_workload begin
+    nonlinear_functions = (
+        (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), 0.1),
+        (NonlinearFunction{false, NoSpecialize}((u, p) -> u .* u .- p), [0.1]),
+        (NonlinearFunction{true, NoSpecialize}((du, u, p) -> du .= u .* u .- p), [0.1])
+    )
+
+    nonlinear_problems = NonlinearProblem[]
+    for (fn, u0) in nonlinear_functions
+        push!(nonlinear_problems, NonlinearProblem(fn, u0, 2.0))
+    end
 
     nonlinear_functions = (
         (NonlinearFunction{false, NoSpecialize}((u, p) -> (u .^ 2 .- p)[1:1]), [0.1, 0.0]),
@@ -85,27 +85,25 @@ include("forward_diff.jl")
         )
     )
 
-#     nlls_problems = NonlinearLeastSquaresProblem[]
-#     for (fn, u0) in nonlinear_functions
-#         push!(nlls_problems, NonlinearLeastSquaresProblem(fn, u0, 2.0))
-#     end
-
-#     @compile_workload begin
-#         @sync begin
-#             for prob in nonlinear_problems
-#                 Threads.@spawn CommonSolve.solve(
-#                     prob, nothing; abstol = 1e-2, verbose = true
-#                 )
-#             end
-
-#             for prob in nlls_problems
-#                 Threads.@spawn CommonSolve.solve(
-#                     prob, nothing; abstol = 1e-2, verbose = true
-#                 )
-#             end
-#         end
-#     end
-# end
+    nlls_problems = NonlinearLeastSquaresProblem[]
+    for (fn, u0) in nonlinear_functions
+        push!(nlls_problems, NonlinearLeastSquaresProblem(fn, u0, 2.0))
+    end
+
+    @compile_workload begin
+        @sync begin
+            for prob in nonlinear_problems
+                Threads.@spawn CommonSolve.solve(
+                )
+            end
+
+            for prob in nlls_problems
+                Threads.@spawn CommonSolve.solve(
+                )
+            end
+        end
+    end
+end
 
 # Rexexports
 @reexport using SciMLBase, NonlinearSolveBase, LineSearch, ADTypes