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

trivialfis · hcho3 · web-flow · commit ec2f56abe624 · 2024-06-18T13:10:21.000+08:00
---------

Co-authored-by: hcho3 &lt;2532981+hcho3@users.noreply.github.com&gt;
Co-authored-by: Philip Hyunsu Cho &lt;chohyu01@cs.washington.edu&gt;
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);
   }

Original file line number	Diff line number	Diff line change
`@@ -728,7 +728,7 @@ class RMMAllocator {`
`728`	`728`	`for (int i = 0; i < n_gpu; ++i) {`
`729`	`729`	`CHECK_EQ(cudaSetDevice(i), cudaSuccess);`
`730`	`730`	`cuda_mr.push_back(std::make_unique<CUDAMemoryResource>());`
`731`		`- pool_mr.push_back(std::make_unique<PoolMemoryResource>(cuda_mr[i].get()));`
	`731`	`+ pool_mr.push_back(std::make_unique<PoolMemoryResource>(cuda_mr[i].get(), 0ul));`
`732`	`732`	`}`
`733`	`733`	`CHECK_EQ(cudaSetDevice(current_device), cudaSuccess);`
`734`	`734`	`}`