Browse Source

!884 fix l2 buffer error

From: @jiming6
Reviewed-by: @wqtshg,@xchu42
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
f6a206e491
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/build/task_generator.cc

+ 1
- 1
ge/graph/build/task_generator.cc View File

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


Loading…
Cancel
Save