Try aquaplanet DYAMOND simulation on GPU

CliMA · Sep 18, 2023 · d5de646 · d5de646
1 parent cb44933
commit d5de646
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/.buildkite/gpu_pipeline/pipeline.yml b/.buildkite/gpu_pipeline/pipeline.yml
@@ -58,3 +58,14 @@ steps:
         agents:
           slurm_gpus: 1
           slurm_time: 23:00:00
+
+      - label: "gpu_aquaplanet_dyamond"
+        command: >
+          julia --project -e 'using CUDA; CUDA.versioninfo()'
+          
+          nsys profile --trace=nvtx,cuda --output=gpu_aquaplanet_dyamond/report julia --color=yes --project=examples examples/hybrid/driver.jl
+          --config_file ${GPU_CONFIG_PATH}gpu_aquaplanet_dyamond.yml
+        artifact_paths: "gpu_aquaplanet_dyamond/*"
+        agents:
+          slurm_gpus: 1
+          slurm_time: 23:00:00
diff --git a/config/gpu_configs/gpu_aquaplanet_dyamond.yml b/config/gpu_configs/gpu_aquaplanet_dyamond.yml
@@ -0,0 +1,20 @@
+h_elem: 60
+z_max: 55000.0
+z_elem: 63
+dz_bottom: 30.0
+dz_top: 3000.0
+kappa_4: 1e14
+moist: "equil" 
+precip_model: "0M" 
+rad: "allskywithclear" 
+idealized_insolation: false
+dt_rad: "1hours"
+vert_diff: "true" 
+surface_setup: "DefaultMoninObukhov" 
+rayleigh_sponge: true
+dt_save_to_disk: "3hours"
+dt: "50secs" 
+t_end: "1days"
+FLOAT_TYPE: "Float64"
+job_id: "gpu_aquaplanet_dyamond" 
+toml: [toml/longrun_aquaplanet_dyamond.toml]