From 1cb0ab36f08b4746fce77a4ff3444507226d3e52 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 15 Jun 2018 14:34:13 +0800 Subject: [PATCH] bug fix --- paddle/fluid/operators/detail/grpc_client.cc | 2 +- paddle/fluid/operators/detail/grpc_client.h | 2 +- paddle/fluid/operators/detail/request_handler_impl.cc | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/detail/grpc_client.cc b/paddle/fluid/operators/detail/grpc_client.cc index 1dff3bfa3cb..9a25ec8fdb4 100644 --- a/paddle/fluid/operators/detail/grpc_client.cc +++ b/paddle/fluid/operators/detail/grpc_client.cc @@ -234,7 +234,7 @@ void GRPCClient::AsyncCheckpointNotify(const std::string& ep, int64_t time_out) { const auto ch = GetChannel(ep); CheckpointNotifyProcessor* s = new CheckpointNotifyProcessor(ch); - s.prepare(time_out); + s->Prepare(time_out); sendrecv::CheckpointMessage req; req.set_notify_type(CHECKPOINT_SAVE_MESSAGE); diff --git a/paddle/fluid/operators/detail/grpc_client.h b/paddle/fluid/operators/detail/grpc_client.h index bc3deff47ce..0c54ec0efef 100644 --- a/paddle/fluid/operators/detail/grpc_client.h +++ b/paddle/fluid/operators/detail/grpc_client.h @@ -177,7 +177,7 @@ class CheckpointNotifyProcessor : public BaseProcessor { virtual void Process() {} sendrecv::VoidMessage reply_; std::unique_ptr stub_; -} +}; class GRPCClient : public RPCClient { public: diff --git a/paddle/fluid/operators/detail/request_handler_impl.cc b/paddle/fluid/operators/detail/request_handler_impl.cc index de6ce72d4dc..ba7d027637a 100644 --- a/paddle/fluid/operators/detail/request_handler_impl.cc +++ b/paddle/fluid/operators/detail/request_handler_impl.cc @@ -124,7 +124,7 @@ bool RequestCheckpointHandler::Handle(const std::string& varname, framework::Variable* invar, framework::Variable** outvar, const std::string& out_var_name) { - executor_->RunPreparedContext(checkpoint_prepared_ctx_); + executor_->RunPreparedContext(checkpoint_prepared_ctx_, scope); return true; } -- GitLab