Skip to content

Commit 66622de

Browse files
author
Chip Kerchner
committed
Hack: Test gemv vs gemm.
1 parent ba47c7f commit 66622de

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

interface/gemm.c

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,22 +47,29 @@
4747
#define SMP_THRESHOLD_MIN 65536.0
4848
#ifdef XDOUBLE
4949
#define ERROR_NAME "QGEMM "
50+
#define GEMV BLASFUNC(qgemv)
5051
#elif defined(DOUBLE)
5152
#define ERROR_NAME "DGEMM "
53+
#define GEMV BLASFUNC(dgemv)
5254
#elif defined(BFLOAT16)
5355
#define ERROR_NAME "SBGEMM "
56+
#define GEMV BLASFUNC(sbgemv)
5457
#else
5558
#define ERROR_NAME "SGEMM "
59+
#define GEMV BLASFUNC(sgemv)
5660
#endif
5761
#else
5862
#define SMP_THRESHOLD_MIN 8192.0
5963
#ifndef GEMM3M
6064
#ifdef XDOUBLE
6165
#define ERROR_NAME "XGEMM "
66+
#define GEMV BLASFUNC(xgemv)
6267
#elif defined(DOUBLE)
6368
#define ERROR_NAME "ZGEMM "
69+
#define GEMV BLASFUNC(zgemv)
6470
#else
6571
#define ERROR_NAME "CGEMM "
72+
#define GEMV BLASFUNC(cgemv)
6673
#endif
6774
#else
6875
#ifdef XDOUBLE
@@ -190,6 +197,16 @@ void NAME(char *TRANSA, char *TRANSB,
190197
IFLOAT *buffer;
191198
IFLOAT *sa, *sb;
192199

200+
#if !defined(COMPLEX) && !defined(DOUBLE) && !defined(BFLOAT16)
201+
#if 1
202+
if (*N == 1) {
203+
GEMV(TRANSA, K, M, alpha, a, ldA, b, N, beta, c, N);
204+
//SUBROUTINE SGEMV(TRANS,M,N,ALPHA,A,LDA,X,INCX,BETA,Y,INCY)
205+
return;
206+
}
207+
#endif
208+
#endif
209+
193210
#ifdef SMP
194211
double MNK;
195212
#if defined(USE_SIMPLE_THREADED_LEVEL3) || !defined(NO_AFFINITY)

0 commit comments

Comments
 (0)