Browse Source

Merge pull request #2533 from martin-frbg/gemmdirect2

Use runtime check for AVX512 capability in DYNAMIC_ARCH builds made on SKX
tags/v0.3.10^2
Martin Kroeker GitHub 5 years ago
parent
commit
018bb3e433
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions
  1. +2
    -0
      common.h
  2. +1
    -1
      interface/gemm.c

+ 2
- 0
common.h View File

@@ -657,6 +657,8 @@ void gotoblas_dynamic_init(void);
void gotoblas_dynamic_quit(void);
void gotoblas_profile_init(void);
void gotoblas_profile_quit(void);
int support_avx512(void);

#ifdef USE_OPENMP



+ 1
- 1
interface/gemm.c View File

@@ -273,7 +273,7 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS

#if !defined(COMPLEX) && !defined(DOUBLE) && defined(USE_SGEMM_KERNEL_DIRECT)
#ifdef DYNAMIC_ARCH
if (gotoblas == &gotoblas_SKYLAKEX)
if (support_avx512() )
#endif
if (beta == 0 && alpha == 1.0 && order == CblasRowMajor && TransA == CblasNoTrans && TransB == CblasNoTrans && sgemm_kernel_direct_performant(m,n,k)) {
sgemm_kernel_direct(m, n, k, a, lda, b, ldb, c, ldc);


Loading…
Cancel
Save