Add BUILD_vartype defines

This commit is contained in:
Martin Kroeker
2020-09-22 23:24:22 +02:00
committed by GitHub
parent 988a6f429e
commit 357bff06b5
2 changed files with 17 additions and 4 deletions

View File

@@ -459,13 +459,16 @@ blas_queue_t *tscq;
} else
#endif
if ((queue -> mode & BLAS_PREC) == BLAS_DOUBLE) {
#ifdef BUILD_DOUBLE
sb = (void *)(((BLASLONG)sa + ((DGEMM_P * DGEMM_Q * sizeof(double)
+ GEMM_ALIGN) & ~GEMM_ALIGN)) + GEMM_OFFSET_B);
#endif
} else if ((queue -> mode & BLAS_PREC) == BLAS_SINGLE) {
#ifdef BUILD_SINGLE
sb = (void *)(((BLASLONG)sa + ((SGEMM_P * SGEMM_Q * sizeof(float)
+ GEMM_ALIGN) & ~GEMM_ALIGN)) + GEMM_OFFSET_B);
} else {
#endif
} else {
/* Other types in future */
}
} else {
@@ -476,11 +479,15 @@ blas_queue_t *tscq;
} else
#endif
if ((queue -> mode & BLAS_PREC) == BLAS_DOUBLE){
#ifdef BUILD_COMPLEX16
sb = (void *)(((BLASLONG)sa + ((ZGEMM_P * ZGEMM_Q * 2 * sizeof(double)
+ GEMM_ALIGN) & ~GEMM_ALIGN)) + GEMM_OFFSET_B);
#endif
} else if ((queue -> mode & BLAS_PREC) == BLAS_SINGLE) {
#ifdef BUILD_COMPLEX
sb = (void *)(((BLASLONG)sa + ((CGEMM_P * CGEMM_Q * 2 * sizeof(float)
+ GEMM_ALIGN) & ~GEMM_ALIGN)) + GEMM_OFFSET_B);
#endif
} else {
/* Other types in future */
}

View File

@@ -2201,9 +2201,15 @@ static void *alloc_mmap(void *address){
#endif
#endif
#ifdef BUILD_DOUBLE
allocsize = DGEMM_P * DGEMM_Q * sizeof(double);
#elif defined(BUILD_COMPLEX16)
allocsize = ZGEMM_P * ZGEMM_Q * sizeof(double);
#elif defined(BUILD_COMPLEX)
allocsize = CGEMM_P * CGEMM_Q * sizeof(double);
#else
allocsize = SGEMM_P * SGEMM_Q * sizeof(double);
#endif
start = (BLASULONG)map_address;
current = (SCALING - 1) * BUFFER_SIZE;