fix l2 buffer error

4 years ago · d666ae0310
parent 169d1efff3
commit d666ae0310
1 changed files with 1 additions and 1 deletions
--- a/ge/graph/build/task_generator.cc
+++ b/ge/graph/build/task_generator.cc
@ -572,7 +572,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);