Skip to content

Commit 5a322f2

Browse files
authored
Merge pull request #5268 from martin-frbg/fix-dyn-sgemmdirect
Fix conditional inclusion of SGEMM_KERNEL_DIRECT
2 parents 0b0bb99 + 6680e05 commit 5a322f2

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

interface/gemm.c

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -417,21 +417,24 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
417417

418418
PRINT_DEBUG_CNAME;
419419

420-
#if !defined(COMPLEX) && !defined(DOUBLE) && !defined(BFLOAT16) && defined(USE_SGEMM_KERNEL_DIRECT)
421-
#if defined(DYNAMIC_ARCH) && defined(ARCH_x86)
422-
if (support_avx512() )
420+
#if !defined(COMPLEX) && !defined(DOUBLE) && !defined(BFLOAT16)
421+
#if defined(ARCH_x86) && (defined(USE_SGEMM_KERNEL_DIRECT)||defined(DYNAMIC_ARCH))
422+
#if defined(DYNAMIC_ARCH)
423+
if (support_avx512() )
424+
#endif
423425
if (beta == 0 && alpha == 1.0 && order == CblasRowMajor && TransA == CblasNoTrans && TransB == CblasNoTrans && SGEMM_DIRECT_PERFORMANT(m,n,k)) {
424426
SGEMM_DIRECT(m, n, k, a, lda, b, ldb, c, ldc);
425427
return;
426428
}
427429
#endif
428-
#if defined(DYNAMIC_ARCH) && defined(ARCH_ARM64)
429-
if (support_sme1()){
430+
#if defined(ARCH_ARM64) && (defined(USE_SGEMM_KERNEL_DIRECT)||defined(DYNAMIC_ARCH))
431+
#if defined(DYNAMIC_ARCH)
432+
if (support_sme1())
433+
#endif
430434
if (beta == 0 && alpha == 1.0 && order == CblasRowMajor && TransA == CblasNoTrans && TransB == CblasNoTrans) {
431435
SGEMM_DIRECT(m, n, k, a, lda, b, ldb, c, ldc);
432436
return;
433437
}
434-
}
435438
#endif
436439
#endif
437440

0 commit comments

Comments
 (0)