|
|
@@ -58,7 +58,7 @@ void AttachStreamLabelPass::FindNodes(const ComputeGraphPtr &graph) { |
|
|
const std::string &type = op_desc->GetType(); |
|
|
const std::string &type = op_desc->GetType(); |
|
|
if ((type == STREAMSWITCH) && op_desc->HasAttr(ATTR_NAME_SWITCH_TRUE_BRANCH_FLAG)) { |
|
|
if ((type == STREAMSWITCH) && op_desc->HasAttr(ATTR_NAME_SWITCH_TRUE_BRANCH_FLAG)) { |
|
|
stream_switch_nodes_.emplace_back(node); |
|
|
stream_switch_nodes_.emplace_back(node); |
|
|
} else if ((type == STREAMMERGE)&& !op_desc->HasAttr(ATTR_NAME_NEXT_ITERATION)) { |
|
|
|
|
|
|
|
|
} else if ((type == STREAMMERGE) && !op_desc->HasAttr(ATTR_NAME_NEXT_ITERATION)) { |
|
|
need_label_nodes_.emplace_back(node); |
|
|
need_label_nodes_.emplace_back(node); |
|
|
} else if ((type == ENTER) || (type == REFENTER)) { |
|
|
} else if ((type == ENTER) || (type == REFENTER)) { |
|
|
enter_nodes_.emplace_back(node); |
|
|
enter_nodes_.emplace_back(node); |
|
|
|