Skip to content

Commit fb52a72

Browse files
committed
aarch64: add b_is_vnni member to brgemm_attr_t same as for x64
1 parent 146b7a6 commit fb52a72

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

src/cpu/aarch64/brgemm/brgemm.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -517,6 +517,7 @@ int brgemm_cmp(const brgemm_t &lhs, const brgemm_t &rhs) {
517517
CMP_BRGEMM_FIELD(brgattr.bd_mask_level);
518518
CMP_BRGEMM_FIELD(brgattr.use_uker);
519519
CMP_BRGEMM_FIELD(brgattr.use_interleave_stores);
520+
CMP_BRGEMM_FIELD(brgattr.b_is_vnni);
520521
CMP_BRGEMM_FIELD(brgattr.fpmath_mode);
521522
CMP_BRGEMM_FIELD(brgattr.LDA2);
522523
CMP_BRGEMM_FIELD(brgattr.LDB2);

src/cpu/aarch64/brgemm/brgemm_types.hpp

+1
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,7 @@ struct DNNL_API brgemm_attr_t {
147147
// interleave stores or not
148148
bool use_interleave_stores;
149149
impl::fpmath_mode_t fpmath_mode = fpmath_mode::strict;
150+
bool b_is_vnni {false};
150151
// Second level leading dimension describing distance between 16-line
151152
// blocks in case of blocked layout. Used to calculate address of next
152153
// bd block. By default are equal to regular leading dimension parameters

0 commit comments

Comments
 (0)