From bbb349fbf075eb67536f1c488cbc395f5fb04d46 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Mon, 18 Jun 2018 14:40:15 +0800 Subject: [PATCH] add RequestCheckpointNotify in grpc --- paddle/fluid/operators/detail/grpc_server.cc | 8 ++------ paddle/fluid/operators/detail/grpc_service.h | 2 +- 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc index 5adb516292..9a8c419672 100644 --- a/paddle/fluid/operators/detail/grpc_server.cc +++ b/paddle/fluid/operators/detail/grpc_server.cc @@ -255,9 +255,9 @@ void AsyncGRPCServer::StartServer() { reqs.reserve(kRequestBufSize); - LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name << " I: " << i; - for (int i = 0; i < kRequestBufSize; i++) { + LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name + << " I: " << i; TryToRegisterNewOne(rpc_name, i); } @@ -313,9 +313,6 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name, 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]; auto& cq = rpc_cq_[rpc_name]; @@ -328,7 +325,6 @@ 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"); diff --git a/paddle/fluid/operators/detail/grpc_service.h b/paddle/fluid/operators/detail/grpc_service.h index 69200a01d3..cb745e125a 100644 --- a/paddle/fluid/operators/detail/grpc_service.h +++ b/paddle/fluid/operators/detail/grpc_service.h @@ -83,7 +83,7 @@ enum class GrpcMethod { }; static const int kGrpcNumMethods = - static_cast(GrpcMethod::kPrefetchVariable) + 1; + static_cast(GrpcMethod::kCheckpointNotify) + 1; inline const char* GrpcMethodName(GrpcMethod id) { switch (id) { -- GitLab