Update gemm.c

This commit is contained in:
Martin Kroeker 2020-08-16 20:45:20 +02:00 committed by GitHub
parent a83cb3966d
commit 54e02aaf11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -275,8 +275,8 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
#ifdef DYNAMIC_ARCH #ifdef DYNAMIC_ARCH
if (support_avx512() ) if (support_avx512() )
#endif #endif
if (beta == 0 && alpha == 1.0 && order == CblasRowMajor && TransA == CblasNoTrans && TransB == CblasNoTrans && sgemm_kernel_direct_performant(m,n,k)) { if (beta == 0 && alpha == 1.0 && order == CblasRowMajor && TransA == CblasNoTrans && TransB == CblasNoTrans && sgemm_direct_performant(m,n,k)) {
sgemm_kernel_direct(m, n, k, a, lda, b, ldb, c, ldc); sgemm_direct(m, n, k, a, lda, b, ldb, c, ldc);
return; return;
} }