Skip to content

Commit 1fafd12

Browse files
ankalinintprimak
authored andcommitted
benchdnn: brgemm: support f16 b matrix vnni layout for non-amx
1 parent a9d8a65 commit 1fafd12

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tests/benchdnn/brgemm/brgemm.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,7 @@ dnnl_status_t brgemm_attr_init(
142142
// PROCESS_KEY_VAL(bd_mask_level);
143143
PROCESS_KEY_VAL(use_uker);
144144
PROCESS_KEY_VAL(use_interleave_stores);
145+
PROCESS_KEY_VAL(b_is_vnni);
145146
PROCESS_KEY_VAL(postops_only);
146147
PROCESS_KEY_VAL(hint_bd_block);
147148
PROCESS_KEY_VAL(hint_bd_block2);
Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,21 @@
11
--reset
2-
32
--dt=f16,f16:f16:f32,f16:f32:f32,f32:f16:f32
43
--bia_dt=undef,f32,f16
54
--beta=0,1
65
--attr-post-ops=,sum:2,relu
6+
--brgemm-attr=b_is_vnni:0,b_is_vnni:1
77
--batch=option_set_f32 # f16 uses f32 blocking for avx512_core_fp16
88

99
# Separate cases for non-default alpha
1010
--reset
1111
--dt=f16
1212
--alpha=2
13+
--brgemm-attr=b_is_vnni:0,b_is_vnni:1
1314
--batch=shapes_2d_no_tail_f32
1415

1516
# Skip-acc feature
1617
--reset
17-
--brgemm-attr=generate_skip_accumulation:1
18+
--brgemm-attr=generate_skip_accumulation:1+b_is_vnni:0,generate_skip_accumulation:1+b_is_vnni:1
1819
--beta=0,1
1920
--bia_dt=f32
2021
--dt=f16
@@ -24,4 +25,5 @@
2425
--reset
2526
--wtag=ba
2627
--dt=f16
28+
--brgemm-attr=b_is_vnni:0,b_is_vnni:1
2729
--batch=option_set_f32

0 commit comments

Comments
 (0)