diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc index 75f0a1f78931170dbebb5fc1c41b7627f87f5717..21bd232260f2e080220c41f9a501be71bc5986f2 100644 --- a/paddle/fluid/operators/detail/grpc_server.cc +++ b/paddle/fluid/operators/detail/grpc_server.cc @@ -203,7 +203,7 @@ class RequestCheckpointNotify final : public RequestBase { virtual ~RequestCheckpointNotify() {} - std::string GetReqName() override { return request_->Varname(); } + std::string GetReqName() override { return "checkpoint_notify"; } void Process() override { auto scope = request_->GetMutableLocalScope(); @@ -214,6 +214,11 @@ class RequestCheckpointNotify final : public RequestBase { request_handler_->Handle(nullptr_str, scope, invar, &outvar, nullptr_str); Finish(reply_, &responder_); } + + protected: + sendrecv::CheckpointMessage request_; + sendrecv::VoidMessage reply_; + ServerAsyncResponseWriter responder_; } void AsyncGRPCServer::WaitServerReady() {