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

save checkpoint bug fix

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