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

add zyx_fsv32 format for onednn deconv (#11873)

Browse Source
This commit is contained in:
Sungeun Kim 2022-06-14 22:04:29 +09:00 committed by GitHub
parent 25e002808a
commit a454b7371e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/plugins/intel_gpu/src/graph/impls/onednn/deconvolution_onednn.cpp
View File

@ -183,6 +183,11 @@ attach_deconvolution_onednn::attach_deconvolution_onednn() {
std::make_tuple(data_types::u8, format::b_fs_yx_fsv32),
std::make_tuple(data_types::i8, format::b_fs_yx_fsv32),
std::make_tuple(data_types::f32, format::b_fs_zyx_fsv32),
std::make_tuple(data_types::f16, format::b_fs_zyx_fsv32),
std::make_tuple(data_types::u8, format::b_fs_zyx_fsv32),
std::make_tuple(data_types::i8, format::b_fs_zyx_fsv32),
std::make_tuple(data_types::f32, format::bs_fs_yx_bsv16_fsv16),
std::make_tuple(data_types::f16, format::bs_fs_yx_bsv16_fsv16),
std::make_tuple(data_types::u8, format::bs_fs_yx_bsv16_fsv16),

4
src/plugins/intel_gpu/src/graph/impls/onednn/utils.cpp
View File

@ -456,6 +456,10 @@ cldnn::format find_format(dnnl::memory::desc desc, bool is_grouped) {
} else if (compare_strides(order, {1, 0, 2, 3, 4})) {
return cldnn::format::is_os_zyx_isa8_osv8_isv2;
}
} else if (desc.data.ndims == 5 && blk.inner_nblks == 4 &&
blk.inner_blks[0] == 4 && blk.inner_blks[1] == 8 && blk.inner_blks[2] == 8 && blk.inner_blks[3] == 4 &&
blk.inner_idxs[0] == 0 && blk.inner_idxs[1] == 1 && blk.inner_idxs[2] == 0 && blk.inner_idxs[3] == 1) {
return cldnn::format::os_is_zyx_osa4_isa8_osv8_isv4;
}
}