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 committed by Wangyang Guo
parent be3349405d
commit 4271cfcc6f
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
interface/gemm.c
View File

@ -145,7 +145,7 @@ 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;
#if defined(USE_SIMPLE_THREADED_LEVEL3) || !defined(NO_AFFINITY) #if defined(USE_SIMPLE_THREADED_LEVEL3) || !defined(NO_AFFINITY)
#ifndef COMPLEX #ifndef COMPLEX
@ -269,8 +269,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
#if defined(USE_SIMPLE_THREADED_LEVEL3) || !defined(NO_AFFINITY) #if defined(USE_SIMPLE_THREADED_LEVEL3) || !defined(NO_AFFINITY)
#ifndef COMPLEX #ifndef COMPLEX
#ifdef XDOUBLE #ifdef XDOUBLE
@ -438,7 +441,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)