fix l2 buffer error

4 years ago · aac7897a44
parent d5c6e8146b
commit aac7897a44
1 changed files with 1 additions and 1 deletions
--- a/ge/graph/build/task_generator.cc
+++ b/ge/graph/build/task_generator.cc
@ -567,7 +567,7 @@ Status TaskGenerator::MarkFirstAndLastOps(const vector<OpDescPtr> &ops, bool is_
      continue;
    }
    string op_type = op_desc->GetType();
-    if (!op_desc->GetSubgraphInstanceNames().empty() || separator_types.count(op_type) != 0) {
+    if ((!is_single_stream && !op_desc->GetSubgraphInstanceNames().empty()) || separator_types.count(op_type) != 0) {
      continuous_op_lists.emplace_back(vector<OpDescPtr>());
    } else {
      continuous_op_lists.back().emplace_back(op_desc);