Add Allgather to collective communicator (#8765)

* Add Allgather to collective communicator
2023-02-08 19:31:22 -08:00
parent 48cefa012e
commit cbf98cb9c6
14 changed files with 187 additions and 4 deletions
--- a/plugin/federated/federated_server.h
+++ b/plugin/federated/federated_server.h
@@ -14,6 +14,9 @@ class FederatedService final : public Federated::Service {
 public:
  explicit FederatedService(int const world_size) : handler_{world_size} {}

+  grpc::Status Allgather(grpc::ServerContext* context, AllgatherRequest const* request,
+                         AllgatherReply* reply) override;
+
  grpc::Status Allreduce(grpc::ServerContext* context, AllreduceRequest const* request,
                         AllreduceReply* reply) override;