Add support for 2D simulations on GPUs

CliMA · Jul 16, 2024 · a689b49 · a689b49
1 parent 5390243
commit a689b49
Show file tree

Hide file tree

Showing 7 changed files with 882 additions and 28 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -904,17 +904,39 @@ steps:
         agents:
           slurm_gpus: 1
 
+      - label: "GPU: Analytic Schar Mountain Test (2D, High Resolution)"
+        command: >
+          julia --color=yes --project=examples examples/hybrid/driver.jl
+          --config_file $CONFIG_PATH/plane_analytic_schar_mountain_high_res_test.yml
+          --job_id gpu_plane_analytic_schar_mountain_high_res_test
+        artifact_paths: "gpu_plane_analytic_schar_mountain_high_res_test/output_active/*"
+        env:
+          CLIMACOMMS_DEVICE: "CUDA"
+        agents:
+          slurm_gpus: 1
+
       - label: "GPU: Analytic Cosine Mountain Test (2D)"
         command: >
           julia --color=yes --project=examples examples/hybrid/driver.jl
-          --config_file $CONFIG_PATH/gpu_plane_analytic_cosine_mountain_test.yml
+          --config_file $CONFIG_PATH/plane_analytic_cosine_mountain_test.yml
           --job_id gpu_plane_analytic_cosine_mountain_test
         artifact_paths: "gpu_plane_analytic_cosine_mountain_test/output_active/*"
         env:
           CLIMACOMMS_DEVICE: "CUDA"
         agents:
           slurm_gpus: 1
 
+      - label: "GPU: Analytic Cosine Mountain Test (Extruded 2D)"
+        command: >
+          julia --color=yes --project=examples examples/hybrid/driver.jl
+          --config_file $CONFIG_PATH/extruded_plane_analytic_cosine_mountain_test.yml
+          --job_id gpu_extruded_plane_analytic_cosine_mountain_test
+        artifact_paths: "gpu_extruded_plane_analytic_cosine_mountain_test/output_active/*"
+        env:
+          CLIMACOMMS_DEVICE: "CUDA"
+        agents:
+          slurm_gpus: 1
+
       - label: "GPU: Analytic Cosine Mountain Test (3D)"
         command: >
           julia --color=yes --project=examples examples/hybrid/driver.jl

diff --git a/...u_plane_analytic_cosine_mountain_test.yml → ...d_plane_analytic_cosine_mountain_test.yml b/...u_plane_analytic_cosine_mountain_test.yml → ...d_plane_analytic_cosine_mountain_test.yml
diff --git a/config/model_configs/plane_analytic_schar_mountain_high_res_test.yml b/config/model_configs/plane_analytic_schar_mountain_high_res_test.yml
@@ -0,0 +1,18 @@
+config: "plane"
+FLOAT_TYPE: "Float64"
+initial_condition: "ConstantBuoyancyFrequencyProfile"
+topography: "Schar"
+x_max: 100e3
+z_max: 21e3
+x_elem: 250
+z_elem: 100
+z_stretch: false
+dt: "0.2secs"
+t_end: "24hours"
+rayleigh_sponge: true
+toml: [toml/analytic_mountain_test.toml]
+analytic_check: true
+output_default_diagnostics: false
+diagnostics:
+  - short_name: [orog, ua, wa, uapredicted, wapredicted, uaerror, waerror]
+    period: 1hours
diff --git a/config/model_configs/plane_analytic_schar_mountain_test.yml b/config/model_configs/plane_analytic_schar_mountain_test.yml
@@ -1,16 +1,14 @@
-config: "box"
+config: "plane"
 FLOAT_TYPE: "Float64"
 initial_condition: "ConstantBuoyancyFrequencyProfile"
 topography: "Schar"
 x_max: 100e3
-y_max: 2e3
 z_max: 21e3
 x_elem: 100
-y_elem: 2
 z_elem: 100
 z_stretch: false
 dt: "0.5secs"
-t_end: "18hours"
+t_end: "24hours"
 rayleigh_sponge: true
 toml: [toml/analytic_mountain_test.toml]
 analytic_check: true