From 36d17d11a41bf08a1b93491183ae249d48e7ffe7 Mon Sep 17 00:00:00 2001
From: tangwei12 <tangwei12@baidu.com>
Date: Mon, 18 Jun 2018 11:13:32 +0800
Subject: [PATCH] add RequestCheckpointNotify in grpc

---
 paddle/fluid/operators/detail/grpc_server.cc | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc
index 75f0a1f78..21bd23226 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<sendrecv::VoidMessage> responder_;
 }
 
 void AsyncGRPCServer::WaitServerReady() {
-- 
GitLab