!884 fix l2 buffer error

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

@ -572,7 +572,7 @@ Status TaskGenerator::MarkFirstAndLastOps(const vector<OpDescPtr> &ops, bool is_
continue; continue;
} }
string op_type = op_desc->GetType(); 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>()); continuous_op_lists.emplace_back(vector<OpDescPtr>());
} else { } else {
continuous_op_lists.back().emplace_back(op_desc); continuous_op_lists.back().emplace_back(op_desc);

Loading…
Cancel
Save