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

Add BUILD_vartype defines

This commit is contained in:
Martin Kroeker 2020-09-22 23:23:33 +02:00 committed by GitHub
parent e5e2fbd593
commit 988a6f429e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
driver/level3/syrk_thread.c
View File

@ -56,12 +56,16 @@ int CNAME(int mode, blas_arg_t *arg, BLASLONG *range_m, BLASLONG *range_n, int (
if (!(mode & BLAS_COMPLEX)) { if (!(mode & BLAS_COMPLEX)) {
switch (mode & BLAS_PREC) { switch (mode & BLAS_PREC) {
#ifdef BUILD_SINGLE
case BLAS_SINGLE: case BLAS_SINGLE:
mask = SGEMM_UNROLL_MN - 1; mask = SGEMM_UNROLL_MN - 1;
break; break;
#endif
#ifdef BUILD_DOUBLE
case BLAS_DOUBLE: case BLAS_DOUBLE:
mask = DGEMM_UNROLL_MN - 1; mask = DGEMM_UNROLL_MN - 1;
break; break;
#endif
#ifdef EXPRECISION #ifdef EXPRECISION
case BLAS_XDOUBLE: case BLAS_XDOUBLE:
mask = MAX(QGEMM_UNROLL_M, QGEMM_UNROLL_N) - 1; mask = MAX(QGEMM_UNROLL_M, QGEMM_UNROLL_N) - 1;
@ -70,12 +74,16 @@ int CNAME(int mode, blas_arg_t *arg, BLASLONG *range_m, BLASLONG *range_n, int (
} }
} else { } else {
switch (mode & BLAS_PREC) { switch (mode & BLAS_PREC) {
#ifdef BUILD_COMPLEX
case BLAS_SINGLE: case BLAS_SINGLE:
mask = CGEMM_UNROLL_MN - 1; mask = CGEMM_UNROLL_MN - 1;
break; break;
#endif
#ifdef BUILD_COMPLEX16
case BLAS_DOUBLE: case BLAS_DOUBLE:
mask = ZGEMM_UNROLL_MN - 1; mask = ZGEMM_UNROLL_MN - 1;
break; break;
#endif
#ifdef EXPRECISION #ifdef EXPRECISION
case BLAS_XDOUBLE: case BLAS_XDOUBLE:
mask = MAX(XGEMM_UNROLL_M, XGEMM_UNROLL_N) - 1; mask = MAX(XGEMM_UNROLL_M, XGEMM_UNROLL_N) - 1;