fix soil/canopy performance run

CliMA · Jun 12, 2024 · 8b192f7 · 8b192f7
1 parent cf46982
commit 8b192f7
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/experiments/integrated/performance/profile_allocations.jl b/experiments/integrated/performance/profile_allocations.jl
@@ -330,10 +330,6 @@ land = SoilCanopyModel{FT}(;
 exp_tendency! = make_exp_tendency(land)
 imp_tendency! = make_imp_tendency(land);
 tendency_jacobian! = make_tendency_jacobian(land);
-jac_kwargs = (;
-    jac_prototype = ClimaLand.ImplicitEquationJacobian(Y),
-    Wfact = tendency_jacobian!,
-);
 
 # Set up timestepping and simulation callbacks
 dt = Float64(150)
@@ -354,6 +350,12 @@ driver_cb = ClimaLand.DriverUpdateCallback(updateat, updatefunc)
 # Set initial conditions
 Y, p = set_initial_conditions(land, t0)
 
+# Set up jacobian info
+jac_kwargs = (;
+    jac_prototype = ClimaLand.ImplicitEquationJacobian(Y),
+    Wfact = tendency_jacobian!,
+);
+
 # Solve simulation
 prob = SciMLBase.ODEProblem(
     CTS.ClimaODEFunction(