bring back compilation workload

Author: jClugstor

lib/NonlinearSolveQuasiNewton/src/NonlinearSolveQuasiNewton.jl

@@ -34,27 +34,27 @@ 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()]
-
-#     @compile_workload begin
-#         @sync for prob in nonlinear_problems, alg in algs
-#             Threads.@spawn CommonSolve.solve(
-#                 prob, alg; abstol = 1e-2, verbose = NonlinearVerbosity())
-#         end
-#     end
-# 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
+
+    algs = [Broyden(), Klement()]
+
+    @compile_workload begin
+        @sync for prob in nonlinear_problems, alg in algs
+            Threads.@spawn CommonSolve.solve(
+                prob, alg; abstol = 1e-2, verbose = NonlinearVerbosity())
+        end
+    end
+end
 
 @reexport using SciMLBase, NonlinearSolveBase
 

src/NonlinearSolve.jl

@@ -52,59 +52,57 @@ 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
-
-#     nonlinear_functions = (
-#         (NonlinearFunction{false, NoSpecialize}((u, p) -> (u .^ 2 .- p)[1:1]), [0.1, 0.0]),
-#         (
-#             NonlinearFunction{false, NoSpecialize}((u, p) -> vcat(u .* u .- p, u .* u .- p)),
-#             [0.1, 0.1]
-#         ),
-#         (
-#             NonlinearFunction{true, NoSpecialize}(
-#                 (du, u, p) -> du[1] = u[1] * u[1] - p, resid_prototype = zeros(1)
-#             ),
-#             [0.1, 0.0]
-#         ),
-#         (
-#             NonlinearFunction{true, NoSpecialize}(
-#                 (du, u, p) -> du .= vcat(u .* u .- p, u .* u .- p), resid_prototype = zeros(4)
-#             ),
-#             [0.1, 0.1]
-#         )
-#     )
-
-#     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
+@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]),
+        (
+            NonlinearFunction{false, NoSpecialize}((u, p) -> vcat(u .* u .- p, u .* u .- p)),
+            [0.1, 0.1]
+        ),
+        (
+            NonlinearFunction{true, NoSpecialize}(
+                (du, u, p) -> du[1] = u[1] * u[1] - p, resid_prototype = zeros(1)
+            ),
+            [0.1, 0.0]
+        ),
+        (
+            NonlinearFunction{true, NoSpecialize}(
+                (du, u, p) -> du .= vcat(u .* u .- p, u .* u .- p), resid_prototype = zeros(4)
+            ),
+            [0.1, 0.1]
+        )
+    )
+
+    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