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

[GPU] Use double blocked format if batch >= 16 (#9357)

Browse Source
This commit is contained in:
Mingyu Kim 2021-12-23 16:58:33 +09:00 committed by GitHub
parent 3b03728807
commit 16490959e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp vendored
View File

@ -920,7 +920,7 @@ layout layout_optimizer::get_expected_layout(layout const& current_layout,
if (i8_u8_input) {
if ((non_grouped || valid_grouped || valid_int8_dw) && onednn_valid_post_ops && is_2d) {
if (input_layout.size.batch[0] % 16 == 0) {
if (input_layout.size.batch[0] >= 16) {
expected_format = cldnn::format::bs_fs_yx_bsv32_fsv32;
} else {
if (data_type_traits::is_floating_point(output_layout.data_type) &&