From fa912ce852b57807c510a1fde5d08ae80c0af788 Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Wed, 8 Oct 2025 11:11:52 +0200 Subject: [PATCH] rework definitions of ?FLOAT16_GEMM_GEMV_FORWARD --- interface/gemm.c | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/interface/gemm.c b/interface/gemm.c index 28f962ad87..a16df81ec4 100644 --- a/interface/gemm.c +++ b/interface/gemm.c @@ -587,8 +587,16 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS args.m, args.n, args.k, args.lda, args.ldb, args.ldc); #endif -#define BFLOAT16_GEMM_GEMV_FORWARD (!defined(BFLOAT16) || (!defined(BGEMM) && defined(SBGEMM_GEMV_FORWARD)) || (defined(BGEMM) && defined(BGEMM_GEMV_FORWARD))) -#define HFLOAT16_GEMM_GEMV_FORWARD (!defined(HFLOAT16) || (!defined(HGEMM) && defined(SHGEMM_GEMV_FORWARD)) || (defined(HGEMM) && defined(HGEMM_GEMV_FORWARD))) +#if (!defined(BFLOAT16) || (!defined(BGEMM) && defined(SBGEMM_GEMV_FORWARD)) || (defined(BGEMM) && defined(BGEMM_GEMV_FORWARD))) +#define BFLOAT16_GEMM_GEMV_FORWARD 1 +#else +#define BFLOAT16_GEMM_GEMV_FORWARD 0 +#endif +#if (!defined(HFLOAT16) || (!defined(HGEMM) && defined(SHGEMM_GEMV_FORWARD)) || (defined(HGEMM) && defined(HGEMM_GEMV_FORWARD))) +#define HFLOAT16_GEMM_GEMV_FORWARD 1 +#else +#define HFLOAT16_GEMM_GEMV_FORWARD 0 +#endif #if defined(GEMM_GEMV_FORWARD) && !defined(GEMM3M) && !defined(COMPLEX) && HFLOAT16_GEMM_GEMV_FORWARD && BFLOAT16_GEMM_GEMV_FORWARD #if defined(ARCH_ARM64)