Browse Source

use macros for sgemm_direct to support dynamic_arch naming via common_s,h

pull/2782/head
Martin Kroeker GitHub 5 years ago
parent
commit
7fe38daee5
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      interface/gemm.c

+ 2
- 2
interface/gemm.c View File

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



Loading…
Cancel
Save