提交 8d46d1dd 编写于 作者: T tangwei12

bug fix

上级 f224948f
......@@ -39,9 +39,10 @@ class CheckpointNotifyOp : public framework::OperatorBase {
detail::RPCClient* rpc_client =
detail::RPCClient::GetInstance<RPCCLIENT_T>();
VLOG(3) << "sending " << ins[i] << " to " << epmap[i] << " to get "
<< outs[i] << " back";
rpc_client->AsyncCheckpointNotify(epmap[i], dir);
for (size_t i = 0; i < epmap.size(); i++) {
VLOG(3) << "sending to " << epmap[i] << " to checkpoint notify ... ";
rpc_client->AsyncCheckpointNotify(epmap[i], dir);
}
rpc_client->Wait();
}
};
......
......@@ -300,10 +300,10 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
}
int checkpoint_point_block_id = Attr<int>(kCheckpointBlockId);
auto *ctx = new ExecutorPrepareContext(*program, checkpoint_point_block_id);
auto ctx = executor.Prepare(*program, checkpoint_point_block_id);
std::shared_ptr<framework::ExecutorPrepareContext> ckpt_pre_context =
std::shared_ptr<ExecutorPrepareContext>(ctx);
std::move(ctx);
auto f =
std::bind(FillRequestCtx, std::placeholders::_1, &recv_scope, &dev_ctx,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册