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

bug fix

上级 bb17604b
......@@ -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);
......
......@@ -177,7 +177,7 @@ class CheckpointNotifyProcessor : public BaseProcessor {
virtual void Process() {}
sendrecv::VoidMessage reply_;
std::unique_ptr<sendrecv::SendRecvService::Stub> stub_;
}
};
class GRPCClient : public RPCClient {
public:
......
......@@ -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;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册