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

Increase gRPC max receive message size for federated learning (#7958)

Browse Source
This commit is contained in:
Rong Ou 2022-05-31 22:21:54 -07:00 committed by GitHub
parent 545fd4548e
commit d3429f2ff6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
plugin/federated/federated_client.h
View File

@ -8,6 +8,7 @@
#include <cstdio> #include <cstdio>
#include <cstdlib> #include <cstdlib>
#include <limits>
#include <string> #include <string>
namespace xgboost { namespace xgboost {
@ -25,8 +26,10 @@ class FederatedClient {
options.pem_root_certs = server_cert; options.pem_root_certs = server_cert;
options.pem_private_key = client_key; options.pem_private_key = client_key;
options.pem_cert_chain = client_cert; options.pem_cert_chain = client_cert;
grpc::ChannelArguments args;
args.SetMaxReceiveMessageSize(std::numeric_limits<int>::max());
return Federated::NewStub( return Federated::NewStub(
grpc::CreateChannel(server_address, grpc::SslCredentials(options))); grpc::CreateCustomChannel(server_address, grpc::SslCredentials(options), args));
}()}, }()},
rank_{rank} {} rank_{rank} {}

1
plugin/federated/federated_server.cc
View File

@ -221,6 +221,7 @@ void RunServer(int port, int world_size, char const* server_key_file, char const
key.private_key = ReadFile(server_key_file); key.private_key = ReadFile(server_key_file);
key.cert_chain = ReadFile(server_cert_file); key.cert_chain = ReadFile(server_cert_file);
options.pem_key_cert_pairs.push_back(key); options.pem_key_cert_pairs.push_back(key);
builder.SetMaxReceiveMessageSize(std::numeric_limits<int>::max());
builder.AddListeningPort(server_address, grpc::SslServerCredentials(options)); builder.AddListeningPort(server_address, grpc::SslServerCredentials(options));
builder.RegisterService(&service); builder.RegisterService(&service);
std::unique_ptr<grpc::Server> server(builder.BuildAndStart()); std::unique_ptr<grpc::Server> server(builder.BuildAndStart());