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

---
 paddle/fluid/operators/detail/grpc_server.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc
index 21bd232260f..4079df7ab10 100644
--- a/paddle/fluid/operators/detail/grpc_server.cc
+++ b/paddle/fluid/operators/detail/grpc_server.cc
@@ -216,10 +216,10 @@ class RequestCheckpointNotify final : public RequestBase {
   }
 
  protected:
-  sendrecv::CheckpointMessage request_;
+  std::shared_ptr<VariableResponse> request_;
   sendrecv::VoidMessage reply_;
   ServerAsyncResponseWriter<sendrecv::VoidMessage> responder_;
-}
+};
 
 void AsyncGRPCServer::WaitServerReady() {
   VLOG(3) << "AsyncGRPCServer is wait server ready";
-- 
GitLab