From 15532c74b19f87b7ea3df16969d76299b366801c Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Mon, 18 Jun 2018 13:04:22 +0800 Subject: [PATCH] add RequestCheckpointNotify in grpc --- paddle/fluid/operators/detail/grpc_server.cc | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc index 61afffeebad..5adb5162923 100644 --- a/paddle/fluid/operators/detail/grpc_server.cc +++ b/paddle/fluid/operators/detail/grpc_server.cc @@ -247,9 +247,6 @@ void AsyncGRPCServer::StartServer() { std::bind(&AsyncGRPCServer::TryToRegisterNewOne, this, std::placeholders::_1, std::placeholders::_2); - LOG(INFO) << "Server StartServer on " - << "TryToRegisterNewOne bind finished"; - for (auto& t : rpc_call_map_) { auto& rpc_name = t.first; auto& cq = rpc_cq_[rpc_name]; @@ -258,7 +255,7 @@ void AsyncGRPCServer::StartServer() { 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++) { TryToRegisterNewOne(rpc_name, i); @@ -313,8 +310,11 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name, return; } - VLOG(4) << "register send rpc_name:" << rpc_name - << ", handler:" << rpc_call_map_[kRequestSend]; + LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name + << " REQ ID: " << req_id; + + // VLOG(4) << "register send rpc_name:" << rpc_name + // << ", handler:" << rpc_call_map_[kRequestSend]; auto& reqs = rpc_reqs_[rpc_name]; auto& handler = rpc_call_map_[rpc_name]; @@ -328,6 +328,7 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name, } else if (rpc_name == kRequestPrefetch) { b = new RequestPrefetch(&service_, cq.get(), handler, req_id); } else if (rpc_name == kRequestCheckpoint) { + LOG(INFO) << "TryToRegisterNewOne on RPC kRequestCheckpoint"; b = new RequestCheckpointNotify(&service_, cq.get(), handler, req_id); } else { PADDLE_ENFORCE(false, "not supported rpc"); -- GitLab