turn p.dt into an FT

Author: nefrathenrici

.buildkite/pipeline.yml

@@ -1281,7 +1281,7 @@ steps:
           CLIMACOMMS_DEVICE: "CUDA"
           CLIMA_NAME_CUDA_KERNELS_FROM_STACK_TRACE: "true"
         agents:
-          slurm_mem: 32GB
+          slurm_mem: 24GB
           slurm_gpus: 1
 
   - group: "Flame graphs"

post_processing/jacobian_summary.jl

@@ -19,7 +19,7 @@ function print_jacobian_summary(integrator)
 
     FT = eltype(Y)
     γs = filter(!iszero, CA.LinearAlgebra.diag(tableau_coefficients))
-    dtγ = FT(dt) * FT(γs[end])
+    dtγ = dt * FT(γs[end])
     scalar_names = CA.scalar_field_names(Y)
     block_keys = Iterators.product(scalar_names, scalar_names)
 

src/cache/cache.jl

@@ -1,5 +1,5 @@
 struct AtmosCache{
-    TT,
+    FT,
     AM,
     NUM,
     CAP,
@@ -20,7 +20,7 @@ struct AtmosCache{
     CONSCHECK,
 }
     """Timestep of the simulation (in seconds). This is also used by callbacks and tendencies"""
-    dt::TT
+    dt::FT
 
     """AtmosModel"""
     atmos::AM
@@ -94,6 +94,7 @@ function build_cache(
 )
     (; dt, start_date, output_dir) = sim_info
     FT = eltype(params)
+    dt = FT(dt)
 
     ᶜcoord = Fields.local_geometry_field(Y.c).coordinates
     ᶠcoord = Fields.local_geometry_field(Y.f).coordinates

src/cache/diagnostic_edmf_precomputed_quantities.jl

@@ -449,7 +449,6 @@ NVTX.@annotate function set_diagnostic_edmf_precomputed_quantities_do_integral!(
     ᶜdz = Fields.Δz_field(axes(Y.c))
     (; params) = p
     (; dt) = p
-    dt = FT(dt)
     (; ᶜΦ, ᶜgradᵥ_ᶠΦ) = p.core
     (; ᶜp, ᶠu³, ᶜts, ᶜK) = p.precomputed
     (;

src/cache/precipitation_precomputed_quantities.jl

@@ -580,8 +580,6 @@ sources from the sub-domains.
 set_precipitation_cache!(Y, p, _, _) = nothing
 function set_precipitation_cache!(Y, p, ::Microphysics0Moment, _)
     (; params, dt) = p
-    FT = eltype(p.params)
-    dt = FT(dt)
     (; ᶜts) = p.precomputed
     (; ᶜS_ρq_tot, ᶜS_ρe_tot) = p.precomputed
     (; ᶜΦ) = p.core

src/cache/prognostic_edmf_precomputed_quantities.jl

@@ -283,7 +283,7 @@ NVTX.@annotate function set_prognostic_edmf_precomputed_quantities_explicit_clos
                 buoyancy_flux_val < 0 || ᶜaʲ_int_val >= $(FT(turbconv_params.surface_area)),
                 entr_int_val,
                 detr_int_val +
-                ($(FT(turbconv_params.surface_area)) / ᶜaʲ_int_val - 1) / FT(dt),
+                ($(FT(turbconv_params.surface_area)) / ᶜaʲ_int_val - 1) / dt,
             ),
             ᶜaʲ_int_val,
             dt,

src/callbacks/callbacks.jl

@@ -128,7 +128,7 @@ NVTX.@annotate function rrtmgp_model_callback!(integrator)
     Y = integrator.u
     p = integrator.p
     t = integrator.t
-
+    FT = eltype(Y)
     (; params) = p
     (; ᶠradiation_flux, rrtmgp_model) = p.radiation
     (; radiation_mode) = p.atmos
@@ -138,7 +138,7 @@ NVTX.@annotate function rrtmgp_model_callback!(integrator)
     set_insolation_variables!(Y, p, t, p.atmos.insolation)
     set_surface_albedo!(Y, p, t, p.atmos.surface_albedo)
 
-    RRTMGPI.update_fluxes!(rrtmgp_model, UInt32(floor(t / integrator.p.dt)))
+    RRTMGPI.update_fluxes!(rrtmgp_model, UInt32(floor(FT(t) / integrator.p.dt)))
     Fields.field2array(ᶠradiation_flux) .= rrtmgp_model.face_flux
     return nothing
 end

src/parameterized_tendencies/microphysics/microphysics_wrappers.jl

@@ -25,15 +25,14 @@ end
 # Helper function to compute the limit of the tendency in the traingle limiter.
 # The limit is defined as the available amont / n times model time step.
 function limit(q, dt, n::Int)
-    FT = eltype(q)
-    return q / FT(dt) / n
+    return q / dt / n
 end
 
 function moisture_fixer(q, qᵥ, dt)
     FT = eltype(q)
     return triangle_inequality_limiter(
-        -min(FT(0), q / FT(dt)),
-        limit(qᵥ, FT(dt), 5),
+        -min(FT(0), q / dt),
+        limit(qᵥ, dt, 5),
         FT(0),
     )
 end
@@ -161,10 +160,9 @@ Returns the qₜ source term due to precipitation formation
 defined as Δm_tot / (m_dry + m_tot) for the 0-moment scheme
 """
 function q_tot_0M_precipitation_sources(thp, cmp::CMP.Parameters0M, dt, qₜ, ts)
-    FT = eltype(qₜ)
     return -triangle_inequality_limiter(
         -CM0.remove_precipitation(cmp, PP(thp, ts)),
-        qₜ / FT(dt),
+        qₜ / dt,
     )
 end
 
@@ -570,7 +568,7 @@ function aerosol_activation_sources(
     return ifelse(
         S_max < S || isnan(n_act) || n_act < nₗ,
         FT(0),
-        (n_act - nₗ) / FT(dt),
+        (n_act - nₗ) / dt,
     )
 end
 

src/prognostic_equations/edmfx_entr_detr.jl

@@ -664,7 +664,7 @@ function edmfx_first_interior_entr_tendency!(
         sgsʲs_ρaₜ_int_val = Fields.field_values(Fields.level(Yₜ.c.sgsʲs.:($j).ρa, 1))
         @. sgsʲs_ρaₜ_int_val += ifelse(buoyancy_flux_val < 0,
             0,
-            max(0, (sgsʲs_ρ_int_val * $(eps(FT)) - sgsʲs_ρa_int_val) / FT(dt)),
+            max(0, (sgsʲs_ρ_int_val * $(eps(FT)) - sgsʲs_ρa_int_val) / dt),
         )
 
         # Apply entrainment tendencies in the first model cell for moist static energy (mse) 
@@ -718,16 +718,16 @@ end
 limit_entrainment(entr::FT, a, dt) where {FT} = max(
     min(
         entr,
-        FT(0.9) * (1 - a) / max(a, eps(FT)) / FT(dt),
-        FT(0.9) * 1 / FT(dt),
+        FT(0.9) * (1 - a) / max(a, eps(FT)) / dt,
+        FT(0.9) * 1 / dt,
     ),
     0,
 )
 limit_detrainment(detr::FT, a, dt) where {FT} =
-    max(min(detr, FT(0.9) * 1 / FT(dt)), 0)
+    max(min(detr, FT(0.9) * 1 / dt), 0)
 
 function limit_turb_entrainment(dyn_entr::FT, turb_entr, dt) where {FT}
-    return max(min((FT(0.9) * 1 / FT(dt)) - dyn_entr, turb_entr), 0)
+    return max(min((FT(0.9) * 1 / dt) - dyn_entr, turb_entr), 0)
 end
 
 # limit entrainment and detrainment rates for diagnostic EDMF

src/prognostic_equations/edmfx_sgs_flux.jl

@@ -448,7 +448,7 @@ function edmfx_sgs_diffusive_flux_tendency!(
                         ᶜtke⁰,
                         ᶜmixing_length_field,
                     ),
-                    Y.c.sgs⁰.ρatke / FT(dt),
+                    Y.c.sgs⁰.ρatke / dt,
                 )
         end
         if !(p.atmos.moisture_model isa DryModel)
@@ -595,7 +595,7 @@ function edmfx_sgs_diffusive_flux_tendency!(
                         ᶜtke⁰,
                         ᶜmixing_length_field,
                     ),
-                    Y.c.sgs⁰.ρatke / FT(dt),
+                    Y.c.sgs⁰.ρatke / dt,
                 )
         end