From 99dbb358358067ca72d2c0afad662c9d4e609038 Mon Sep 17 00:00:00 2001 From: "Min, Byungil" Date: Thu, 14 Jul 2022 11:04:14 +0900 Subject: [PATCH] [GPU] modify fusing condition for reduce (#12119) Signed-off-by: Min, Byungil --- .../src/graph/graph_optimizer/prepare_primitive_fusing.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/plugins/intel_gpu/src/graph/graph_optimizer/prepare_primitive_fusing.cpp b/src/plugins/intel_gpu/src/graph/graph_optimizer/prepare_primitive_fusing.cpp index 28dbec14e55..4d872410353 100644 --- a/src/plugins/intel_gpu/src/graph/graph_optimizer/prepare_primitive_fusing.cpp +++ b/src/plugins/intel_gpu/src/graph/graph_optimizer/prepare_primitive_fusing.cpp @@ -624,7 +624,8 @@ void prepare_primitive_fusing::fuse_simple_primitives(program &p) { (find(axes.begin(), axes.end(), reduce::along_y) != axes.end() && node.input().get_output_layout().spatial(1) > 16) || (find(axes.begin(), axes.end(), reduce::along_f) != axes.end() && - node.input().get_output_layout().feature() > 16))) + node.input().get_output_layout().feature() > 16) || + (node.get_output_layout().count() > 256))) return false; if (keep_dims)