提交 620999c9 编写于 作者: T tangwei12

save checkpoint bug fix

上级 97648442
...@@ -39,8 +39,8 @@ class CheckpointNotifyOp : public framework::OperatorBase { ...@@ -39,8 +39,8 @@ class CheckpointNotifyOp : public framework::OperatorBase {
std::string dir = Attr<std::string>("dir"); std::string dir = Attr<std::string>("dir");
std::string lookup_table_name = Attr<std::string>("lookup_table"); std::string lookup_table_name = Attr<std::string>("lookup_table");
detail::RPCClient* rpc_client = distributed::RPCClient* rpc_client =
detail::RPCClient::GetInstance<RPCCLIENT_T>(); distributed::RPCClient::GetInstance<RPCCLIENT_T>();
for (size_t i = 0; i < epmap.size(); i++) { for (size_t i = 0; i < epmap.size(); i++) {
VLOG(3) << "checkpoint notify sending " << dir << " to " << epmap[i]; VLOG(3) << "checkpoint notify sending " << dir << " to " << epmap[i];
auto serial_looku_table = auto serial_looku_table =
......
...@@ -268,7 +268,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope, ...@@ -268,7 +268,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
request_get_handler_.get()); request_get_handler_.get());
rpc_service_->RegisterRPC(distributed::kRequestPrefetch, rpc_service_->RegisterRPC(distributed::kRequestPrefetch,
request_prefetch_handler_.get()); request_prefetch_handler_.get());
rpc_service_->RegisterRPC(detail::kRequestCheckpoint, rpc_service_->RegisterRPC(distributed::kRequestCheckpoint,
request_checkpoint_handler_.get()); request_checkpoint_handler_.get());
auto *optimize_block = Attr<framework::BlockDesc *>(kOptimizeBlock); auto *optimize_block = Attr<framework::BlockDesc *>(kOptimizeBlock);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册