From b62899ebb698735c69399740ef18ca0c66ff81e7 Mon Sep 17 00:00:00 2001 From: Andrey Kalinin Date: Tue, 24 Sep 2024 13:34:20 -0700 Subject: [PATCH] aarch64: add b_is_vnni member to brgemm_attr_t same as for x64 --- src/cpu/aarch64/brgemm/brgemm.cpp | 1 + src/cpu/aarch64/brgemm/brgemm_types.hpp | 1 + 2 files changed, 2 insertions(+) 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