diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc index 9a8c4196720d0054eeaa25d7e54a38b9691b4c2e..b7f4032c5af69b83f9b5d7470688dd598965a1a1 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_); }