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

[GPU] add errata for onednn layout selection (#13758)

Browse Source
This commit is contained in:
Sungeun Kim 2022-11-04 20:44:36 +09:00 committed by GitHub
parent 0646d49d35
commit f1cf1948d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/plugins/intel_gpu/src/graph/layout_optimizer.cpp
View File

@ -236,6 +236,14 @@ bool layout_optimizer::can_fuse_reorder(program_node& prev, program_node& next,
if (next.is_type<reorder>())
return true;
// Errata for onednn layout selection
if (next.is_type<convolution>() && next.get_dependencies().size() == 1 &&
next.get_preferred_impl_type() == impl_types::onednn &&
((fmt_prev == format::byxf && fmt_next == format::byxf) ||
(fmt_prev == format::bfyx && fmt_next == format::byxf))) {
return true;
}
// Do not remove reorder if it is necessary to fulfill required_input
auto& reorder_node = next.get_dependency(0);
auto reorder_layout = reorder_node.get_output_layout();