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

Update rapids (#10435)

Browse Source 
* [CI] Update RAPIDS to latest stable

* RMM.

---------

Co-authored-by: hcho3 <2532981+hcho3@users.noreply.github.com>
Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan 2024-06-18 05:01:57 +08:00 committed by GitHub
parent b4cc350ec5
commit b9e5229ff2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/buildkite/conftest.sh
View File

@ -24,8 +24,8 @@ set -x
CUDA_VERSION=11.8.0
NCCL_VERSION=2.16.5-1
RAPIDS_VERSION=24.04
DEV_RAPIDS_VERSION=24.06
RAPIDS_VERSION=24.06
DEV_RAPIDS_VERSION=24.08
SPARK_VERSION=3.5.1
JDK_VERSION=8
R_VERSION=4.3.2

2
tests/cpp/helpers.cc
View File

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