diff --git a/paddle/fluid/operators/detail/grpc_client.cc b/paddle/fluid/operators/detail/grpc_client.cc index 1dff3bfa3cbb9709e40e526523b5467b650268ff..9a25ec8fdb4599ea9128bed151a525c13d4127a0 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 bc3deff47cec1499056c8b13de5d5e7db9ef2175..0c54ec0efefcc889440a257d59f6ab123ab8b06a 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 de6ce72d4dcd30e53dfa770ce615335f3a2ecfc3..ba7d027637a24d0af91bbd3dca1a75f0e5f03f6a 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; }