From 7b976959f4284b97ae5b9ea5b620ce5b67d67078 Mon Sep 17 00:00:00 2001
From: Zhaoyi Shen <11598433+szy21@users.noreply.github.com>
Date: Sun, 23 Jun 2024 22:25:38 -0700
Subject: [PATCH] move progedmf longrun to gpu

---
 .buildkite/longruns/pipeline.yml              | 24 -------------------
 .buildkite/longruns_gpu/pipeline.yml          | 11 +++++++++
 ...5km_nz63_clearsky_progedmf_diffonly_0M.yml |  2 +-
 .../longrun_ssp_bw_rhoe_equil_highres.yml     | 15 ------------
 4 files changed, 12 insertions(+), 40 deletions(-)
 delete mode 100644 config/longrun_configs/longrun_ssp_bw_rhoe_equil_highres.yml

diff --git a/.buildkite/longruns/pipeline.yml b/.buildkite/longruns/pipeline.yml
index 98f0e77d6e..ce326f49b5 100644
--- a/.buildkite/longruns/pipeline.yml
+++ b/.buildkite/longruns/pipeline.yml
@@ -55,27 +55,3 @@ steps:
       #     slurm_time: 24:00:00
       #   env:
       #     JOB_NAME: "longrun_zalesak_tracer_energy_bw_rhoe_equil_highres"
-
-      - label: ":computer: SSP baroclinic wave (ρe_tot) equilmoist high resolution centered diff"
-        command:
-          - srun julia --project=examples examples/hybrid/driver.jl --config_file $CONFIG_PATH/$$JOB_NAME.yml --job_id $$JOB_NAME
-        artifact_paths: "$$JOB_NAME/output_active/*"
-        agents:
-          slurm_ntasks: 32
-          slurm_nodes: 2
-          slurm_time: 24:00:00
-        env:
-          CLIMACOMMS_CONTEXT: "MPI"
-          JOB_NAME: "longrun_ssp_bw_rhoe_equil_highres"
-
-      - label: ":computer: aquaplanet equilmoist clearsky radiation + prognostic edmf diffusion only + 0M microphysics"
-        command:
-          - srun julia --project=examples examples/hybrid/driver.jl --config_file $CONFIG_PATH/$$JOB_NAME.yml --job_id $$JOB_NAME
-        artifact_paths: "$$JOB_NAME/output_active/*"
-        agents:
-          slurm_ntasks: 64
-          slurm_nodes: 4
-          slurm_time: 24:00:00
-        env:
-          CLIMACOMMS_CONTEXT: "MPI"
-          JOB_NAME: "longrun_aquaplanet_rhoe_equil_55km_nz63_clearsky_progedmf_diffonly_0M"
diff --git a/.buildkite/longruns_gpu/pipeline.yml b/.buildkite/longruns_gpu/pipeline.yml
index c91b7e2b7e..89341cf89a 100644
--- a/.buildkite/longruns_gpu/pipeline.yml
+++ b/.buildkite/longruns_gpu/pipeline.yml
@@ -177,6 +177,17 @@ steps:
         env:
           CLIMACOMMS_DEVICE: "CUDA"
           JOB_NAME: "longrun_aquaplanet_rhoe_equil_55km_nz63_allsky_diagedmf_0M"
+      
+      - label: ":computer: aquaplanet equilmoist clearsky radiation + prognostic edmf diffusion only + 0M microphysics"
+        command:
+          - srun julia --project=examples examples/hybrid/driver.jl --config_file $CONFIG_PATH/$$JOB_NAME.yml --job_id $$JOB_NAME
+        artifact_paths: "$$JOB_NAME/output_active/*"
+        agents:
+          slurm_gpus: 1
+          slurm_time: 12:00:00
+        env:
+          CLIMACOMMS_DEVICE: "CUDA"
+          JOB_NAME: "longrun_aquaplanet_rhoe_equil_55km_nz63_clearsky_progedmf_diffonly_0M"
 
       - label: ":computer: aquaplanet equilmoist clearsky radiation + 0M microphysics + earth topography"
         command:
diff --git a/config/longrun_configs/longrun_aquaplanet_rhoe_equil_55km_nz63_clearsky_progedmf_diffonly_0M.yml b/config/longrun_configs/longrun_aquaplanet_rhoe_equil_55km_nz63_clearsky_progedmf_diffonly_0M.yml
index 3cfdf7b03f..6a3034c2e1 100644
--- a/config/longrun_configs/longrun_aquaplanet_rhoe_equil_55km_nz63_clearsky_progedmf_diffonly_0M.yml
+++ b/config/longrun_configs/longrun_aquaplanet_rhoe_equil_55km_nz63_clearsky_progedmf_diffonly_0M.yml
@@ -26,5 +26,5 @@ edmfx_sgs_diffusive_flux: true
 rayleigh_sponge: true
 dt_save_state_to_disk: "10days"
 dt: "100secs"
-t_end: "30days"
+t_end: "120days"
 toml: [toml/longrun_aquaplanet_rhoe_equil_55km_nz63_clearsky_progedmf_0M.toml]
diff --git a/config/longrun_configs/longrun_ssp_bw_rhoe_equil_highres.yml b/config/longrun_configs/longrun_ssp_bw_rhoe_equil_highres.yml
deleted file mode 100644
index d4e827c21e..0000000000
--- a/config/longrun_configs/longrun_ssp_bw_rhoe_equil_highres.yml
+++ /dev/null
@@ -1,15 +0,0 @@
-dt_save_state_to_disk: "10days"
-initial_condition: "MoistBaroclinicWave"
-approximate_linear_solve_iters: 2
-z_elem: 45
-dt: "150secs"
-t_end: "120days"
-dz_bottom: 30.0
-h_elem: 16
-ode_algo: "SSP333"
-precip_model: "0M"
-moist: "equil"
-apply_limiter: true
-diagnostics:
-  - short_name: [pfull, wa, va, rv, hus]
-    period: 1days