Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Set max message size in insecure gRPC (#8203)

Browse Source
This commit is contained in:
Rong Ou 2022-08-26 01:33:51 -07:00 committed by GitHub
parent 651f0a8889
commit d6e2013c5f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
plugin/federated/federated_client.h
View File

@ -35,8 +35,12 @@ class FederatedClient {
/** @brief Insecure client for connecting to localhost only. */ /** @brief Insecure client for connecting to localhost only. */
FederatedClient(std::string const &server_address, int rank) FederatedClient(std::string const &server_address, int rank)
: stub_{Federated::NewStub( : stub_{[&] {
grpc::CreateChannel(server_address, grpc::InsecureChannelCredentials()))}, grpc::ChannelArguments args;
args.SetMaxReceiveMessageSize(std::numeric_limits<int>::max());
return Federated::NewStub(
grpc::CreateCustomChannel(server_address, grpc::InsecureChannelCredentials(), args));
}()},
rank_{rank} {} rank_{rank} {}
std::string Allgather(std::string const &send_buffer) { std::string Allgather(std::string const &send_buffer) {