Skip to content

Commit

Permalink
Merge pull request #1533 from CliMA/sb/shared-depot
Browse files Browse the repository at this point in the history
try using a shared depot
  • Loading branch information
Sbozzolo authored Nov 8, 2023
2 parents 3b94a8a + 3620c08 commit cf10bd9
Showing 1 changed file with 11 additions and 5 deletions.
16 changes: 11 additions & 5 deletions .buildkite/pipeline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,16 +5,22 @@ agents:

env:
JULIA_LOAD_PATH: "${JULIA_LOAD_PATH}:${BUILDKITE_BUILD_CHECKOUT_PATH}/.buildkite"
JULIA_DEPOT_PATH: "${BUILDKITE_BUILD_PATH}/${BUILDKITE_PIPELINE_SLUG}/depot/default"
OPENBLAS_NUM_THREADS: 1
JULIA_NVTX_CALLBACKS: gc
OMPI_MCA_opal_warn_on_missing_libcuda: 0
# JULIA_CPU_TARGET: broadwell # arch used by GPU nodes
JULIA_CPU_TARGET: 'broadwell;skylake'
OMPI_MCA_opal_warn_on_missing_libcuda: 0

steps:
- label: "initialize"
key: "init_cpu_env"
command:
- |
if [ -n "$$CLEAR_DEPOT" ]; then
echo "--- Clearing depot"
rm -rf ${JULIA_DEPOT_PATH}
fi
- echo "--- Instantiate project"
- "julia --project -e 'using Pkg; Pkg.instantiate(;verbose=true); Pkg.precompile(;strict=true)'"
# force the initialization of the CUDA runtime as it is lazily loaded by default
Expand Down Expand Up @@ -43,9 +49,9 @@ steps:
agents:
slurm_mem: 80G
slurm_gpus: 1
slurm_cpus_per_task: 8
slurm_cpus_per_task: 12
env:
JULIA_NUM_PRECOMPILE_TASKS: 8
JULIA_NUM_PRECOMPILE_TASKS: 12

- wait

Expand All @@ -59,7 +65,7 @@ steps:
- label: "Unit: cuda"
key: unit_cuda_functional
command:
- "julia --project -e 'using CUDA; CUDA.versioninfo()'"
- "julia --project=test -e 'using CUDA; CUDA.versioninfo()'"
- "julia --color=yes --check-bounds=yes --project=test test/gpu/cuda.jl CUDA"
agents:
slurm_gpus: 1
Expand Down

0 comments on commit cf10bd9

Please sign in to comment.