Skip to content

Commit

Permalink
add atmos long run pipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
kmdeck committed Jul 2, 2024
1 parent cc89f7e commit 90e5748
Showing 1 changed file with 49 additions and 0 deletions.
49 changes: 49 additions & 0 deletions .buildkite/longruns_gpu/pipeline.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
agents:
queue: clima
slurm_mem: 8G
modules: julia/1.10.0 cuda/julia-pref openmpi/4.1.5-mpitrampoline nsight-systems/2024.2.1

env:
JULIA_MPI_HAS_CUDA: "true"
JULIA_NVTX_CALLBACKS: gc
JULIA_MAX_NUM_PRECOMPILE_FILES: 100
OPENBLAS_NUM_THREADS: 1
OMPI_MCA_opal_warn_on_missing_libcuda: 0
SLURM_KILL_BAD_EXIT: 1
SLURM_GRES_FLAGS: "allow-task-sharing"
CONFIG_PATH: "config/longrun_configs"

timeout_in_minutes: 1440

steps:
- label: "init :GPU:"
key: "init_gpu_env"
command:
- echo "--- Instantiate"
- julia --project=.buildkite -e 'using Pkg; Pkg.instantiate(;verbose=true)'
- julia --project=.buildkite -e 'using Pkg; Pkg.precompile()'
- julia --project=.buildkite -e 'using CUDA; CUDA.precompile_runtime()'
- julia --project=.buildkite -e 'using Pkg; Pkg.status()'

agents:
slurm_gpus: 1
slurm_cpus_per_task: 8
env:
JULIA_NUM_PRECOMPILE_TASKS: 8
JULIA_MAX_NUM_PRECOMPILE_FILES: 50

- wait

- group: "Global Land Models"
steps:

- label: ":seedling: Soil-Canopy"
command:
- srun julia --color=yes --project=.buildkite experiments/long_runs/land.jl
artifact_paths: "$$JOB_NAME/output_active/*"
agents:
slurm_gpus: 1
slurm_time: 12:00:00
env:
CLIMACOMMS_DEVICE: "CUDA"
JOB_NAME: "soil_canopy"

0 comments on commit 90e5748

Please sign in to comment.