Improve allgather functions (#9649)

2023-10-12 08:31:43 -07:00
parent d1dee4ad99
commit e164d51c43
20 changed files with 346 additions and 122 deletions
--- a/tests/cpp/plugin/helpers.h
+++ b/tests/cpp/plugin/helpers.h
@@ -23,7 +23,7 @@ class ServerForTest {
  std::unique_ptr<grpc::Server> server_;

 public:
-  explicit ServerForTest(std::int32_t world_size) {
+  explicit ServerForTest(std::size_t world_size) {
    server_thread_.reset(new std::thread([this, world_size] {
      grpc::ServerBuilder builder;
      xgboost::federated::FederatedService service{world_size};