c5c8f643f2
XGBoost now uses CTK-11.8 for binary packages, there's no need to maintain a cub submodule anymore.
27 lines
782 B
Bash
Executable File
27 lines
782 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
source tests/buildkite/conftest.sh
|
|
|
|
echo "--- Build with CUDA ${CUDA_VERSION}, RMM enabled"
|
|
|
|
if [[ ($is_pull_request == 1) || ($is_release_branch == 0) ]]
|
|
then
|
|
arch_flag="-DGPU_COMPUTE_VER=75"
|
|
else
|
|
arch_flag=""
|
|
fi
|
|
|
|
command_wrapper="tests/ci_build/ci_build.sh rmm docker --build-arg "`
|
|
`"CUDA_VERSION_ARG=$CUDA_VERSION --build-arg "`
|
|
`"RAPIDS_VERSION_ARG=$RAPIDS_VERSION --build-arg "`
|
|
`"NCCL_VERSION_ARG=$NCCL_VERSION"
|
|
|
|
echo "--- Build libxgboost from the source"
|
|
$command_wrapper tests/ci_build/build_via_cmake.sh --conda-env=gpu_test -DUSE_CUDA=ON \
|
|
-DUSE_NCCL=ON -DPLUGIN_RMM=ON ${arch_flag}
|
|
|
|
echo "-- Stash C++ test executable (testxgboost)"
|
|
buildkite-agent artifact upload build/testxgboost
|