add GPU AMIP albedo map runs

CliMA · Mar 4, 2024 · 01db1e4 · 01db1e4
1 parent ab10ab9
commit 01db1e4
Show file tree

Hide file tree

Showing 6 changed files with 65 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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"