提交 6abf0769 编写于 作者: T tangwei12

checkpoint_notify_id rename

上级 16ecead8
...@@ -247,7 +247,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope, ...@@ -247,7 +247,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
PADDLE_ENFORCE(!rpc_service_); PADDLE_ENFORCE(!rpc_service_);
std::string endpoint = Attr<std::string>("endpoint"); std::string endpoint = Attr<std::string>("endpoint");
int checkpoint_point_block_id = Attr<int>(kCheckpointBlockId); int checkpoint_notify_id = Attr<int>(kCheckpointBlockId);
LOG(INFO) << "sync_mode:" << sync_mode << ", fan_in:" << fan_in LOG(INFO) << "sync_mode:" << sync_mode << ", fan_in:" << fan_in
<< ", end_point:" << endpoint << ", end_point:" << endpoint
...@@ -271,7 +271,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope, ...@@ -271,7 +271,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
std::shared_ptr<framework::ExecutorPrepareContext> ckpt_pre_context = nullptr; std::shared_ptr<framework::ExecutorPrepareContext> ckpt_pre_context = nullptr;
if (checkpoint_notify_id != -1) { if (checkpoint_notify_id != -1) {
auto ctx = executor.Prepare(*program, checkpoint_point_block_id); auto ctx = executor.Prepare(*program, checkpoint_notify_id);
ckpt_pre_context = std::move(ctx); ckpt_pre_context = std::move(ctx);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册