[coll] Add federated coll. (#9738)
- Define a new data type, the proto file is copied for now. - Merge client and communicator into `FederatedColl`. - Define CUDA variant. - Migrate tests for CPU, add tests for CUDA.
This commit is contained in:
@@ -163,6 +163,12 @@ inline void AssertGPUSupport() {
|
||||
#endif // XGBOOST_USE_CUDA
|
||||
}
|
||||
|
||||
inline void AssertNCCLSupport() {
|
||||
#if !defined(XGBOOST_USE_NCCL)
|
||||
LOG(FATAL) << "XGBoost version not compiled with NCCL support.";
|
||||
#endif // !defined(XGBOOST_USE_NCCL)
|
||||
}
|
||||
|
||||
inline void AssertOneAPISupport() {
|
||||
#ifndef XGBOOST_USE_ONEAPI
|
||||
LOG(FATAL) << "XGBoost version not compiled with OneAPI support.";
|
||||
|
||||
Reference in New Issue
Block a user