IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] modify fusing condition for reduce (#12119)

Browse Source 
Signed-off-by: Min, Byungil <byungil.min@intel.com>
This commit is contained in:
Min, Byungil 2022-07-14 11:04:14 +09:00 committed by GitHub
parent 3c1ed2ce1f
commit 99dbb35835
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/plugins/intel_gpu/src/graph/graph_optimizer/prepare_primitive_fusing.cpp
View File

@ -624,7 +624,8 @@ void prepare_primitive_fusing::fuse_simple_primitives(program &p) {
(find(axes.begin(), axes.end(), reduce::along_y) != axes.end() && (find(axes.begin(), axes.end(), reduce::along_y) != axes.end() &&
node.input().get_output_layout().spatial(1) > 16) || node.input().get_output_layout().spatial(1) > 16) ||
(find(axes.begin(), axes.end(), reduce::along_f) != axes.end() && (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; return false;
if (keep_dims) if (keep_dims)