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

[GPU] Do not use OneDNN for FP32 pipeline (#8737)

Browse Source
This commit is contained in:
Sergey Shlyapnikov authored and GitHub committed 2021-11-25 10:58:37 +03:00
1 parent be04b856e2
commit c10a981f1b
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp
Vendored
+1 -1
View File
@@ -856,7 +856,7 @@ layout layout_optimizer::get_expected_layout(layout const& current_layout,
bool non_grouped = prim->groups == 1;
bool is_2d = input_layout.format.spatial_num() == 2;
bool onednn_valid_post_ops = get_post_ops_count(node) <= 32;
bool use_onednn_impls = _optimization_attributes.use_onednn_impls;
bool use_onednn_impls = _optimization_attributes.use_onednn_impls && input_layout.data_type != data_types::f32;
bool i8_u8_input = input_layout.data_type == data_types::u8 || input_layout.data_type == data_types::i8;
if (use_onednn_impls && onednn_valid_post_ops) {