Flatten buoyancy_gradients call

Author: charleskawczynski

src/cache/diagnostic_edmf_precomputed_quantities.jl

@@ -782,16 +782,12 @@ NVTX.@annotate function set_diagnostic_edmf_precomputed_quantities_env_closures!
         BuoyGradMean(),
         thermo_params,
         moisture_model,
-        EnvBuoyGradVars(
-            ᶜts,
-            projected_vector_buoy_grad_vars(
-                C3,
-                p.precomputed.ᶜgradᵥ_θ_virt,    # ∂θv∂z_unsat
-                p.precomputed.ᶜgradᵥ_q_tot,     # ∂qt∂z_sat
-                p.precomputed.ᶜgradᵥ_θ_liq_ice, # ∂θl∂z_sat
-                ᶜlg,
-            ),
-        ),
+        ᶜts,
+        C3,
+        p.precomputed.ᶜgradᵥ_θ_virt,    # ∂θv∂z_unsat
+        p.precomputed.ᶜgradᵥ_q_tot,     # ∂qt∂z_sat
+        p.precomputed.ᶜgradᵥ_θ_liq_ice, # ∂θl∂z_sat
+        ᶜlg,
     )
 
     # TODO: Currently the shear production only includes vertical gradients

src/cache/prognostic_edmf_precomputed_quantities.jl

@@ -272,16 +272,12 @@ NVTX.@annotate function set_prognostic_edmf_precomputed_quantities_closures!(
         BuoyGradMean(),
         thermo_params,
         moisture_model,
-        EnvBuoyGradVars(
-            ᶜts⁰,
-            projected_vector_buoy_grad_vars(
-                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
-                ᶜlg,
-            ),
-        ),
+        ᶜ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
+        ᶜlg,
     )
 
     # TODO: Currently the shear production only includes vertical gradients

src/prognostic_equations/buoyancy_gradients.jl

@@ -34,6 +34,36 @@ The dispatch on EnvBuoyGradVars type is performed at the EnvBuoyGradVars constru
 used here are consistent for both mean fields and conditional fields obtained from assumed distributions
 over the conserved thermodynamic variables.
 """
+function buoyancy_gradients end
+
+function buoyancy_gradients(
+    ebgc::AbstractEnvBuoyGradClosure,
+    thermo_params,
+    moisture_model,
+    ts,
+    ::Type{C3},
+    ∂θv∂z_unsat,
+    ∂qt∂z_sat,
+    ∂θl∂z_sat,
+    ᶜlg,
+) where {C3}
+    return buoyancy_gradients(
+        ebgc,
+        thermo_params,
+        moisture_model,
+        EnvBuoyGradVars(
+            ts,
+            projected_vector_buoy_grad_vars(
+                C3,
+                ∂θv∂z_unsat,
+                ∂qt∂z_sat,
+                ∂θl∂z_sat,
+                ᶜlg,
+            ),
+        ),
+    )
+end
+
 function buoyancy_gradients(
     ebgc::AbstractEnvBuoyGradClosure,
     thermo_params,

src/prognostic_equations/gm_sgs_closures.jl

@@ -36,16 +36,12 @@ NVTX.@annotate function compute_gm_mixing_length!(ᶜmixing_length, Y, p)
         BuoyGradMean(),
         thermo_params,
         p.atmos.moisture_model,
-        EnvBuoyGradVars(
-            ᶜts,
-            projected_vector_buoy_grad_vars(
-                C3,
-                p.precomputed.ᶜgradᵥ_θ_virt,    # ∂θv∂z_unsat
-                p.precomputed.ᶜgradᵥ_q_tot,     # ∂qt∂z_sat
-                p.precomputed.ᶜgradᵥ_θ_liq_ice, # ∂θl∂z_sat
-                ᶜlg,
-            ),
-        ),
+        ᶜts,
+        C3,
+        p.precomputed.ᶜgradᵥ_θ_virt,    # ∂θv∂z_unsat
+        p.precomputed.ᶜgradᵥ_q_tot,     # ∂qt∂z_sat
+        p.precomputed.ᶜgradᵥ_θ_liq_ice, # ∂θl∂z_sat
+        ᶜlg,
     )
 
     ᶠu = p.scratch.ᶠtemp_C123