Skip to content

Commit 2d5e090

Browse files
committed
benchdnn: reorder: support nCw4c, nChw4c and nCdhw4c
1 parent 07ddf70 commit 2d5e090

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

tests/benchdnn/dnn_types.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,16 +81,21 @@ data_kind_t fmt2data_kind(mkldnn_memory_format_t fmt) {
8181

8282
case mkldnn_ncw:
8383
case mkldnn_nwc:
84+
case mkldnn_nCw4c:
85+
case mkldnn_nCw8c:
8486
case mkldnn_nCw16c:
8587

8688
case mkldnn_nchw:
8789
case mkldnn_nhwc:
8890
case mkldnn_chwn:
91+
case mkldnn_nChw4c:
8992
case mkldnn_nChw8c:
9093
case mkldnn_nChw16c:
9194

9295
case mkldnn_ncdhw:
9396
case mkldnn_ndhwc:
97+
case mkldnn_nCdhw4c:
98+
case mkldnn_nCdhw8c:
9499
case mkldnn_nCdhw16c:
95100
return DATA;
96101

tests/benchdnn/inputs/reorder/test_default

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,16 +4,17 @@
44
--dt=f32,s32,s8,u8
55

66
--attr=irmode=nearest;oscale=per_oc:0.
7-
--fmt=nchw,nhwc,nChw8c,nChw16c 2x64x3x3
7+
--fmt=nchw,nhwc,nChw4c,nChw8c,nChw16c 2x64x3x3
88
--fmt=oihw,hwio 2x64x3x3
99
--fmt=goihw,gOIhw16i16o 3x32x32x2x2
1010

1111
--both-dir-fmt=true
1212
--ifmt=nChw8c --ofmt=nChw16c 2x40x3x3 # blocked with tail
13+
--ifmt=nChw4c --ofmt=nChw16c 2x40x3x3 # blocked with tail
1314

1415
--attr=irmode=down;oscale=common:0.
1516
--fmt=ncw,nwc 2x8x8 # special 8x8 kernel
16-
--fmt=nchw,nhwc,nChw8c,nChw16c 2x64x3x3
17+
--fmt=nchw,nhwc,nChw4c,nChw8c,nChw16c 2x64x3x3
1718
--fmt=oihw,hwio 2x64x3x3
1819

1920
#

0 commit comments

Comments
 (0)