enable RCCL

2023-10-30 14:05:04 -07:00
parent d7f1235b7d
commit b6b5218245
5 changed files with 11 additions and 5 deletions
--- a/src/collective/coll.cc
+++ b/src/collective/coll.cc
@@ -87,7 +87,7 @@ namespace xgboost::collective {
  }
 }

-#if !defined(XGBOOST_USE_NCCL)
+#if !defined(XGBOOST_USE_NCCL) && !defined(XGBOOST_USE_RCCL)
 Coll* Coll::MakeCUDAVar() {
  LOG(FATAL) << "NCCL is required for device communication.";
  return nullptr;
--- a/src/collective/coll.cu
+++ b/src/collective/coll.cu
@@ -1,7 +1,7 @@
 /**
 * Copyright 2023, XGBoost Contributors
 */
-#if defined(XGBOOST_USE_NCCL)
+#if defined(XGBOOST_USE_NCCL) || defined(XGBOOST_USE_RCCL)
 #include <cstdint>  // for int8_t, int64_t

 #include "../common/cuda_context.cuh"
--- a/src/collective/comm.cu
+++ b/src/collective/comm.cu
@@ -1,7 +1,7 @@
 /**
 * Copyright 2023, XGBoost Contributors
 */
-#if defined(XGBOOST_USE_NCCL)
+#if defined(XGBOOST_USE_NCCL) || defined(XGBOOST_USE_RCCL)
 #include <algorithm>  // for sort
 #include <cstddef>    // for size_t
 #include <cstdint>    // for uint64_t, int8_t
--- a/src/collective/comm.cuh
+++ b/src/collective/comm.cuh
@@ -3,7 +3,7 @@
 */
 #pragma once

-#ifdef XGBOOST_USE_NCCL
+#if defined(XGBOOST_USE_NCCL) || defined(XGBOOST_USE_RCCL)
 #include "nccl.h"
 #endif  // XGBOOST_USE_NCCL
 #include "../common/device_helpers.cuh"