Define SBGEMM_ALIGN_K for DYNAMIC_ARCH build

This commit is contained in:
Honglin Zhu
2022-10-27 14:10:26 +08:00
parent 843e9fd0b9
commit 4989e039a5
5 changed files with 18 additions and 18 deletions

View File

@@ -62,6 +62,8 @@ gotoblas_t TABLE_NAME = {
MAX(SBGEMM_DEFAULT_UNROLL_M, SBGEMM_DEFAULT_UNROLL_N),
#endif
SBGEMM_ALIGN_K,
sbstobf16_kTS, sbdtobf16_kTS, sbf16tos_kTS, dbf16tod_kTS,
samax_kTS, samin_kTS, smax_kTS, smin_kTS,
@@ -973,12 +975,6 @@ static void init_parameter(void) {
TABLE_NAME.xgemm3m_r = TABLE_NAME.qgemm_r;
#endif
#endif
#if defined(NEOVERSEN2) && BUILD_BFLOAT16 == 1
TABLE_NAME.align_k = 4;
#else
TABLE_NAME.align_k = 1;
#endif
}
#else // (ARCH_ARM64)