Skip to content

Commit

Permalink
add GPU DYAMOND runs
Browse files Browse the repository at this point in the history
  • Loading branch information
juliasloan25 committed Mar 6, 2024
1 parent ef2b3ce commit 4feeb75
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 1 deletion.
15 changes: 14 additions & 1 deletion .buildkite/longruns/pipeline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,6 @@ steps:
steps:

# DYAMOND AMIP: 1 day (convection resolving)

- label: "MPI AMIP SUPERFINE: dyamond_target"
key: "dyamond_target"
command: "srun julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/dyamond_target.yml"
Expand Down Expand Up @@ -357,6 +356,20 @@ steps:
slurm_nodes: 1
slurm_mem_per_cpu: 16G

- group: "GPU AMIP targets"

steps:

# DYAMOND AMIP: 1 day (convection resolving)
- label: "GPU AMIP SUPERFINE: dyamond_target"
key: "gpu_dyamond_target"
command: "julia --color=yes --project=experiments/AMIP/ experiments/AMIP/coupler_driver.jl --config_file $CONFIG_PATH/gpu_dyamond_target.yml"
artifact_paths: "experiments/AMIP/output/amip/gpu_dyamond_target_artifacts/*"
agents:
slurm_mem: 80GB
slurm_gpus: 1
gres: "gpu:v100:1,gpu:h100:1"

- wait

# plot job performance history
Expand Down
15 changes: 15 additions & 0 deletions config/longrun_configs/gpu_dyamond_target.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
anim: false
atmos_config_file: "config/longrun_configs/longrun_aquaplanet_dyamond.yml"
dt_cpl: 50
dt_save_state_to_disk: "0.5days"
dt_save_to_sol: "0.5days"
energy_check: false
job_id: "gpu_dyamond_target"
land_albedo_type: "map_temporal"
mode_name: "amip"
mono_surface: false
monthly_checkpoint: false
run_name: "gpu_dyamond_target"
start_date: "19790301"
t_end: "1days"
turb_flux_partition: "CombinedStateFluxes"

0 comments on commit 4feeb75

Please sign in to comment.