Browse Source

Add BUILD_vartype defines

tags/v0.3.11^2
Martin Kroeker GitHub 5 years ago
parent
commit
988a6f429e
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
  1. +8
    -0
      driver/level3/syrk_thread.c

+ 8
- 0
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;


Loading…
Cancel
Save