ws 1 GPU h_elem 30 [skip ci]

CliMA · Mar 8, 2024 · 1f675c1 · 1f675c1
1 parent 0b22c9b
commit 1f675c1
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 34 deletions.
diff --git 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
@@ -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"