enable federated

2023-10-31 16:31:56 -07:00
parent 123af45327
commit 129bb76941
6 changed files with 15 additions and 3 deletions
--- a/src/collective/comm.cc
+++ b/src/collective/comm.cc
@@ -49,7 +49,7 @@ Result ConnectTrackerImpl(proto::PeerInfo info, std::chrono::seconds timeout, st
                            this->Rank(), this->World());
 }

-#if !defined(XGBOOST_USE_NCCL)
+#if !defined(XGBOOST_USE_NCCL) && !defined(XGBOOST_USE_RCCL)
 Comm* Comm::MakeCUDAVar(Context const*, std::shared_ptr<Coll>) const {
  common::AssertGPUSupport();
  common::AssertNCCLSupport();