From 188434e96909536d2293d65ad2dbd3ab122eaddd Mon Sep 17 00:00:00 2001 From: Roman Lyamin Date: Wed, 16 Aug 2023 08:59:43 +0400 Subject: [PATCH] [GPU] image_2d_weights_winograd_6x3_s1 formats fix (#19205) --- src/plugins/intel_gpu/src/kernel_selector/tensor_type.cpp | 4 ++-- src/plugins/intel_gpu/src/runtime/format.cpp | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/plugins/intel_gpu/src/kernel_selector/tensor_type.cpp b/src/plugins/intel_gpu/src/kernel_selector/tensor_type.cpp index 43c534d3cdcdc0..c3ca64875b6fdd 100644 --- a/src/plugins/intel_gpu/src/kernel_selector/tensor_type.cpp +++ b/src/plugins/intel_gpu/src/kernel_selector/tensor_type.cpp @@ -100,8 +100,8 @@ WeightsTensor::WeightsChannelArray WeightsTensor::weightsChannelArray {{ { WeightsLayout::winograd_2x3_s1_weights, { 3, 2, -1, 1, 0, -1 } }, { WeightsLayout::winograd_2x3_s1_fused_weights, { 0, 1, -1, 2, 3, -1 } }, { WeightsLayout::winograd_6x3_s1_fused_weights, { 0, 1, -1, 2, 3, -1 } }, - { WeightsLayout::image_2d_weights_winograd_6x3_s1_fbxyb, { 0, 1, -1, 2, 3, -1 } }, - { WeightsLayout::image_2d_weights_winograd_6x3_s1_xfbyb, { 0, 1, -1, 2, 3, -1 } }, + { WeightsLayout::image_2d_weights_winograd_6x3_s1_fbxyb, { 1, 0, -1, 3, 2, -1 } }, + { WeightsLayout::image_2d_weights_winograd_6x3_s1_xfbyb, { 3, 0, -1, 2, 1, -1 } }, { WeightsLayout::dlstm_dir_io, { 1, 0, -1, 2, 3, -1 } }, { WeightsLayout::os_is_yx_isa8_osv8_isv4, { 0, 1, -1, 2, 3, -1 } }, { WeightsLayout::os_is_yx_isa8_osv16_isv4, { 0, 1, -1, 2, 3, -1 } }, diff --git a/src/plugins/intel_gpu/src/runtime/format.cpp b/src/plugins/intel_gpu/src/runtime/format.cpp index ca94a647e1a2e2..7b501a4daf04bc 100644 --- a/src/plugins/intel_gpu/src/runtime/format.cpp +++ b/src/plugins/intel_gpu/src/runtime/format.cpp @@ -88,8 +88,8 @@ static const std::map format_traits_map { FMT_TRAITS(winograd_2x3_s1_weights, 1, 1, 2, 0, {0, 1, 2, 3}, "oiyx", "oixy", {}), FMT_TRAITS(winograd_2x3_s1_fused_weights, 1, 1, 2, 0, {3, 2, 1, 0}, "xyio", "oixy", {}), FMT_TRAITS(winograd_6x3_s1_fused_weights, 1, 1, 2, 0, {3, 2, 1, 0}, "xyio", "oixy", {}), - FMT_TRAITS(image_2d_weights_winograd_6x3_s1_fbxyb, 1, 1, 2, 0, {3, 2, 1, 0}, "xyio", "oixy", {}), - FMT_TRAITS(image_2d_weights_winograd_6x3_s1_xfbyb, 1, 1, 2, 0, {3, 2, 1, 0}, "xyio", "oixy", {}), + FMT_TRAITS(image_2d_weights_winograd_6x3_s1_fbxyb, 1, 1, 2, 0, {1, 0, 3, 2}, "ioxy", "oixy", {}), + FMT_TRAITS(image_2d_weights_winograd_6x3_s1_xfbyb, 1, 1, 2, 0, {3, 1, 0, 2}, "xioy", "oixy", {}), FMT_TRAITS(image_2d_weights_c4_fyx_b, 1, 1, 2, 0, {0, 1, 2, 3}, "oiyx", "oixy", {}), FMT_TRAITS(image_2d_weights_c1_b_fyx, 1, 1, 2, 0, {0, 1, 2, 3}, "oiyx", "oixy", {}), FMT_TRAITS(lstm_weights_dio, 1, 1, 2, 0, {0, 1, 3, 2}, "oixy", "oixy", {}),