From 1f675c1c05e645a92cdc6fefa90f574b132381a0 Mon Sep 17 00:00:00 2001
From: Julia Sloan <jsloan@caltech.edu>
Date: Fri, 8 Mar 2024 14:54:20 -0800
Subject: [PATCH] ws 1 GPU h_elem 30 [skip ci]

---
 .buildkite/gpu/pipeline.yml             | 61 ++++++++++++-------------
 config/gpu_configs/gpu_amip_chap_ws.yml |  6 +--
 2 files changed, 33 insertions(+), 34 deletions(-)

diff --git a/.buildkite/gpu/pipeline.yml b/.buildkite/gpu/pipeline.yml
index b2a7fb6922..ee78539191 100644
--- a/.buildkite/gpu/pipeline.yml
+++ b/.buildkite/gpu/pipeline.yml
@@ -39,8 +39,8 @@ steps:
 
   - wait
 
-  - group: "CHAP GPU strong scaling"
-    steps:
+  # - group: "CHAP GPU strong scaling"
+  #   steps:
 
   #     - label: "GPU AMIP CHAP - strong scaling - 1 GPU"
   #       key: "gpu_amip_chap"
@@ -56,20 +56,20 @@ steps:
   #         slurm_mem: 32G
   #         slurm_exclusive:
 
-      - label: "GPU AMIP CHAP - strong scaling - 2 GPUs"
-        key: "gpu_amip_chap_2process"
-        command:
-          - >
-            srun --cpu-bind=threads --cpus-per-task=4
-            julia --threads=3 --color=yes --project=experiments/AMIP experiments/AMIP/coupler_driver.jl
-            --config_file $GPU_CONFIG_PATH/gpu_amip_chap_2process.yml
-        artifact_paths: "gpu_amip_chap_2process/*"
-        agents:
-          slurm_gpus_per_task: 1
-          slurm_cpus_per_task: 4
-          slurm_ntasks: 2
-          slurm_mem: 32G
-          slurm_exclusive:
+      # - label: "GPU AMIP CHAP - strong scaling - 2 GPUs"
+      #   key: "gpu_amip_chap_2process"
+      #   command:
+      #     - >
+      #       srun --cpu-bind=threads --cpus-per-task=4
+      #       julia --threads=3 --color=yes --project=experiments/AMIP experiments/AMIP/coupler_driver.jl
+      #       --config_file $GPU_CONFIG_PATH/gpu_amip_chap_2process.yml
+      #   artifact_paths: "gpu_amip_chap_2process/*"
+      #   agents:
+      #     slurm_gpus_per_task: 1
+      #     slurm_cpus_per_task: 4
+      #     slurm_ntasks: 2
+      #     slurm_mem: 32G
+      #     slurm_exclusive:
 
   #     - label: "GPU AMIP CHAP - strong scaling - 4 GPUs"
   #       key: "gpu_amip_chap_4process"
@@ -86,22 +86,21 @@ steps:
   #         slurm_mem: 32G
   #         slurm_exclusive:
 
-  # - group: "CHAP GPU weak scaling"
-  #   steps:
+  - group: "CHAP GPU weak scaling"
+    steps:
 
-  #     - label: "GPU AMIP CHAP - weak scaling - 1 GPU"
-  #       key: "gpu_amip_chap_ws"
-  #       command:
-  #         - >
-  #           julia --threads=3 --color=yes --project=experiments/AMIP experiments/AMIP/coupler_driver.jl
-  #           --config_file $GPU_CONFIG_PATH/gpu_amip_chap_ws.yml
-  #       artifact_paths: "gpu_amip_chap_ws/*"
-  #       agents:
-  #         slurm_gpus_per_task: 1
-  #         slurm_cpus_per_task: 4
-  #         slurm_ntasks: 1
-  #         slurm_mem: 32G
-  #         slurm_exclusive:
+      - label: "GPU AMIP CHAP - weak scaling - 1 GPU"
+        key: "gpu_amip_chap_ws"
+        command:
+          - >
+            julia --threads=3 --color=yes --project=experiments/AMIP experiments/AMIP/coupler_driver.jl --config_file $GPU_CONFIG_PATH/gpu_amip_chap_ws.yml
+        artifact_paths: "gpu_amip_chap_ws/*"
+        agents:
+          slurm_gpus_per_task: 1
+          slurm_cpus_per_task: 4
+          slurm_ntasks: 1
+          slurm_mem: 32G
+          slurm_exclusive:
 
       # - label: "GPU AMIP CHAP - weak scaling - 2 GPUs"
       #   key: "gpu_amip_chap_ws_2process"
diff --git a/config/gpu_configs/gpu_amip_chap_ws.yml b/config/gpu_configs/gpu_amip_chap_ws.yml
index d22cad8259..e24b359bd9 100644
--- a/config/gpu_configs/gpu_amip_chap_ws.yml
+++ b/config/gpu_configs/gpu_amip_chap_ws.yml
@@ -1,15 +1,15 @@
 anim: false
 apply_limiter: false
 atmos_config_file: "config/gpu_configs/gpu_aquaplanet_chap_ws_1process.yml"
-dt: "50secs"
+dt: "100secs"
 dt_cloud_fraction: "1hours"
-dt_cpl: 50
+dt_cpl: 100
 dt_rad: "1hours"
 dt_save_state_to_disk: "Inf"
 dt_save_to_sol: "Inf"
 energy_check: false
 evolving_ocean: false
-h_elem: 42
+h_elem: 30
 hourly_checkpoint: false
 job_id: "gpu_amip_chap_ws"
 land_albedo_type: "map_static"