From 4fb9922892f424cf8c0ab3dcdfd82bf8ab449ac2 Mon Sep 17 00:00:00 2001 From: sriharshakandala Date: Mon, 9 Oct 2023 10:11:50 -0700 Subject: [PATCH] Try aquaplanet DYAMOND simulation on GPU --- .buildkite/gpu_pipeline/pipeline.yml | 12 +++++++++++ config/gpu_configs/gpu_aquaplanet_dyamond.yml | 20 +++++++++++++++++++ 2 files changed, 32 insertions(+) create mode 100644 config/gpu_configs/gpu_aquaplanet_dyamond.yml diff --git a/.buildkite/gpu_pipeline/pipeline.yml b/.buildkite/gpu_pipeline/pipeline.yml index e4d401a82b..61a1818be2 100644 --- a/.buildkite/gpu_pipeline/pipeline.yml +++ b/.buildkite/gpu_pipeline/pipeline.yml @@ -56,6 +56,18 @@ steps: slurm_gpus: 1 slurm_time: 23:00:00 + - label: "gpu_aquaplanet_dyamond" + command: + - mkdir -p gpu_aquaplanet_dyamond + - > + 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 + - label: "moist Held-Suarez" key: "gpu_hs_rhoe_equilmoist_nz63_0M_55km_rs35km" command: diff --git a/config/gpu_configs/gpu_aquaplanet_dyamond.yml b/config/gpu_configs/gpu_aquaplanet_dyamond.yml new file mode 100644 index 0000000000..6c485f00eb --- /dev/null +++ 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]