[backport] Update rapids (dmlc#10435)

* [CI] Update RAPIDS to latest stable * RMM. --------- Co-authored-by: hcho3 <[email protected]> Co-authored-by: Philip Hyunsu Cho <[email protected]>
trivialfis · Jun 18, 2024 · f638800 · f638800
1 parent 63b49f3
commit f638800
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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<CUDAMemoryResource>());
-      pool_mr.push_back(std::make_unique<PoolMemoryResource>(cuda_mr[i].get()));
+      pool_mr.push_back(std::make_unique<PoolMemoryResource>(cuda_mr[i].get(), 0ul));
     }
     CHECK_EQ(cudaSetDevice(current_device), cudaSuccess);
   }