[GPU] Allow 3d fsv16 layout for quantize and reduce prims (#9697)
This commit is contained in:
parent
dd8a073aa4
commit
4c36f8d16d
@ -102,58 +102,73 @@ attach_quantize_impl::attach_quantize_impl() {
|
|||||||
std::make_tuple(data_types::f32, format::fs_b_yx_fsv32),
|
std::make_tuple(data_types::f32, format::fs_b_yx_fsv32),
|
||||||
std::make_tuple(data_types::i8, format::fs_b_yx_fsv32),
|
std::make_tuple(data_types::i8, format::fs_b_yx_fsv32),
|
||||||
std::make_tuple(data_types::u8, format::fs_b_yx_fsv32),
|
std::make_tuple(data_types::u8, format::fs_b_yx_fsv32),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::f32, format::b_fs_yx_fsv16),
|
||||||
std::make_tuple(data_types::f16, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::f16, format::b_fs_yx_fsv16),
|
||||||
std::make_tuple(data_types::i8, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::i8, format::b_fs_yx_fsv16),
|
||||||
std::make_tuple(data_types::u8, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::u8, format::b_fs_yx_fsv16),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::byxf),
|
std::make_tuple(data_types::f32, format::byxf),
|
||||||
std::make_tuple(data_types::f16, format::byxf),
|
std::make_tuple(data_types::f16, format::byxf),
|
||||||
std::make_tuple(data_types::u8, format::byxf),
|
std::make_tuple(data_types::u8, format::byxf),
|
||||||
std::make_tuple(data_types::i8, format::byxf),
|
std::make_tuple(data_types::i8, format::byxf),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::b_fs_yx_fsv4),
|
std::make_tuple(data_types::f32, format::b_fs_yx_fsv4),
|
||||||
std::make_tuple(data_types::f16, format::b_fs_yx_fsv4),
|
std::make_tuple(data_types::f16, format::b_fs_yx_fsv4),
|
||||||
std::make_tuple(data_types::u8, format::b_fs_yx_fsv4),
|
std::make_tuple(data_types::u8, format::b_fs_yx_fsv4),
|
||||||
std::make_tuple(data_types::i8, format::b_fs_yx_fsv4),
|
std::make_tuple(data_types::i8, format::b_fs_yx_fsv4),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::b_fs_yx_fsv32),
|
std::make_tuple(data_types::f32, format::b_fs_yx_fsv32),
|
||||||
std::make_tuple(data_types::f16, format::b_fs_yx_fsv32),
|
std::make_tuple(data_types::f16, format::b_fs_yx_fsv32),
|
||||||
std::make_tuple(data_types::u8, format::b_fs_yx_fsv32),
|
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::i8, format::b_fs_yx_fsv32),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::b_fs_zyx_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::f16, format::b_fs_zyx_fsv32),
|
||||||
std::make_tuple(data_types::u8, 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::i8, format::b_fs_zyx_fsv32),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bs_fs_yx_bsv16_fsv16),
|
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::f16, format::bs_fs_yx_bsv16_fsv16),
|
||||||
std::make_tuple(data_types::u8, format::bs_fs_yx_bsv16_fsv16),
|
std::make_tuple(data_types::u8, format::bs_fs_yx_bsv16_fsv16),
|
||||||
std::make_tuple(data_types::i8, format::bs_fs_yx_bsv16_fsv16),
|
std::make_tuple(data_types::i8, format::bs_fs_yx_bsv16_fsv16),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bs_fs_zyx_bsv16_fsv16),
|
std::make_tuple(data_types::f32, format::bs_fs_zyx_bsv16_fsv16),
|
||||||
std::make_tuple(data_types::f16, format::bs_fs_zyx_bsv16_fsv16),
|
std::make_tuple(data_types::f16, format::bs_fs_zyx_bsv16_fsv16),
|
||||||
std::make_tuple(data_types::u8, format::bs_fs_zyx_bsv16_fsv16),
|
std::make_tuple(data_types::u8, format::bs_fs_zyx_bsv16_fsv16),
|
||||||
std::make_tuple(data_types::i8, format::bs_fs_zyx_bsv16_fsv16),
|
std::make_tuple(data_types::i8, format::bs_fs_zyx_bsv16_fsv16),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bfyx),
|
std::make_tuple(data_types::f32, format::bfyx),
|
||||||
std::make_tuple(data_types::f16, format::bfyx),
|
std::make_tuple(data_types::f16, format::bfyx),
|
||||||
std::make_tuple(data_types::i32, format::bfyx),
|
std::make_tuple(data_types::i32, format::bfyx),
|
||||||
std::make_tuple(data_types::u8, format::bfyx),
|
std::make_tuple(data_types::u8, format::bfyx),
|
||||||
std::make_tuple(data_types::i8, format::bfyx),
|
std::make_tuple(data_types::i8, format::bfyx),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::byxf),
|
std::make_tuple(data_types::f32, format::byxf),
|
||||||
std::make_tuple(data_types::f16, format::byxf),
|
std::make_tuple(data_types::f16, format::byxf),
|
||||||
std::make_tuple(data_types::i32, format::byxf),
|
std::make_tuple(data_types::i32, format::byxf),
|
||||||
std::make_tuple(data_types::u8, format::byxf),
|
std::make_tuple(data_types::u8, format::byxf),
|
||||||
std::make_tuple(data_types::i8, format::byxf),
|
std::make_tuple(data_types::i8, format::byxf),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::yxfb),
|
std::make_tuple(data_types::f32, format::yxfb),
|
||||||
std::make_tuple(data_types::f16, format::yxfb),
|
std::make_tuple(data_types::f16, format::yxfb),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bfzyx),
|
std::make_tuple(data_types::f32, format::bfzyx),
|
||||||
std::make_tuple(data_types::f16, format::bfzyx),
|
std::make_tuple(data_types::f16, format::bfzyx),
|
||||||
std::make_tuple(data_types::i32, format::bfzyx),
|
std::make_tuple(data_types::i32, format::bfzyx),
|
||||||
std::make_tuple(data_types::u8, format::bfzyx),
|
std::make_tuple(data_types::u8, format::bfzyx),
|
||||||
std::make_tuple(data_types::i8, format::bfzyx),
|
std::make_tuple(data_types::i8, format::bfzyx),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bfwzyx),
|
std::make_tuple(data_types::f32, format::bfwzyx),
|
||||||
std::make_tuple(data_types::f16, format::bfwzyx),
|
std::make_tuple(data_types::f16, format::bfwzyx),
|
||||||
std::make_tuple(data_types::i32, format::bfwzyx),
|
std::make_tuple(data_types::i32, format::bfwzyx),
|
||||||
std::make_tuple(data_types::u8, format::bfwzyx),
|
std::make_tuple(data_types::u8, format::bfwzyx),
|
||||||
std::make_tuple(data_types::i8, format::bfwzyx),
|
std::make_tuple(data_types::i8, format::bfwzyx),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::b_fs_zyx_fsv16),
|
std::make_tuple(data_types::f32, format::b_fs_zyx_fsv16),
|
||||||
std::make_tuple(data_types::f16, format::b_fs_zyx_fsv16),
|
std::make_tuple(data_types::f16, format::b_fs_zyx_fsv16),
|
||||||
|
std::make_tuple(data_types::u8, format::b_fs_zyx_fsv16),
|
||||||
|
std::make_tuple(data_types::i8, format::b_fs_zyx_fsv16),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bs_fs_yx_bsv32_fsv32),
|
std::make_tuple(data_types::f32, format::bs_fs_yx_bsv32_fsv32),
|
||||||
std::make_tuple(data_types::f16, format::bs_fs_yx_bsv32_fsv32),
|
std::make_tuple(data_types::f16, format::bs_fs_yx_bsv32_fsv32),
|
||||||
|
@ -86,21 +86,31 @@ attach_reduce_impl::attach_reduce_impl() {
|
|||||||
std::make_tuple(data_types::i32, format::bfyx),
|
std::make_tuple(data_types::i32, format::bfyx),
|
||||||
std::make_tuple(data_types::i8, format::bfyx),
|
std::make_tuple(data_types::i8, format::bfyx),
|
||||||
std::make_tuple(data_types::u8, format::bfyx),
|
std::make_tuple(data_types::u8, format::bfyx),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bfzyx),
|
std::make_tuple(data_types::f32, format::bfzyx),
|
||||||
std::make_tuple(data_types::f16, format::bfzyx),
|
std::make_tuple(data_types::f16, format::bfzyx),
|
||||||
std::make_tuple(data_types::i32, format::bfzyx),
|
std::make_tuple(data_types::i32, format::bfzyx),
|
||||||
std::make_tuple(data_types::i8, format::bfzyx),
|
std::make_tuple(data_types::i8, format::bfzyx),
|
||||||
std::make_tuple(data_types::u8, format::bfzyx),
|
std::make_tuple(data_types::u8, format::bfzyx),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::bfwzyx),
|
std::make_tuple(data_types::f32, format::bfwzyx),
|
||||||
std::make_tuple(data_types::f16, format::bfwzyx),
|
std::make_tuple(data_types::f16, format::bfwzyx),
|
||||||
std::make_tuple(data_types::i32, format::bfwzyx),
|
std::make_tuple(data_types::i32, format::bfwzyx),
|
||||||
std::make_tuple(data_types::i8, format::bfwzyx),
|
std::make_tuple(data_types::i8, format::bfwzyx),
|
||||||
std::make_tuple(data_types::u8, format::bfwzyx),
|
std::make_tuple(data_types::u8, format::bfwzyx),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::f32, format::b_fs_yx_fsv16),
|
||||||
std::make_tuple(data_types::f16, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::f16, format::b_fs_yx_fsv16),
|
||||||
std::make_tuple(data_types::i32, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::i32, format::b_fs_yx_fsv16),
|
||||||
std::make_tuple(data_types::i8, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::i8, format::b_fs_yx_fsv16),
|
||||||
std::make_tuple(data_types::u8, format::b_fs_yx_fsv16),
|
std::make_tuple(data_types::u8, format::b_fs_yx_fsv16),
|
||||||
|
|
||||||
|
std::make_tuple(data_types::f32, format::b_fs_zyx_fsv16),
|
||||||
|
std::make_tuple(data_types::f16, format::b_fs_zyx_fsv16),
|
||||||
|
std::make_tuple(data_types::i32, format::b_fs_zyx_fsv16),
|
||||||
|
std::make_tuple(data_types::i8, format::b_fs_zyx_fsv16),
|
||||||
|
std::make_tuple(data_types::u8, format::b_fs_zyx_fsv16),
|
||||||
|
|
||||||
std::make_tuple(data_types::f32, format::b_fs_yx_fsv32),
|
std::make_tuple(data_types::f32, format::b_fs_yx_fsv32),
|
||||||
std::make_tuple(data_types::f16, format::b_fs_yx_fsv32),
|
std::make_tuple(data_types::f16, format::b_fs_yx_fsv32),
|
||||||
std::make_tuple(data_types::i32, format::b_fs_yx_fsv32),
|
std::make_tuple(data_types::i32, format::b_fs_yx_fsv32),
|
||||||
|
Loading…
Reference in New Issue
Block a user