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

Throw error when not compiled with NCCL. (#5170)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-01-07 11:15:19 +08:00 committed by GitHub
parent 2d95b9a4b6
commit 2b9a62a806
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/device_helpers.cu
View File

@ -71,6 +71,10 @@ void AllReducer::Init(int _device_ordinal) {
dh::safe_nccl(ncclCommInitRank(&comm, rabit::GetWorldSize(), id, rank)); dh::safe_nccl(ncclCommInitRank(&comm, rabit::GetWorldSize(), id, rank));
safe_cuda(cudaStreamCreate(&stream)); safe_cuda(cudaStreamCreate(&stream));
initialised_ = true; initialised_ = true;
#else
if (rabit::IsDistributed()) {
LOG(FATAL) << "XGBoost is not compiled with NCCL.";
}
#endif // XGBOOST_USE_NCCL #endif // XGBOOST_USE_NCCL
} }