diff --git a/src/cpu/aarch64/brgemm/brgemm.cpp b/src/cpu/aarch64/brgemm/brgemm.cpp index 6ed6cc59597..b7d962b9487 100644 --- a/src/cpu/aarch64/brgemm/brgemm.cpp +++ b/src/cpu/aarch64/brgemm/brgemm.cpp @@ -517,6 +517,7 @@ int brgemm_cmp(const brgemm_t &lhs, const brgemm_t &rhs) { CMP_BRGEMM_FIELD(brgattr.bd_mask_level); CMP_BRGEMM_FIELD(brgattr.use_uker); CMP_BRGEMM_FIELD(brgattr.use_interleave_stores); + CMP_BRGEMM_FIELD(brgattr.b_is_vnni); CMP_BRGEMM_FIELD(brgattr.fpmath_mode); CMP_BRGEMM_FIELD(brgattr.LDA2); CMP_BRGEMM_FIELD(brgattr.LDB2); diff --git a/src/cpu/aarch64/brgemm/brgemm_types.hpp b/src/cpu/aarch64/brgemm/brgemm_types.hpp index d6eb16cd6ff..bec29c3c8f1 100644 --- a/src/cpu/aarch64/brgemm/brgemm_types.hpp +++ b/src/cpu/aarch64/brgemm/brgemm_types.hpp @@ -147,6 +147,7 @@ struct DNNL_API brgemm_attr_t { // interleave stores or not bool use_interleave_stores; impl::fpmath_mode_t fpmath_mode = fpmath_mode::strict; + bool b_is_vnni {false}; // Second level leading dimension describing distance between 16-line // blocks in case of blocked layout. Used to calculate address of next // bd block. By default are equal to regular leading dimension parameters