[backport] Update rapids (#10435) (#10443)

--------- Co-authored-by: hcho3 <[email protected]> Co-authored-by: Philip Hyunsu Cho <[email protected]>
dmlc · Jun 18, 2024 · ec2f56a · ec2f56a
1 parent 13f96c4
commit ec2f56a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/buildkite/conftest.sh b/tests/buildkite/conftest.sh
@@ -24,7 +24,7 @@ set -x
 
 CUDA_VERSION=11.8.0
 NCCL_VERSION=2.16.5-1
-RAPIDS_VERSION=24.04
+RAPIDS_VERSION=24.06
 DEV_RAPIDS_VERSION=24.06
 SPARK_VERSION=3.5.1
 JDK_VERSION=8

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);
   }