From 8af8da4fe4fa59e35b0d033286df574df7117067 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Mon, 18 Jun 2018 22:14:39 +0800 Subject: [PATCH] move checkpoint message to variable message --- paddle/fluid/operators/detail/grpc_server.cc | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc index 9a8c4196720..b7f4032c5af 100644 --- a/paddle/fluid/operators/detail/grpc_server.cc +++ b/paddle/fluid/operators/detail/grpc_server.cc @@ -201,15 +201,19 @@ class RequestCheckpointNotify final : public RequestBase { virtual ~RequestCheckpointNotify() {} - std::string GetReqName() override { return "checkpoint_notify"; } + std::string GetReqName() override { return request_->Varname(); } void Process() override { auto scope = request_->GetMutableLocalScope(); - std::string nullptr_str = nullptr; + + std::string checkpoint_notify = request_->Varname(); + std::string checkpoint_dir = request_->Varname(); + framework::Variable* invar = nullptr; framework::Variable* outvar = nullptr; - request_handler_->Handle(nullptr_str, scope, invar, &outvar, nullptr_str); + request_handler_->Handle(checkpoint_notify, scope, invar, &outvar, + checkpoint_dir); Finish(reply_, &responder_); } -- GitLab