|
|
@ -137,6 +137,8 @@ void StartServerNet(bool is_sparse) {
|
|
|
|
attrs.insert({"GradList", std::vector<std::string>({"x1"})});
|
|
|
|
attrs.insert({"GradList", std::vector<std::string>({"x1"})});
|
|
|
|
attrs.insert({"OptimizeBlock", optimize_block});
|
|
|
|
attrs.insert({"OptimizeBlock", optimize_block});
|
|
|
|
attrs.insert({"PrefetchBlock", prefetch_block});
|
|
|
|
attrs.insert({"PrefetchBlock", prefetch_block});
|
|
|
|
|
|
|
|
attrs.insert({"grad_map", {}});
|
|
|
|
|
|
|
|
attrs.insert({"sync_mode", true});
|
|
|
|
listen_and_serv_op =
|
|
|
|
listen_and_serv_op =
|
|
|
|
f::OpRegistry::CreateOp("listen_and_serv", {{"X", {"x1"}}}, {}, attrs);
|
|
|
|
f::OpRegistry::CreateOp("listen_and_serv", {{"X", {"x1"}}}, {}, attrs);
|
|
|
|
listen_and_serv_op->Run(scope, place);
|
|
|
|
listen_and_serv_op->Run(scope, place);
|
|
|
|