diff --git a/ge/graph/build/task_generator.cc b/ge/graph/build/task_generator.cc index 21e82d11..bb72fa8a 100755 --- a/ge/graph/build/task_generator.cc +++ b/ge/graph/build/task_generator.cc @@ -572,7 +572,7 @@ Status TaskGenerator::MarkFirstAndLastOps(const vector &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()); } else { continuous_op_lists.back().emplace_back(op_desc);