6d8afb2218
* Update to C++17 * Turn off unity build * Update CMake to 3.18 * Use MSVC 2022 + CUDA 11.8 * Re-create stack for worker images * Allocate more disk space for Windows * Tempiorarily disable clang-tidy * RAPIDS now requires Python 3.10+ * Unpin cuda-python * Use latest NCCL * Use Ubuntu 20.04 in RMM image * Mark failing mgpu test as xfail
21 lines
485 B
Bash
Executable File
21 lines
485 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
source tests/buildkite/conftest.sh
|
|
|
|
echo "--- Build XGBoost JVM packages with CUDA"
|
|
|
|
if [[ ($is_pull_request == 1) || ($is_release_branch == 0) ]]
|
|
then
|
|
arch_flag="-DGPU_COMPUTE_VER=75"
|
|
else
|
|
arch_flag=""
|
|
fi
|
|
|
|
tests/ci_build/ci_build.sh jvm_gpu_build nvidia-docker \
|
|
--build-arg CUDA_VERSION_ARG=${CUDA_VERSION} \
|
|
--build-arg NCCL_VERSION_ARG=${NCCL_VERSION} \
|
|
tests/ci_build/build_jvm_packages.sh \
|
|
${SPARK_VERSION} -Duse.cuda=ON ${arch_flag}
|