port
tangwei12 7 years ago
parent 5600b13512
commit 32fa832b4b

@ -42,8 +42,9 @@ class CheckpointNotifyOp : public framework::OperatorBase {
detail::RPCClient* rpc_client =
detail::RPCClient::GetInstance<RPCCLIENT_T>();
for (size_t i = 0; i < epmap.size(); i++) {
VLOG(3) << "sending " << dir <<" to " << epmap[i] << " to checkpoint notify ... ";
auto serial_looku_table = string::Sprintf("%s/%s_%d", dir, lookup_table_name, i);
VLOG(3) << "checkpoint notify sending " << dir << " to " << epmap[i];
auto serial_looku_table =
string::Sprintf("%s/%s_%d", dir, lookup_table_name, i);
rpc_client->AsyncCheckpointNotify(epmap[i], serial_looku_table);
}
rpc_client->Wait();
@ -60,8 +61,8 @@ class CheckpointNotifyOpMaker : public framework::OpProtoAndCheckerMaker {
.SetDefault({"127.0.0.1:6164"});
AddAttr<std::string>(
"dir", "(string, default '') indicate the folder checkpoint will use");
AddAttr<std::string>(
"lookup_table", "(string, default '') the lookup table name");
AddAttr<std::string>("lookup_table",
"(string, default '') the lookup table name");
AddComment(R"DOC(
Prefetch operator

Loading…
Cancel
Save