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

[GPU] Pass eltwise unit tests on DG2 (#12113)

Browse Source 
* check fusion in onednn too
This commit is contained in:
Felix Dohyun Kim
2022-07-14 10:15:57 +09:00
committed by GitHub
parent 9d0e623af6
commit 3c1ed2ce1f
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -634,12 +634,10 @@ void prepare_primitive_fusing::fuse_simple_primitives(program &p) {
};
auto eltwise_supports_fusings = [&](eltwise_node& node) -> bool {
if (_lo.get_optimization_attributes().use_onednn_impls == 0) {
auto out_layout = node.get_output_layout();
if (out_layout.data_type == data_types::f16 && out_layout.batch() > 1 &&
(_lo.get_optimization_attributes().fs_b_yx_fsv32_network || out_layout.format == format::fs_b_yx_fsv32)) {
return false;
}
auto out_layout = node.get_output_layout();
if (out_layout.data_type == data_types::f16 && out_layout.batch() > 1 &&
(_lo.get_optimization_attributes().fs_b_yx_fsv32_network || out_layout.format == format::fs_b_yx_fsv32)) {
return false;
}
return true;
};