!891 fix l2 buffer error

From: @jiming6
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @startzgf168,@ji_chen
pull/891/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 1cd83211d6

@ -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);

Loading…
Cancel
Save