add aquaplanet setup for chap proposal

CliMA · Feb 2, 2024 · 5e69334 · 5e69334
1 parent 108a00f
commit 5e69334
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 6 deletions.
diff --git a/.buildkite/gpu_pipeline/pipeline.yml b/.buildkite/gpu_pipeline/pipeline.yml
@@ -67,6 +67,18 @@ steps:
         agents:
           slurm_gpus: 1
           slurm_cpus_per_task: 4
+
+      - label: "gpu_aquaplanet_chap"
+        command:
+          - mkdir -p gpu_aquaplanet_chap
+          - >
+            nsys profile --trace=nvtx,mpi,cuda,osrt --output=gpu_aquaplanet_chap/report
+            julia --threads=3 --color=yes --project=examples examples/hybrid/driver.jl
+            --config_file ${GPU_CONFIG_PATH}gpu_aquaplanet_chap.yml
+        artifact_paths: "gpu_aquaplanet_chap/*"
+        agents:
+          slurm_gpus: 1
+          slurm_cpus_per_task: 4
 
       - label: "moist Held-Suarez"
         key: "gpu_hs_rhoe_equil_55km_nz63_0M"

diff --git a/config/gpu_configs/gpu_aquaplanet_chap.yml b/config/gpu_configs/gpu_aquaplanet_chap.yml
@@ -0,0 +1,29 @@
+job_id: gpu_aquaplanet_chap
+dt_save_state_to_disk: "Inf"
+dt_save_to_sol: "Inf"
+output_default_diagnostics: false
+h_elem: 16
+z_max: 55000.0
+z_elem: 63
+dz_bottom: 30.0
+dz_top: 3000.0
+moist: equil 
+surface_setup: DefaultMoninObukhov 
+rad: allskywithclear
+idealized_insolation: false
+dt_rad: 1hours
+dt_cloud_fraction: 1hours
+turbconv: diagnostic_edmfx 
+implicit_diffusion: true
+approximate_linear_solve_iters: 2
+prognostic_tke: true
+edmfx_upwinding: first_order 
+edmfx_entr_model: "Generalized" 
+edmfx_detr_model: "Generalized" 
+edmfx_nh_pressure: true 
+edmfx_sgs_mass_flux: true
+edmfx_sgs_diffusive_flux: true
+precip_model: 0M
+dt: 100secs 
+t_end: 1days 
+toml: [toml/diagnostic_edmfx_box.toml]
diff --git a/config/gpu_configs/gpu_diagnostic_edmfx_aquaplanet.yml b/config/gpu_configs/gpu_diagnostic_edmfx_aquaplanet.yml
@@ -1,10 +1,11 @@
-dt_save_state_to_disk: "Inf"
+dt_save_state_to_disk: "3hours"
 dt_save_to_sol: "Inf"
-output_default_diagnostics: false
 job_id: gpu_diagnostic_edmfx_aquaplanet
-surface_setup: DefaultExchangeCoefficients 
-rad: gray
+surface_setup: DefaultMoninObukhov 
+rad: clearsky
 turbconv: diagnostic_edmfx 
+implicit_diffusion: true
+approximate_linear_solve_iters: 2
 prognostic_tke: true
 edmfx_upwinding: first_order 
 edmfx_entr_model: "Generalized" 
@@ -14,6 +15,6 @@ edmfx_sgs_mass_flux: true
 edmfx_sgs_diffusive_flux: true
 moist: equil 
 precip_model: 0M
-dt: 10secs 
-t_end: 1hours 
+dt: 100secs 
+t_end: 6hours 
 toml: [toml/diagnostic_edmfx_box.toml]