提交 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,
}
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()) {
return true;
}
......
......@@ -1133,7 +1133,8 @@ to transpile() call.")
inputs={
'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(
index=op_index + 2,
type="send",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册