Skip to content

Commit

Permalink
first pass long run
Browse files Browse the repository at this point in the history
  • Loading branch information
kmdeck committed Jul 2, 2024
1 parent c4d4279 commit 95826ff
Show file tree
Hide file tree
Showing 2 changed files with 678 additions and 0 deletions.
47 changes: 47 additions & 0 deletions .buildkite/longruns_gpu/pipeline.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
agents:
queue: clima
slurm_mem: 8G
modules: common

env:
JULIA_NVTX_CALLBACKS: gc
JULIA_MAX_NUM_PRECOMPILE_FILES: 100
OPENBLAS_NUM_THREADS: 1
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:
- 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"
Loading

0 comments on commit 95826ff

Please sign in to comment.