Browse Source

fix l2 buffer error

tags/v1.2.0
wjm 3 years ago
parent
commit
d666ae0310
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; 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