@ -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",