Don't split input data in federated mode (#8279)

Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Rong Ou
2022-10-05 19:19:28 -07:00
committed by GitHub
parent 66fd9f5207
commit 8d4038da57
6 changed files with 23 additions and 8 deletions

View File

@@ -53,6 +53,8 @@ class RabitCommunicator : public Communicator {
bool IsDistributed() const override { return rabit::IsDistributed(); }
bool IsFederated() const override { return false; }
void AllReduce(void *send_receive_buffer, std::size_t count, DataType data_type,
Operation op) override {
switch (data_type) {