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

[CI] Use RAPIDS 22.10 (#8298)

Browse Source 
* [CI] Use RAPIDS 22.10

* Store CUDA and RAPIDS versions in one place

* Fix

* Add missing #include

* Update gputreeshap submodule

* Fix

* Remove outdated distributed tests
This commit is contained in:
Philip Hyunsu Cho
2022-10-04 00:18:07 -07:00
committed by GitHub
parent 37886a5dff
commit ca0547bb65
23 changed files with 25 additions and 258 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/buildkite/build-cuda-with-rmm.sh
View File

@@ -2,8 +2,6 @@
set -euo pipefail
CUDA_VERSION=11.0.3
source tests/buildkite/conftest.sh
echo "--- Build with CUDA ${CUDA_VERSION}, RMM enabled"
@@ -16,7 +14,8 @@ else
fi
command_wrapper="tests/ci_build/ci_build.sh rmm docker --build-arg "`
`"CUDA_VERSION_ARG=$CUDA_VERSION"
`"CUDA_VERSION_ARG=$CUDA_VERSION --build-arg "`
`"RAPIDS_VERSION_ARG=$RAPIDS_VERSION"
echo "--- Build libxgboost from the source"
$command_wrapper tests/ci_build/build_via_cmake.sh --conda-env=gpu_test -DUSE_CUDA=ON \

5
tests/buildkite/build-cuda.sh
View File

@@ -2,13 +2,12 @@
set -euo pipefail
CUDA_VERSION=11.0.3
WHEEL_TAG=manylinux2014_x86_64
echo "--- Build with CUDA ${CUDA_VERSION}"
source tests/buildkite/conftest.sh
echo "--- Build with CUDA ${CUDA_VERSION}"
if [[ ($is_pull_request == 1) || ($is_release_branch == 0) ]]
then
arch_flag="-DGPU_COMPUTE_VER=75"

2
tests/buildkite/build-gpu-rpkg.sh
View File

@@ -2,8 +2,6 @@
set -euo pipefail
CUDA_VERSION=11.0.3
source tests/buildkite/conftest.sh
echo "--- Build XGBoost R package with CUDA"

3
tests/buildkite/build-jvm-packages-gpu.sh
View File

@@ -2,9 +2,6 @@
set -euo pipefail
SPARK_VERSION=3.0.1
CUDA_VERSION=11.0.3
source tests/buildkite/conftest.sh
echo "--- Build XGBoost JVM packages with CUDA"

2
tests/buildkite/build-jvm-packages.sh
View File

@@ -2,8 +2,6 @@
set -euo pipefail
SPARK_VERSION=3.0.1
source tests/buildkite/conftest.sh
echo "--- Build XGBoost JVM packages"

5
tests/buildkite/conftest.sh
View File

@@ -3,6 +3,11 @@
set -euo pipefail
set -x
CUDA_VERSION=11.0.3
RAPIDS_VERSION=22.10
SPARK_VERSION=3.0.1
JDK_VERSION=8
if [[ -z ${BUILDKITE:-} ]]
then
echo "$0 is not meant to run locally; it should run inside BuildKite."

3
tests/buildkite/deploy-jvm-packages.sh
View File

@@ -2,9 +2,6 @@
set -euo pipefail
SPARK_VERSION=3.0.1
CUDA_VERSION=11.0.3
source tests/buildkite/conftest.sh
if [[ ($is_pull_request == 0) && ($is_release_branch == 1) ]]

2
tests/buildkite/run-clang-tidy.sh
View File

@@ -2,8 +2,6 @@
set -euo pipefail
CUDA_VERSION=11.0.3
echo "--- Run clang-tidy"
source tests/buildkite/conftest.sh

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

@@ -2,20 +2,21 @@
set -euo pipefail
CUDA_VERSION=11.0.3
source tests/buildkite/conftest.sh
echo "--- Run Google Tests with CUDA, using 4 GPUs"
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/testxgboost
--build-arg CUDA_VERSION_ARG=$CUDA_VERSION \
--build-arg RAPIDS_VERSION_ARG=$RAPIDS_VERSION \
build/testxgboost
echo "--- Run Google Tests with CUDA, using 4 GPUs, 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 bash -c \
--build-arg CUDA_VERSION_ARG=$CUDA_VERSION \
--build-arg RAPIDS_VERSION_ARG=$RAPIDS_VERSION bash -c \
"source activate gpu_test && build/testxgboost --use-rmm-pool"

3
tests/buildkite/test-integration-jvm-packages.sh
View File

@@ -2,9 +2,6 @@
set -euo pipefail
JDK_VERSION=8
SPARK_VERSION=3.0.1
source tests/buildkite/conftest.sh
echo "--- Test XGBoost4J on a machine with JDK ${JDK_VERSION}, Spark ${SPARK_VERSION}"

5
tests/buildkite/test-python-gpu.sh
View File

@@ -2,8 +2,6 @@
set -euo pipefail
CUDA_VERSION=11.0.3
if [ "$#" -lt 1 ]
then
suite=''
@@ -25,7 +23,8 @@ chmod +x build/testxgboost
export CI_DOCKER_EXTRA_PARAMS_INIT='--shm-size=4g'
command_wrapper="tests/ci_build/ci_build.sh gpu nvidia-docker --build-arg "`
`"CUDA_VERSION_ARG=$CUDA_VERSION"
`"CUDA_VERSION_ARG=$CUDA_VERSION --build-arg "`
`"RAPIDS_VERSION_ARG=$RAPIDS_VERSION"
# Run specified test suite
case "$suite" in