From 87d06c5bc8c3010b3682062797679904289c761a Mon Sep 17 00:00:00 2001 From: Hyunsu Philip Cho Date: Sat, 11 Feb 2023 22:23:57 -0800 Subject: [PATCH] Allocate extra space in /dev/shm to enable NCCL --- tests/buildkite/test-cpp-gpu.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/buildkite/test-cpp-gpu.sh b/tests/buildkite/test-cpp-gpu.sh index 9bfed6864126..0033b792212e 100755 --- a/tests/buildkite/test-cpp-gpu.sh +++ b/tests/buildkite/test-cpp-gpu.sh @@ -4,6 +4,9 @@ set -euo pipefail source tests/buildkite/conftest.sh +# Allocate extra space in /dev/shm to enable NCCL +export CI_DOCKER_EXTRA_PARAMS_INIT='--shm-size=4g' + echo "--- Run Google Tests with CUDA, using a GPU" buildkite-agent artifact download "build/testxgboost" . --step build-cuda chmod +x build/testxgboost