Hoist ᶜgradᵥ-ᶠinterp args

Author: charleskawczynski

src/cache/precomputed_quantities.jl

@@ -101,6 +101,9 @@ function precomputed_quantities(Y, atmos)
             ᶜentrʲs = similar(Y.c, NTuple{n, FT}),
             ᶜdetrʲs = similar(Y.c, NTuple{n, FT}),
             ᶠnh_pressure₃ʲs = similar(Y.f, NTuple{n, C3{FT}}),
+            ᶜgradᵥ_θ_virt⁰ = Fields.Field(C3{FT}, cspace),
+            ᶜgradᵥ_q_tot⁰ = Fields.Field(C3{FT}, cspace),
+            ᶜgradᵥ_θ_liq_ice⁰ = Fields.Field(C3{FT}, cspace),
             precipitation_sgs_quantities...,
         ) : (;)
     sgs_quantities = (;

src/cache/prognostic_edmf_precomputed_quantities.jl

@@ -267,16 +267,21 @@ NVTX.@annotate function set_prognostic_edmf_precomputed_quantities_closures!(
         )
     end
 
+    (; ᶜgradᵥ_θ_virt⁰, ᶜgradᵥ_q_tot⁰, ᶜgradᵥ_θ_liq_ice⁰) = p.precomputed
     # First order approximation: Use environmental mean fields.
+    @. ᶜgradᵥ_θ_virt⁰ = ᶜgradᵥ(ᶠinterp(TD.virtual_pottemp(thermo_params, ᶜts⁰)))       # ∂θv∂z_unsat
+    @. ᶜgradᵥ_q_tot⁰ = ᶜgradᵥ(ᶠinterp(ᶜq_tot⁰))                                        # ∂qt∂z_sat
+    @. ᶜgradᵥ_θ_liq_ice⁰ =
+        ᶜgradᵥ(ᶠinterp(TD.liquid_ice_pottemp(thermo_params, ᶜts⁰)))                    # ∂θl∂z_sat
     @. ᶜlinear_buoygrad = buoyancy_gradients(
         BuoyGradMean(),
         thermo_params,
         moisture_model,
         ᶜts⁰,
         C3,
-        ᶜgradᵥ(ᶠinterp(TD.virtual_pottemp(thermo_params, ᶜts⁰))),    # ∂θv∂z_unsat
-        ᶜgradᵥ(ᶠinterp(ᶜq_tot⁰)),                                    # ∂qt∂z_sat
-        ᶜgradᵥ(ᶠinterp(TD.liquid_ice_pottemp(thermo_params, ᶜts⁰))), # ∂θl∂z_sat
+        ᶜgradᵥ_θ_virt⁰,
+        ᶜgradᵥ_q_tot⁰,
+        ᶜgradᵥ_θ_liq_ice⁰,
         ᶜlg,
     )