From 5e693343049d292c47060a33622db94c068a1bdb Mon Sep 17 00:00:00 2001
From: Zhaoyi Shen <11598433+szy21@users.noreply.github.com>
Date: Thu, 1 Feb 2024 14:00:12 -0800
Subject: [PATCH] add aquaplanet setup for chap proposal

---
 .buildkite/gpu_pipeline/pipeline.yml          | 12 ++++++++
 config/gpu_configs/gpu_aquaplanet_chap.yml    | 29 +++++++++++++++++++
 .../gpu_diagnostic_edmfx_aquaplanet.yml       | 13 +++++----
 3 files changed, 48 insertions(+), 6 deletions(-)
 create mode 100644 config/gpu_configs/gpu_aquaplanet_chap.yml

diff --git a/.buildkite/gpu_pipeline/pipeline.yml b/.buildkite/gpu_pipeline/pipeline.yml
index cee97ac7a6..31e787a7b5 100644
--- 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
new file mode 100644
index 0000000000..9841e94ddf
--- /dev/null
+++ 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
index 5864810eb3..dd6d3491cd 100644
--- 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]