floraachy/OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Define SBGEMM_ALIGN_K for DYNAMIC_ARCH build

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

11
driver/level3/level3.c
View File

@@ -305,13 +305,12 @@ int CNAME(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n,
}
BLASLONG pad_min_l = min_l;
#if defined(HALF) && defined(DYNAMIC_ARCH)
pad_min_l = (min_l + gotoblas->align_k - 1) & ~(gotoblas->align_k-1);
#if defined(HALF)
#if defined(DYNAMIC_ARCH)
pad_min_l = (min_l + gotoblas->sbgemm_align_k - 1) & ~(gotoblas->sbgemm_align_k-1);
#else
pad_min_l = (min_l + SBGEMM_ALIGN_K - 1) & ~(SBGEMM_ALIGN_K - 1);;
#endif
#if defined(HALF) && !defined(DYNAMIC_ARCH) && defined(NEOVERSEN2)
pad_min_l = (min_l + 3) & ~3;
#endif
/* First, we have to move data A to L2 cache */

10
driver/level3/level3_thread.c
View File

@@ -327,12 +327,12 @@ static int inner_thread(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n,
BLASLONG pad_min_l = min_l;
#if defined(HALF) && defined(DYNAMIC_ARCH)
pad_min_l = (min_l + gotoblas->align_k - 1) & ~(gotoblas->align_k-1);
#if defined(HALF)
#if defined(DYNAMIC_ARCH)
pad_min_l = (min_l + gotoblas->sbgemm_align_k - 1) & ~(gotoblas->sbgemm_align_k-1);
#else
pad_min_l = (min_l + SBGEMM_ALIGN_K - 1) & ~(SBGEMM_ALIGN_K - 1);;
#endif
#if defined(HALF) && !defined(DYNAMIC_ARCH) && defined(NEOVERSEN2)
pad_min_l = (min_l + 3) & ~3;
#endif
/* Determine step size in m