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

Fix gemm interface bug for small matrix.

This commit is contained in:
Xianyi Zhang 2020-04-28 23:15:20 +08:00
parent 43bef4aaac
commit 92372c70fc
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
interface/gemm.c
View File

@ -143,8 +143,11 @@ void NAME(char *TRANSA, char *TRANSB,
IFLOAT *buffer; IFLOAT *buffer;
IFLOAT *sa, *sb; IFLOAT *sa, *sb;
#ifdef SMP #if defined (SMP) || defined(SMALL_MATRIX_OPT)
double MNK; double MNK;
#endif
#ifdef SMP
#ifndef COMPLEX #ifndef COMPLEX
#ifdef XDOUBLE #ifdef XDOUBLE
int mode = BLAS_XDOUBLE | BLAS_REAL; int mode = BLAS_XDOUBLE | BLAS_REAL;
@ -265,8 +268,11 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
XFLOAT *buffer; XFLOAT *buffer;
XFLOAT *sa, *sb; XFLOAT *sa, *sb;
#ifdef SMP #if defined (SMP) || defined(SMALL_MATRIX_OPT)
double MNK; double MNK;
#endif
#ifdef SMP
#ifndef COMPLEX #ifndef COMPLEX
#ifdef XDOUBLE #ifdef XDOUBLE
int mode = BLAS_XDOUBLE | BLAS_REAL; int mode = BLAS_XDOUBLE | BLAS_REAL;
@ -432,7 +438,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
FUNCTION_PROFILE_START(); FUNCTION_PROFILE_START();
#if defined(SMP) || defined(SMALL_MATRIX_OPT)
MNK = (double) args.m * (double) args.n * (double) args.k; MNK = (double) args.m * (double) args.n * (double) args.k;
#endif
#ifdef SMALL_MATRIX_OPT #ifdef SMALL_MATRIX_OPT
#if !defined(COMPLEX) #if !defined(COMPLEX)