提交 1cb0ab36 编写于 作者: T tangwei12

bug fix

上级 bb17604b
...@@ -234,7 +234,7 @@ void GRPCClient::AsyncCheckpointNotify(const std::string& ep, ...@@ -234,7 +234,7 @@ void GRPCClient::AsyncCheckpointNotify(const std::string& ep,
int64_t time_out) { int64_t time_out) {
const auto ch = GetChannel(ep); const auto ch = GetChannel(ep);
CheckpointNotifyProcessor* s = new CheckpointNotifyProcessor(ch); CheckpointNotifyProcessor* s = new CheckpointNotifyProcessor(ch);
s.prepare(time_out); s->Prepare(time_out);
sendrecv::CheckpointMessage req; sendrecv::CheckpointMessage req;
req.set_notify_type(CHECKPOINT_SAVE_MESSAGE); req.set_notify_type(CHECKPOINT_SAVE_MESSAGE);
......
...@@ -177,7 +177,7 @@ class CheckpointNotifyProcessor : public BaseProcessor { ...@@ -177,7 +177,7 @@ class CheckpointNotifyProcessor : public BaseProcessor {
virtual void Process() {} virtual void Process() {}
sendrecv::VoidMessage reply_; sendrecv::VoidMessage reply_;
std::unique_ptr<sendrecv::SendRecvService::Stub> stub_; std::unique_ptr<sendrecv::SendRecvService::Stub> stub_;
} };
class GRPCClient : public RPCClient { class GRPCClient : public RPCClient {
public: public:
......
...@@ -124,7 +124,7 @@ bool RequestCheckpointHandler::Handle(const std::string& varname, ...@@ -124,7 +124,7 @@ bool RequestCheckpointHandler::Handle(const std::string& varname,
framework::Variable* invar, framework::Variable* invar,
framework::Variable** outvar, framework::Variable** outvar,
const std::string& out_var_name) { const std::string& out_var_name) {
executor_->RunPreparedContext(checkpoint_prepared_ctx_); executor_->RunPreparedContext(checkpoint_prepared_ctx_, scope);
return true; return true;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册