Browse Source

modify control passes

pull/401/head
chenyemeng 5 years ago
parent
commit
d1a1a86866
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/attach_stream_label_pass.cc

+ 1
- 1
ge/graph/passes/attach_stream_label_pass.cc View File

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


Loading…
Cancel
Save