提交 48982e9d 编写于 作者: T tangwei12

fix lookuptable in reduce strategy

上级 0e722c5e
...@@ -71,7 +71,7 @@ bool IsDistTrainOp(ir::Node *node, const std::vector<std::string> &send_vars, ...@@ -71,7 +71,7 @@ bool IsDistTrainOp(ir::Node *node, const std::vector<std::string> &send_vars,
} }
if (!(var.find(".block") == std::string::npos && if (!(var.find(".block") == std::string::npos &&
var.find(".pserver") != std::string::npos) && var.find(".pserver") == std::string::npos) &&
std::find(rpc_vars.begin(), rpc_vars.end(), var) != rpc_vars.end()) { std::find(rpc_vars.begin(), rpc_vars.end(), var) != rpc_vars.end()) {
return true; return true;
} }
......
...@@ -1133,7 +1133,8 @@ to transpile() call.") ...@@ -1133,7 +1133,8 @@ to transpile() call.")
inputs={ inputs={
'Ids': [program.global_block().vars[table_grad_name]] 'Ids': [program.global_block().vars[table_grad_name]]
}, },
outputs={"Out": self.trainer_side_table_grad_list}) outputs={"Out": self.trainer_side_table_grad_list},
attrs={RPC_OP_ROLE_ATTR_NAME: DIST_OP_ROLE_ATTR_VALUE})
program.global_block()._insert_op( program.global_block()._insert_op(
index=op_index + 2, index=op_index + 2,
type="send", type="send",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册