From f63880036b89976ef22553c2c748c37ec8482800 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Tue, 18 Jun 2024 05:01:57 +0800 Subject: [PATCH] [backport] Update rapids (#10435) * [CI] Update RAPIDS to latest stable * RMM. --------- Co-authored-by: hcho3 <2532981+hcho3@users.noreply.github.com> Co-authored-by: Philip Hyunsu Cho --- tests/buildkite/conftest.sh | 4 ++-- tests/cpp/helpers.cc | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/buildkite/conftest.sh b/tests/buildkite/conftest.sh index 44043910bc96..4cb5164a70fb 100755 --- a/tests/buildkite/conftest.sh +++ b/tests/buildkite/conftest.sh @@ -24,8 +24,8 @@ set -x CUDA_VERSION=11.8.0 NCCL_VERSION=2.16.5-1 -RAPIDS_VERSION=24.04 -SPARK_VERSION=3.4.0 +RAPIDS_VERSION=24.06 +SPARK_VERSION=3.5.1 JDK_VERSION=8 R_VERSION=4.3.2 diff --git a/tests/cpp/helpers.cc b/tests/cpp/helpers.cc index a4761063688d..fc5ec303448d 100644 --- a/tests/cpp/helpers.cc +++ b/tests/cpp/helpers.cc @@ -728,7 +728,7 @@ class RMMAllocator { for (int i = 0; i < n_gpu; ++i) { CHECK_EQ(cudaSetDevice(i), cudaSuccess); cuda_mr.push_back(std::make_unique()); - pool_mr.push_back(std::make_unique(cuda_mr[i].get())); + pool_mr.push_back(std::make_unique(cuda_mr[i].get(), 0ul)); } CHECK_EQ(cudaSetDevice(current_device), cudaSuccess); }