Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CI] Build pip wheel with RMM support (#9383)

Browse Source
This commit is contained in:
Philip Hyunsu Cho
2023-07-18 01:52:26 -07:00
committed by GitHub
parent 6e18d3a290
commit e082718c66
8 changed files with 64 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/buildkite/test-cpp-gpu.sh
View File

@@ -16,8 +16,8 @@ echo "--- Run Google Tests with CUDA, using a GPU, RMM enabled"
rm -rfv build/
buildkite-agent artifact download "build/testxgboost" . --step build-cuda-with-rmm
chmod +x build/testxgboost
tests/ci_build/ci_build.sh rmm nvidia-docker \
tests/ci_build/ci_build.sh gpu nvidia-docker \
--build-arg CUDA_VERSION_ARG=$CUDA_VERSION \
--build-arg RAPIDS_VERSION_ARG=$RAPIDS_VERSION \
--build-arg NCCL_VERSION_ARG=$NCCL_VERSION bash -c \
"source activate gpu_test && build/testxgboost --use-rmm-pool"
--build-arg NCCL_VERSION_ARG=$NCCL_VERSION \
build/testxgboost --use-rmm-pool