提交 15532c74 编写于 作者: T tangwei12

add RequestCheckpointNotify in grpc

上级 54013a93
...@@ -247,9 +247,6 @@ void AsyncGRPCServer::StartServer() { ...@@ -247,9 +247,6 @@ void AsyncGRPCServer::StartServer() {
std::bind(&AsyncGRPCServer::TryToRegisterNewOne, this, std::bind(&AsyncGRPCServer::TryToRegisterNewOne, this,
std::placeholders::_1, std::placeholders::_2); std::placeholders::_1, std::placeholders::_2);
LOG(INFO) << "Server StartServer on "
<< "TryToRegisterNewOne bind finished";
for (auto& t : rpc_call_map_) { for (auto& t : rpc_call_map_) {
auto& rpc_name = t.first; auto& rpc_name = t.first;
auto& cq = rpc_cq_[rpc_name]; auto& cq = rpc_cq_[rpc_name];
...@@ -258,7 +255,7 @@ void AsyncGRPCServer::StartServer() { ...@@ -258,7 +255,7 @@ void AsyncGRPCServer::StartServer() {
reqs.reserve(kRequestBufSize); reqs.reserve(kRequestBufSize);
LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name << "I: " << i; LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name << " I: " << i;
for (int i = 0; i < kRequestBufSize; i++) { for (int i = 0; i < kRequestBufSize; i++) {
TryToRegisterNewOne(rpc_name, i); TryToRegisterNewOne(rpc_name, i);
...@@ -313,8 +310,11 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name, ...@@ -313,8 +310,11 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name,
return; return;
} }
VLOG(4) << "register send rpc_name:" << rpc_name LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name
<< ", handler:" << rpc_call_map_[kRequestSend]; << " REQ ID: " << req_id;
// VLOG(4) << "register send rpc_name:" << rpc_name
// << ", handler:" << rpc_call_map_[kRequestSend];
auto& reqs = rpc_reqs_[rpc_name]; auto& reqs = rpc_reqs_[rpc_name];
auto& handler = rpc_call_map_[rpc_name]; auto& handler = rpc_call_map_[rpc_name];
...@@ -328,6 +328,7 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name, ...@@ -328,6 +328,7 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name,
} else if (rpc_name == kRequestPrefetch) { } else if (rpc_name == kRequestPrefetch) {
b = new RequestPrefetch(&service_, cq.get(), handler, req_id); b = new RequestPrefetch(&service_, cq.get(), handler, req_id);
} else if (rpc_name == kRequestCheckpoint) { } else if (rpc_name == kRequestCheckpoint) {
LOG(INFO) << "TryToRegisterNewOne on RPC kRequestCheckpoint";
b = new RequestCheckpointNotify(&service_, cq.get(), handler, req_id); b = new RequestCheckpointNotify(&service_, cq.get(), handler, req_id);
} else { } else {
PADDLE_ENFORCE(false, "not supported rpc"); PADDLE_ENFORCE(false, "not supported rpc");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册