From 6680e0592f9f2e4e0551b13d7c1f6fc3e225fe95 Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Sat, 17 May 2025 05:12:15 -0700 Subject: [PATCH] Fix conditional inclusion of SGEMM_KERNEL_DIRECT --- interface/gemm.c | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/interface/gemm.c b/interface/gemm.c index d369256291..54e5604fd3 100644 --- a/interface/gemm.c +++ b/interface/gemm.c @@ -417,21 +417,24 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS PRINT_DEBUG_CNAME; -#if !defined(COMPLEX) && !defined(DOUBLE) && !defined(BFLOAT16) && defined(USE_SGEMM_KERNEL_DIRECT) -#if defined(DYNAMIC_ARCH) && defined(ARCH_x86) - if (support_avx512() ) +#if !defined(COMPLEX) && !defined(DOUBLE) && !defined(BFLOAT16) +#if defined(ARCH_x86) && (defined(USE_SGEMM_KERNEL_DIRECT)||defined(DYNAMIC_ARCH)) +#if defined(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); return; } #endif -#if defined(DYNAMIC_ARCH) && defined(ARCH_ARM64) - if (support_sme1()){ +#if defined(ARCH_ARM64) && (defined(USE_SGEMM_KERNEL_DIRECT)||defined(DYNAMIC_ARCH)) +#if defined(DYNAMIC_ARCH) + if (support_sme1()) +#endif if (beta == 0 && alpha == 1.0 && order == CblasRowMajor && TransA == CblasNoTrans && TransB == CblasNoTrans) { SGEMM_DIRECT(m, n, k, a, lda, b, ldb, c, ldc); return; } - } #endif #endif