Skip to content

Commit f8d7c2e

Browse files
nivas-x86tprimak
nivas-x86
authored and
tprimak
committed
x64: ip: add avx2_vnni_2 xf16 tags
1 parent f5654f5 commit f8d7c2e

File tree

8 files changed

+123
-14
lines changed

8 files changed

+123
-14
lines changed

‎include/oneapi/dnnl/dnnl.hpp‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2427,6 +2427,22 @@ struct memory : public handle<dnnl_memory_t> {
24272427
ABcde4b24a4b = dnnl_ABcde4b24a4b,
24282428
OhwI24o = dnnl_OhwI24o,
24292429
gOhwI24o = dnnl_gOhwI24o,
2430+
AB8b24a2b = dnnl_AB8b24a2b,
2431+
ABc8b24a2b = dnnl_ABc8b24a2b,
2432+
ABcd8b24a2b = dnnl_ABcd8b24a2b,
2433+
ABcde8b24a2b = dnnl_ABcde8b24a2b,
2434+
AB8b8a2b = dnnl_AB8b8a2b,
2435+
ABc8b8a2b = dnnl_ABc8b8a2b,
2436+
ABcd8b8a2b = dnnl_ABcd8b8a2b,
2437+
ABcde8b8a2b = dnnl_ABcde8b8a2b,
2438+
OI8i8o2i = dnnl_OI8i8o2i,
2439+
OI8i24o2i = dnnl_OI8i24o2i,
2440+
OIw8i8o2i = dnnl_OIw8i8o2i,
2441+
OIw8i24o2i = dnnl_OIw8i24o2i,
2442+
OIhw8i8o2i = dnnl_OIhw8i8o2i,
2443+
OIhw8i24o2i = dnnl_OIhw8i24o2i,
2444+
OIdhw8i8o2i = dnnl_OIdhw8i8o2i,
2445+
OIdhw8i24o2i = dnnl_OIdhw8i24o2i,
24302446
};
24312447

24322448
/// A memory descriptor.

‎include/oneapi/dnnl/dnnl_types.h‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -882,6 +882,14 @@ typedef enum {
882882
dnnl_ABcd4b24a4b,
883883
dnnl_ABcde4b8a4b,
884884
dnnl_ABcde4b24a4b,
885+
dnnl_AB8b24a2b,
886+
dnnl_ABc8b24a2b,
887+
dnnl_ABcd8b24a2b,
888+
dnnl_ABcde8b24a2b,
889+
dnnl_AB8b8a2b,
890+
dnnl_ABc8b8a2b,
891+
dnnl_ABcd8b8a2b,
892+
dnnl_ABcde8b8a2b,
885893

886894
/// Just a sentinel, not real memory format tag. Must be changed after new
887895
/// format tag is added.
@@ -1084,7 +1092,9 @@ typedef enum {
10841092
dnnl_OI16i16o = dnnl_AB16b16a,
10851093
dnnl_OI16i32o = dnnl_AB16b32a,
10861094
dnnl_OI16i64o = dnnl_AB16b64a,
1095+
dnnl_OI8i8o2i = dnnl_AB8b8a2b,
10871096
dnnl_OI8i16o2i = dnnl_AB8b16a2b,
1097+
dnnl_OI8i24o2i = dnnl_AB8b24a2b,
10881098
dnnl_OI8i32o2i = dnnl_AB8b32a2b,
10891099
dnnl_OI8i64o2i = dnnl_AB8b64a2b,
10901100
dnnl_OI4i8o4i = dnnl_AB4b8a4b,
@@ -1118,7 +1128,9 @@ typedef enum {
11181128
dnnl_OIw4i4o = dnnl_ABc4b4a,
11191129
dnnl_OIw4o4i = dnnl_ABc4a4b,
11201130
dnnl_Oiw4o = dnnl_Abc4a,
1131+
dnnl_OIw8i8o2i = dnnl_ABc8b8a2b,
11211132
dnnl_OIw8i16o2i = dnnl_ABc8b16a2b,
1133+
dnnl_OIw8i24o2i = dnnl_ABc8b24a2b,
11221134
dnnl_OIw8i32o2i = dnnl_ABc8b32a2b,
11231135
dnnl_OIw8i64o2i = dnnl_ABc8b64a2b,
11241136
dnnl_OIw8i8o = dnnl_ABc8b8a,
@@ -1169,8 +1181,10 @@ typedef enum {
11691181
dnnl_OIhw4i4o = dnnl_ABcd4b4a,
11701182
dnnl_OIhw4o4i = dnnl_ABcd4a4b,
11711183
dnnl_Oihw4o = dnnl_Abcd4a,
1184+
dnnl_OIhw8i8o2i = dnnl_ABcd8b8a2b,
11721185
dnnl_OIhw8i16o2i = dnnl_ABcd8b16a2b,
11731186
dnnl_OIhw8i32o2i = dnnl_ABcd8b32a2b,
1187+
dnnl_OIhw8i24o2i = dnnl_ABcd8b24a2b,
11741188
dnnl_OIhw8i64o2i = dnnl_ABcd8b64a2b,
11751189
dnnl_OIhw8i8o = dnnl_ABcd8b8a,
11761190
dnnl_OIhw8o16i2o = dnnl_ABcd8a16b2a,
@@ -1202,8 +1216,10 @@ typedef enum {
12021216
dnnl_OIdhw4i4o = dnnl_ABcde4b4a,
12031217
dnnl_OIdhw4o4i = dnnl_ABcde4a4b,
12041218
dnnl_Oidhw4o = dnnl_Abcde4a,
1219+
dnnl_OIdhw8i8o2i = dnnl_ABcde8b8a2b,
12051220
dnnl_OIdhw8i16o2i = dnnl_ABcde8b16a2b,
12061221
dnnl_OIdhw8i32o2i = dnnl_ABcde8b32a2b,
1222+
dnnl_OIdhw8i24o2i = dnnl_ABcde8b24a2b,
12071223
dnnl_OIdhw8i64o2i = dnnl_ABcde8b64a2b,
12081224
dnnl_OIdhw8i8o = dnnl_ABcde8b8a,
12091225
dnnl_OIdhw8o16i2o = dnnl_ABcde8a16b2a,

‎src/common/c_types_map.hpp‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -878,6 +878,14 @@ const format_tag_t ABc8b16a = dnnl_ABc8b16a;
878878
const format_tag_t ABcd8b16a = dnnl_ABcd8b16a;
879879
const format_tag_t ABcde8b16a = dnnl_ABcde8b16a;
880880
const format_tag_t AB8b8a = dnnl_AB8b8a;
881+
const format_tag_t AB8b8a2b = dnnl_AB8b8a2b;
882+
const format_tag_t ABc8b8a2b = dnnl_ABc8b8a2b;
883+
const format_tag_t ABcd8b8a2b = dnnl_ABcd8b8a2b;
884+
const format_tag_t ABcde8b8a2b = dnnl_ABcde8b8a2b;
885+
const format_tag_t AB8b24a2b = dnnl_AB8b24a2b;
886+
const format_tag_t ABc8b24a2b = dnnl_ABc8b24a2b;
887+
const format_tag_t ABcd8b24a2b = dnnl_ABcd8b24a2b;
888+
const format_tag_t ABcde8b24a2b = dnnl_ABcde8b24a2b;
881889

882890
const format_tag_t last = dnnl_format_tag_last;
883891

@@ -1573,6 +1581,14 @@ const format_tag_t OIw8i16o = dnnl_OIw8i16o;
15731581
const format_tag_t OIhw8i16o = dnnl_OIhw8i16o;
15741582
const format_tag_t OIdhw8i16o = dnnl_OIdhw8i16o;
15751583
const format_tag_t OI8i8o = dnnl_OI8i8o;
1584+
const format_tag_t OI8i8o2i = dnnl_OI8i8o2i;
1585+
const format_tag_t OIw8i8o2i = dnnl_OIw8i8o2i;
1586+
const format_tag_t OIhw8i8o2i = dnnl_OIhw8i8o2i;
1587+
const format_tag_t OIdhw8i8o2i = dnnl_OIdhw8i8o2i;
1588+
const format_tag_t OI8i24o2i = dnnl_OI8i24o2i;
1589+
const format_tag_t OIw8i24o2i = dnnl_OIw8i24o2i;
1590+
const format_tag_t OIhw8i24o2i = dnnl_OIhw8i24o2i;
1591+
const format_tag_t OIdhw8i24o2i = dnnl_OIdhw8i24o2i;
15761592
} // namespace format_tag
15771593

15781594
using normalization_flags_t = dnnl_normalization_flags_t;

‎src/common/dnnl_debug_autogenerated.cpp‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -784,6 +784,14 @@ const char *dnnl_fmt_tag2str(dnnl_format_tag_t v) {
784784
if (v == dnnl_ABcd4b24a4b) return "ABcd4b24a4b";
785785
if (v == dnnl_ABcde4b8a4b) return "ABcde4b8a4b";
786786
if (v == dnnl_ABcde4b24a4b) return "ABcde4b24a4b";
787+
if (v == dnnl_AB8b24a2b) return "AB8b24a2b";
788+
if (v == dnnl_ABc8b24a2b) return "ABc8b24a2b";
789+
if (v == dnnl_ABcd8b24a2b) return "ABcd8b24a2b";
790+
if (v == dnnl_ABcde8b24a2b) return "ABcde8b24a2b";
791+
if (v == dnnl_AB8b8a2b) return "AB8b8a2b";
792+
if (v == dnnl_ABc8b8a2b) return "ABc8b8a2b";
793+
if (v == dnnl_ABcd8b8a2b) return "ABcd8b8a2b";
794+
if (v == dnnl_ABcde8b8a2b) return "ABcde8b8a2b";
787795
if (v == dnnl_format_tag_last) return "format_tag_last";
788796
if (v == dnnl_x) return "x";
789797
if (v == dnnl_nc) return "nc";
@@ -873,7 +881,9 @@ const char *dnnl_fmt_tag2str(dnnl_format_tag_t v) {
873881
if (v == dnnl_OI16i16o) return "OI16i16o";
874882
if (v == dnnl_OI16i32o) return "OI16i32o";
875883
if (v == dnnl_OI16i64o) return "OI16i64o";
884+
if (v == dnnl_OI8i8o2i) return "OI8i8o2i";
876885
if (v == dnnl_OI8i16o2i) return "OI8i16o2i";
886+
if (v == dnnl_OI8i24o2i) return "OI8i24o2i";
877887
if (v == dnnl_OI8i32o2i) return "OI8i32o2i";
878888
if (v == dnnl_OI8i64o2i) return "OI8i64o2i";
879889
if (v == dnnl_OI4i8o4i) return "OI4i8o4i";
@@ -905,7 +915,9 @@ const char *dnnl_fmt_tag2str(dnnl_format_tag_t v) {
905915
if (v == dnnl_OIw4i4o) return "OIw4i4o";
906916
if (v == dnnl_OIw4o4i) return "OIw4o4i";
907917
if (v == dnnl_Oiw4o) return "Oiw4o";
918+
if (v == dnnl_OIw8i8o2i) return "OIw8i8o2i";
908919
if (v == dnnl_OIw8i16o2i) return "OIw8i16o2i";
920+
if (v == dnnl_OIw8i24o2i) return "OIw8i24o2i";
909921
if (v == dnnl_OIw8i32o2i) return "OIw8i32o2i";
910922
if (v == dnnl_OIw8i64o2i) return "OIw8i64o2i";
911923
if (v == dnnl_OIw8i8o) return "OIw8i8o";
@@ -954,8 +966,10 @@ const char *dnnl_fmt_tag2str(dnnl_format_tag_t v) {
954966
if (v == dnnl_OIhw4i4o) return "OIhw4i4o";
955967
if (v == dnnl_OIhw4o4i) return "OIhw4o4i";
956968
if (v == dnnl_Oihw4o) return "Oihw4o";
969+
if (v == dnnl_OIhw8i8o2i) return "OIhw8i8o2i";
957970
if (v == dnnl_OIhw8i16o2i) return "OIhw8i16o2i";
958971
if (v == dnnl_OIhw8i32o2i) return "OIhw8i32o2i";
972+
if (v == dnnl_OIhw8i24o2i) return "OIhw8i24o2i";
959973
if (v == dnnl_OIhw8i64o2i) return "OIhw8i64o2i";
960974
if (v == dnnl_OIhw8i8o) return "OIhw8i8o";
961975
if (v == dnnl_OIhw8o16i2o) return "OIhw8o16i2o";
@@ -985,8 +999,10 @@ const char *dnnl_fmt_tag2str(dnnl_format_tag_t v) {
985999
if (v == dnnl_OIdhw4i4o) return "OIdhw4i4o";
9861000
if (v == dnnl_OIdhw4o4i) return "OIdhw4o4i";
9871001
if (v == dnnl_Oidhw4o) return "Oidhw4o";
1002+
if (v == dnnl_OIdhw8i8o2i) return "OIdhw8i8o2i";
9881003
if (v == dnnl_OIdhw8i16o2i) return "OIdhw8i16o2i";
9891004
if (v == dnnl_OIdhw8i32o2i) return "OIdhw8i32o2i";
1005+
if (v == dnnl_OIdhw8i24o2i) return "OIdhw8i24o2i";
9901006
if (v == dnnl_OIdhw8i64o2i) return "OIdhw8i64o2i";
9911007
if (v == dnnl_OIdhw8i8o) return "OIdhw8i8o";
9921008
if (v == dnnl_OIdhw8o16i2o) return "OIdhw8o16i2o";

‎src/common/memory_desc_wrapper.cpp‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -843,6 +843,15 @@ status_t memory_desc_wrapper::compute_blocking(
843843
C(ABcd8b16a, {0, 1, 2, 3}, {8, 16}, {1, 0});
844844
C(ABcde8b16a, {0, 1, 2, 3, 4}, {8, 16}, {1, 0});
845845
C(AB8b8a, {0, 1}, {8, 8}, {1, 0});
846+
847+
C(AB8b24a2b, {0, 1}, {8, 24, 2}, {1, 0, 1});
848+
C(ABc8b24a2b, {0, 1, 2}, {8, 24, 2}, {1, 0, 1});
849+
C(ABcd8b24a2b, {0, 1, 2, 3}, {8, 24, 2}, {1, 0, 1});
850+
C(ABcde8b24a2b, {0, 1, 2, 3, 4}, {8, 24, 2}, {1, 0, 1});
851+
C(AB8b8a2b, {0, 1}, {8, 8, 2}, {1, 0, 1});
852+
C(ABc8b8a2b, {0, 1, 2}, {8, 8, 2}, {1, 0, 1});
853+
C(ABcd8b8a2b, {0, 1, 2, 3}, {8, 8, 2}, {1, 0, 1});
854+
C(ABcde8b8a2b, {0, 1, 2, 3, 4}, {8, 8, 2}, {1, 0, 1});
846855
default: break;
847856
}
848857

‎src/common/tag_traits.hpp‎

Lines changed: 27 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,9 @@ enum class inner_blk_t {
125125
_4b32a4b,
126126
_4b64a4b,
127127
_2b8a4b,
128+
_8b8a2b,
128129
_8b16a2b,
130+
_8b24a2b,
129131
_8b32a2b,
130132
_8b64a2b,
131133
_8b16c2b,
@@ -190,19 +192,20 @@ constexpr int AB_or_BC_blk_off(int x0, int x1) {
190192
ib::_16b2c, ib::_16b4c, ib::_2c8b4c, ib::_8a16b2a,
191193
ib::_4b64a4b, ib::_4b32a4b, ib::_4b24a4b, ib::_4b16a4b,
192194
ib::_4b8a4b, ib::_2b8a4b, ib::_8b64a2b, ib::_8b32a2b,
193-
ib::_8b16a2b, ib::_8b16c2b, ib::_4c16b4c, ib::_8c16b2c,
194-
ib::_2b4c2b, ib::_2c4b2c, ib::_4b8c2b, ib::_4c8b2c,
195-
ib::_16a32b, ib::_16a48b, ib::_16a64b, ib::_16a16b2a,
196-
ib::_16a32b2a, ib::_16a48b2a, ib::_16a64b2a, ib::_16a16b4a,
197-
ib::_16a32b4a, ib::_16a48b4a, ib::_16a64b4a, ib::_16b16a2b,
198-
ib::_16b16a4b, ib::_16b16c2b, ib::_16c16b2c, ib::_16c16b4c,
199-
ib::_2a8b8a2b, ib::_2b8c8b2c, ib::_4a8b8a4b, ib::_4b8c8b4c,
200-
ib::_16b32a2b, ib::_16b48a2b, ib::_16b64a2b, ib::_16b32a4b,
201-
ib::_16b48a4b, ib::_16b64a4b, ib::_16c32b2c, ib::_16c48b2c,
202-
ib::_16c64b2c, ib::_16c32b4c, ib::_16c48b4c, ib::_16c64b4c,
203-
ib::_16b32c, ib::_16b48c, ib::_16b64c, ib::_16b32c2b,
204-
ib::_16b48c2b, ib::_16b64c2b, ib::_16b16c4b, ib::_16b32c4b,
205-
ib::_16b48c4b, ib::_16b64c4b, ib::_24a2b, ib::_24b2c),
195+
ib::_8b24a2b, ib::_8b16a2b, ib::_8b8a2b, ib::_8b16c2b,
196+
ib::_4c16b4c, ib::_8c16b2c, ib::_2b4c2b, ib::_2c4b2c,
197+
ib::_4b8c2b, ib::_4c8b2c, ib::_16a32b, ib::_16a48b,
198+
ib::_16a64b, ib::_16a16b2a, ib::_16a32b2a, ib::_16a48b2a,
199+
ib::_16a64b2a, ib::_16a16b4a, ib::_16a32b4a, ib::_16a48b4a,
200+
ib::_16a64b4a, ib::_16b16a2b, ib::_16b16a4b, ib::_16b16c2b,
201+
ib::_16c16b2c, ib::_16c16b4c, ib::_2a8b8a2b, ib::_2b8c8b2c,
202+
ib::_4a8b8a4b, ib::_4b8c8b4c, ib::_16b32a2b, ib::_16b48a2b,
203+
ib::_16b64a2b, ib::_16b32a4b, ib::_16b48a4b, ib::_16b64a4b,
204+
ib::_16c32b2c, ib::_16c48b2c, ib::_16c64b2c, ib::_16c32b4c,
205+
ib::_16c48b4c, ib::_16c64b4c, ib::_16b32c, ib::_16b48c,
206+
ib::_16b64c, ib::_16b32c2b, ib::_16b48c2b, ib::_16b64c2b,
207+
ib::_16b16c4b, ib::_16b32c4b, ib::_16b48c4b, ib::_16b64c4b,
208+
ib::_24a2b, ib::_24b2c),
206209
"unexpected inner_blk format");
207210

208211
// clang-format off
@@ -237,7 +240,9 @@ constexpr int AB_or_BC_blk_off(int x0, int x1) {
237240
: (f == ib::_2b8a4b || f == ib::_2c8b4c) ? (x1 / 4) * 32 + x0 * 4 + x1 % 4
238241
: (f == ib::_16b16a2b || f == ib::_16c16b2c) ? (x1 / 2) * 32 + x0 * 2 + x1 % 2
239242
: (f == ib::_16b16a4b || f == ib::_16c16b4c) ? (x1 / 4) * 64 + x0 * 4 + x1 % 4
243+
: (f == ib::_8b8a2b) ? (x1 / 2) * 16 + x0 * 2 + x1 % 2
240244
: (f == ib::_8b16a2b || f == ib::_8c16b2c) ? (x1 / 2) * 32 + x0 * 2 + x1 % 2
245+
: (f == ib::_8b24a2b) ? (x1 / 2) * 48 + x0 * 2 + x1 % 2
241246
: (f == ib::_8b32a2b) ? (x1 / 2) * 64 + x0 * 2 + x1 % 2
242247
: (f == ib::_8b64a2b) ? (x1 / 2) * 128 + x0 * 2 + x1 % 2
243248
: (f == ib::_2b4c2b || f == ib::_2c4b2c) ? (x0 / 2) * 8 + x1 * 2 + x0 % 2
@@ -790,6 +795,15 @@ DECL_TRAITS(ABc8b16a, _AB, _8b16a, 3);
790795
DECL_TRAITS(ABcd8b16a, _AB, _8b16a, 4);
791796
DECL_TRAITS(ABcde8b16a, _AB, _8b16a, 5);
792797
DECL_TRAITS(AB8b8a, _AB, _8b8a, 2);
798+
799+
DECL_TRAITS(AB8b8a2b, _AB, _8b8a2b, 2);
800+
DECL_TRAITS(ABc8b8a2b, _AB, _8b8a2b, 3);
801+
DECL_TRAITS(ABcd8b8a2b, _AB, _8b8a2b, 4);
802+
DECL_TRAITS(ABcde8b8a2b, _AB, _8b8a2b, 5);
803+
DECL_TRAITS(AB8b24a2b, _AB, _8b24a2b, 2);
804+
DECL_TRAITS(ABc8b24a2b, _AB, _8b24a2b, 3);
805+
DECL_TRAITS(ABcd8b24a2b, _AB, _8b24a2b, 4);
806+
DECL_TRAITS(ABcde8b24a2b, _AB, _8b24a2b, 5);
793807
} // namespace impl
794808
} // namespace dnnl
795809

‎src/cpu/x64/jit_brgemm_inner_product_utils.cpp‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,9 +176,15 @@ std::unordered_map<int, format_tag_t> get_desired_weights_tag(
176176
{32,
177177
pick(n_sp_dims, OI8i32o2i, OIw8i32o2i, OIhw8i32o2i,
178178
OIdhw8i32o2i)},
179+
{24,
180+
pick(n_sp_dims, OI8i24o2i, OIw8i24o2i, OIhw8i24o2i,
181+
OIdhw8i24o2i)},
179182
{16,
180183
pick(n_sp_dims, OI8i16o2i, OIw8i16o2i, OIhw8i16o2i,
181-
OIdhw8i16o2i)}};
184+
OIdhw8i16o2i)},
185+
{8,
186+
pick(n_sp_dims, OI8i8o2i, OIw8i8o2i, OIhw8i8o2i,
187+
OIdhw8i8o2i)}};
182188
}
183189
} else if (jbgp.wei_dt == data_type::s8) {
184190
if (jbgp.is_amx) {

‎tests/benchdnn/dnnl_debug_autogenerated.cpp‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -771,6 +771,14 @@ dnnl_format_tag_t str2fmt_tag(const char *str) {
771771
CASE(ABcd4b24a4b);
772772
CASE(ABcde4b8a4b);
773773
CASE(ABcde4b24a4b);
774+
CASE(AB8b24a2b);
775+
CASE(ABc8b24a2b);
776+
CASE(ABcd8b24a2b);
777+
CASE(ABcde8b24a2b);
778+
CASE(AB8b8a2b);
779+
CASE(ABc8b8a2b);
780+
CASE(ABcd8b8a2b);
781+
CASE(ABcde8b8a2b);
774782
CASE(x);
775783
CASE(nc);
776784
CASE(cn);
@@ -859,7 +867,9 @@ dnnl_format_tag_t str2fmt_tag(const char *str) {
859867
CASE(OI16i16o);
860868
CASE(OI16i32o);
861869
CASE(OI16i64o);
870+
CASE(OI8i8o2i);
862871
CASE(OI8i16o2i);
872+
CASE(OI8i24o2i);
863873
CASE(OI8i32o2i);
864874
CASE(OI8i64o2i);
865875
CASE(OI4i8o4i);
@@ -891,7 +901,9 @@ dnnl_format_tag_t str2fmt_tag(const char *str) {
891901
CASE(OIw4i4o);
892902
CASE(OIw4o4i);
893903
CASE(Oiw4o);
904+
CASE(OIw8i8o2i);
894905
CASE(OIw8i16o2i);
906+
CASE(OIw8i24o2i);
895907
CASE(OIw8i32o2i);
896908
CASE(OIw8i64o2i);
897909
CASE(OIw8i8o);
@@ -940,8 +952,10 @@ dnnl_format_tag_t str2fmt_tag(const char *str) {
940952
CASE(OIhw4i4o);
941953
CASE(OIhw4o4i);
942954
CASE(Oihw4o);
955+
CASE(OIhw8i8o2i);
943956
CASE(OIhw8i16o2i);
944957
CASE(OIhw8i32o2i);
958+
CASE(OIhw8i24o2i);
945959
CASE(OIhw8i64o2i);
946960
CASE(OIhw8i8o);
947961
CASE(OIhw8o16i2o);
@@ -971,8 +985,10 @@ dnnl_format_tag_t str2fmt_tag(const char *str) {
971985
CASE(OIdhw4i4o);
972986
CASE(OIdhw4o4i);
973987
CASE(Oidhw4o);
988+
CASE(OIdhw8i8o2i);
974989
CASE(OIdhw8i16o2i);
975990
CASE(OIdhw8i32o2i);
991+
CASE(OIdhw8i24o2i);
976992
CASE(OIdhw8i64o2i);
977993
CASE(OIdhw8i8o);
978994
CASE(OIdhw8o16i2o);

0 commit comments

Comments
 (0)