From 01db1e4e97d9a5e9aec46c1ae27bf69a06e1ac59 Mon Sep 17 00:00:00 2001
From: Julia Sloan <jsloan@caltech.edu>
Date: Thu, 29 Feb 2024 21:21:27 -0800
Subject: [PATCH] add GPU AMIP albedo map runs

---
 .buildkite/pipeline.yml                       | 18 +++++++++++++++
 .../gpu_amip_albedo_function.yml              |  2 +-
 .../gpu_amip_albedo_static_map.yml            | 22 +++++++++++++++++++
 .../gpu_amip_albedo_temporal_map.yml          | 22 +++++++++++++++++++
 .../gpu_slabplanet_albedo_temporal_map.yml    |  2 +-
 .../slabplanet_albedo_temporal_map.yml        |  2 +-
 6 files changed, 65 insertions(+), 3 deletions(-)
 create mode 100644 config/model_configs/gpu_amip_albedo_static_map.yml
 create mode 100644 config/model_configs/gpu_amip_albedo_temporal_map.yml

diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
index 40493b3397..628d703043 100644
--- a/.buildkite/pipeline.yml
+++ b/.buildkite/pipeline.yml
@@ -396,6 +396,24 @@ steps:
           slurm_mem: 20GB
           slurm_gpus: 1
 
+      - label: "GPU AMIP: albedo from static map"
+        key: "gpu_amip_albedo_static_map"
+        command: "julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/gpu_amip_albedo_static_map.yml"
+        artifact_paths: "experiments/AMIP/output/amip/gpu_amip_albedo_static_map_artifacts/*"
+        agents:
+          slurm_mem: 20GB
+          slurm_gpus: 1
+
+      - label: "GPU AMIP: albedo from temporal map"
+        key: "gpu_amip_albedo_temporal_map"
+        command: "julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/gpu_amip_albedo_temporal_map.yml"
+        artifact_paths: "experiments/AMIP/output/amip/gpu_amip_albedo_temporal_map_artifacts/*"
+        agents:
+          slurm_mem: 20GB
+          slurm_gpus: 1
+
+
+
       - wait
 
   - wait
diff --git a/config/model_configs/gpu_amip_albedo_function.yml b/config/model_configs/gpu_amip_albedo_function.yml
index e588f58459..8d126c791b 100644
--- a/config/model_configs/gpu_amip_albedo_function.yml
+++ b/config/model_configs/gpu_amip_albedo_function.yml
@@ -7,7 +7,7 @@ dt_rad: "1hours"
 dt_save_to_sol: "1days"
 dz_bottom: 30
 dz_top: 3000
-h_elem: 6
+h_elem: 4
 job_id: "gpu_amip_albedo_function"
 land_albedo_type: "function"
 mode_name: "amip"
diff --git a/config/model_configs/gpu_amip_albedo_static_map.yml b/config/model_configs/gpu_amip_albedo_static_map.yml
new file mode 100644
index 0000000000..eb95c07a72
--- /dev/null
+++ b/config/model_configs/gpu_amip_albedo_static_map.yml
@@ -0,0 +1,22 @@
+alpha_rayleigh_uh: 0
+alpha_rayleigh_w: 10
+apply_limiter: false
+dt: "150secs"
+dt_cpl: 150
+dt_rad: "1hours"
+dt_save_to_sol: "1days"
+dz_bottom: 30
+dz_top: 3000
+h_elem: 4
+job_id: "gpu_amip_albedo_static_map"
+land_albedo_type: "map_static"
+mode_name: "amip"
+moist: "equil"
+precip_model: "0M"
+rad: "gray"
+rayleigh_sponge: true
+run_name: "gpu_amip_albedo_static_map"
+t_end: "300secs"
+vert_diff: "true"
+z_elem: 50
+z_stretch: false
diff --git a/config/model_configs/gpu_amip_albedo_temporal_map.yml b/config/model_configs/gpu_amip_albedo_temporal_map.yml
new file mode 100644
index 0000000000..084eeb0e2b
--- /dev/null
+++ b/config/model_configs/gpu_amip_albedo_temporal_map.yml
@@ -0,0 +1,22 @@
+alpha_rayleigh_uh: 0
+alpha_rayleigh_w: 10
+apply_limiter: false
+dt: "150secs"
+dt_cpl: 150
+dt_rad: "1hours"
+dt_save_to_sol: "1days"
+dz_bottom: 30
+dz_top: 3000
+h_elem: 4
+job_id: "gpu_amip_albedo_temporal_map"
+land_albedo_type: "map_temporal"
+mode_name: "amip"
+moist: "equil"
+precip_model: "0M"
+rad: "gray"
+rayleigh_sponge: true
+run_name: "gpu_amip_albedo_temporal_map"
+t_end: "300secs"
+vert_diff: "true"
+z_elem: 50
+z_stretch: false
diff --git a/config/model_configs/gpu_slabplanet_albedo_temporal_map.yml b/config/model_configs/gpu_slabplanet_albedo_temporal_map.yml
index bdd6d61a08..c263918c09 100644
--- a/config/model_configs/gpu_slabplanet_albedo_temporal_map.yml
+++ b/config/model_configs/gpu_slabplanet_albedo_temporal_map.yml
@@ -4,7 +4,7 @@ dt: "200secs"
 dt_cpl: 200
 dt_save_to_sol: "3600secs"
 energy_check: true
-h_elem: 6
+h_elem: 4
 job_id: "gpu_slabplanet_albedo_temporal_map"
 land_albedo_type: "map_temporal"
 mode_name: "slabplanet"
diff --git a/config/model_configs/slabplanet_albedo_temporal_map.yml b/config/model_configs/slabplanet_albedo_temporal_map.yml
index 5d0dc21533..22bb505c97 100644
--- a/config/model_configs/slabplanet_albedo_temporal_map.yml
+++ b/config/model_configs/slabplanet_albedo_temporal_map.yml
@@ -3,7 +3,7 @@ dt: "200secs"
 dt_cpl: 200
 dt_save_to_sol: "3600secs"
 energy_check: true
-h_elem: 6
+h_elem: 4
 job_id: "slabplanet_albedo_temporal_map"
 land_albedo_type: "map_temporal"
 mode_name: "slabplanet"