3689695d16
* [CI] Run RMM gtests. * Update test-cpp-gpu.sh --------- Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
24 lines
840 B
Bash
Executable File
24 lines
840 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
source tests/buildkite/conftest.sh
|
|
|
|
echo "--- Run Google Tests with CUDA, using a GPU"
|
|
buildkite-agent artifact download "build/testxgboost" . --step build-cuda
|
|
chmod +x build/testxgboost
|
|
tests/ci_build/ci_build.sh gpu nvidia-docker \
|
|
--build-arg CUDA_VERSION_ARG=$CUDA_VERSION \
|
|
--build-arg RAPIDS_VERSION_ARG=$RAPIDS_VERSION \
|
|
build/testxgboost
|
|
|
|
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 \
|
|
--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"
|