resintate setup workload

Author: jClugstor

lib/NonlinearSolveFirstOrder/src/NonlinearSolveFirstOrder.jl

@@ -40,17 +40,17 @@ include("pseudo_transient.jl")
 include("poly_algs.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]),
@@ -73,13 +73,13 @@ include("forward_diff.jl")
         )
     )
 
-#     nlls_problems = NonlinearLeastSquaresProblem[]
-#     for (fn, u0) in nonlinear_functions
-#         push!(nlls_problems, NonlinearLeastSquaresProblem(fn, u0, 2.0))
-#     end
+    nlls_problems = NonlinearLeastSquaresProblem[]
+    for (fn, u0) in nonlinear_functions
+        push!(nlls_problems, NonlinearLeastSquaresProblem(fn, u0, 2.0))
+    end
 
-#     nlp_algs = [NewtonRaphson(), TrustRegion(), LevenbergMarquardt()]
-#     nlls_algs = [GaussNewton(), TrustRegion(), LevenbergMarquardt()]
+    nlp_algs = [NewtonRaphson(), TrustRegion(), LevenbergMarquardt()]
+    nlls_algs = [GaussNewton(), TrustRegion(), LevenbergMarquardt()]
 
     @compile_workload begin
         @sync begin