From 4d07ddc8792da701c95081a6ceda7c8cb94b913c Mon Sep 17 00:00:00 2001
From: Zhaoyi Shen <11598433+szy21@users.noreply.github.com>
Date: Tue, 14 Nov 2023 15:44:18 -0800
Subject: [PATCH] add gpu prognostic edmf aquaplanet

---
 .buildkite/pipeline.yml                       | 10 +++++++
 .../gpu_prognostic_edmfx_aquaplanet.yml       | 26 +++++++++++++++++++
 .../prognostic_edmfx_aquaplanet.yml           |  1 +
 3 files changed, 37 insertions(+)
 create mode 100644 config/gpu_configs/gpu_prognostic_edmfx_aquaplanet.yml

diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
index c097cbc77d..81fb9b35ea 100644
--- a/.buildkite/pipeline.yml
+++ b/.buildkite/pipeline.yml
@@ -701,6 +701,16 @@ steps:
         agents:
           slurm_gpus: 1
           slurm_mem: 20G
+      
+      - label: "GPU: Prognostic EDMFX aquaplanet"
+        command: >
+          julia --color=yes --project=examples examples/hybrid/driver.jl
+          --config_file $GPU_CONFIG_PATH/gpu_prognostic_edmfx_aquaplanet.yml
+        artifact_paths: "gpu_prognostic_edmfx_aquaplanet/*"
+        agents:
+          slurm_gpus: 1
+          slurm_mem: 20G
+        soft_fail: true
 
       - label: ":umbrella: GPU: Aquaplanet clearsky radiation and 1M microphysics"
         command: >
diff --git a/config/gpu_configs/gpu_prognostic_edmfx_aquaplanet.yml b/config/gpu_configs/gpu_prognostic_edmfx_aquaplanet.yml
new file mode 100644
index 0000000000..fee1ea6dc2
--- /dev/null
+++ b/config/gpu_configs/gpu_prognostic_edmfx_aquaplanet.yml
@@ -0,0 +1,26 @@
+job_id: gpu_prognostic_edmfx_aquaplanet
+surface_setup: DefaultMoninObukhov 
+rad: clearsky
+turbconv: prognostic_edmfx 
+prognostic_tke: true
+edmfx_upwinding: first_order 
+edmfx_entr_model: "Generalized" 
+edmfx_detr_model: "ConstantArea" 
+edmfx_nh_pressure: true 
+edmfx_velocity_relaxation: true
+edmfx_sgs_mass_flux: true
+edmfx_sgs_diffusive_flux: true
+moist: equil 
+precip_model: 0M
+dt: 10secs 
+t_end: 1hours 
+dt_save_to_disk: 600secs
+toml: [toml/prognostic_edmfx_box.toml]
+output_default_diagnostics: false
+diagnostics:
+  - short_name: [ts, ta, thetaa, ha, pfull, rhoa, ua, va, wa, hfes, hur, hus, cl, clw, cli, evspsbl, rsd, rsu, rld, rlu]
+    reduction_time: average
+    period: 1hours
+  - short_name: [arup, waup, taup, thetaaup, haup, husup, hurup, clwup, cliup, waen, tke, lmix]
+    reduction_time: average
+    period: 1hours
diff --git a/config/model_configs/prognostic_edmfx_aquaplanet.yml b/config/model_configs/prognostic_edmfx_aquaplanet.yml
index c663570057..177455a8d9 100644
--- a/config/model_configs/prognostic_edmfx_aquaplanet.yml
+++ b/config/model_configs/prognostic_edmfx_aquaplanet.yml
@@ -7,6 +7,7 @@ edmfx_upwinding: first_order
 edmfx_entr_model: "Generalized" 
 edmfx_detr_model: "ConstantArea" 
 edmfx_nh_pressure: true 
+edmfx_velocity_relaxation: true
 edmfx_sgs_mass_flux: true
 edmfx_sgs_diffusive_flux: true
 moist: equil