From d666ae031009d8da4c343db9a3f5943a77ec1e35 Mon Sep 17 00:00:00 2001 From: wjm Date: Wed, 6 Jan 2021 17:01:59 +0800 Subject: [PATCH] fix l2 buffer error --- ge/graph/build/task_generator.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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);