Skip to content

Commit 277618d

Browse files
authored
Merge pull request #2945 from CliMA/zs/progedmf_gpu_scaling
add aquaplanet with prognostic edmf to gpu scaling pipeline
2 parents c8305b3 + 0288881 commit 277618d

File tree

2 files changed

+50
-0
lines changed

2 files changed

+50
-0
lines changed

.buildkite/gpu_pipeline/pipeline.yml

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,3 +250,21 @@ steps:
250250
slurm_ntasks: 1
251251
slurm_mem: 32G
252252
slurm_exclusive:
253+
254+
- group: "Prognostic EDMF GPU"
255+
steps:
256+
257+
- label: "gpu_aquaplanet_progedmf - 1 GPU"
258+
command:
259+
- mkdir -p gpu_aquaplanet_progedmf
260+
- >
261+
nsys profile --trace=nvtx,mpi,cuda,osrt --output=gpu_aquaplanet_progedmf/output_active/report
262+
julia --threads=3 --color=yes --project=examples examples/hybrid/driver.jl
263+
--config_file ${GPU_CONFIG_PATH}gpu_aquaplanet_progedmf.yml
264+
artifact_paths: "gpu_aquaplanet_progedmf/output_active/*"
265+
agents:
266+
slurm_gpus_per_task: 1
267+
slurm_cpus_per_task: 4
268+
slurm_ntasks: 1
269+
slurm_mem: 32G
270+
slurm_exclusive:
Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
job_id: gpu_aquaplanet_progedmf
2+
dt_save_state_to_disk: "Inf"
3+
dt_save_to_sol: "Inf"
4+
output_default_diagnostics: false
5+
h_elem: 30
6+
z_max: 55000.0
7+
z_elem: 63
8+
dz_bottom: 30.0
9+
dz_top: 3000.0
10+
moist: equil
11+
surface_setup: DefaultMoninObukhov
12+
rad: allskywithclear
13+
idealized_insolation: false
14+
dt_rad: 1hours
15+
dt_cloud_fraction: 1hours
16+
turbconv: prognostic_edmfx
17+
implicit_diffusion: true
18+
implicit_sgs_advection: true
19+
approximate_linear_solve_iters: 2
20+
max_newton_iters_ode: 3
21+
prognostic_tke: true
22+
edmfx_upwinding: first_order
23+
edmfx_entr_model: "Generalized"
24+
edmfx_detr_model: "Generalized"
25+
edmfx_nh_pressure: true
26+
edmfx_velocity_relaxation: true
27+
edmfx_sgs_mass_flux: true
28+
edmfx_sgs_diffusive_flux: true
29+
precip_model: 0M
30+
dt: 10secs
31+
t_end: 1hours
32+
toml: [toml/prognostic_edmfx.toml]

0 commit comments

Comments
 (0)