From e9637ac7af39d6ca537b6cb8427e1fd3a6eb8921 Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Tue, 10 Dec 2024 14:17:18 +0100 Subject: [PATCH] try inlining the small_matrix_permit --- interface/gemm.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/interface/gemm.c b/interface/gemm.c index 576e94593..4b290437f 100644 --- a/interface/gemm.c +++ b/interface/gemm.c @@ -566,7 +566,11 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS #if USE_SMALL_MATRIX_OPT #if !defined(COMPLEX) +#if defined(ARCH_ARM64) + if(args.m*args.n*args.k<=64*64*64){ +#else if(GEMM_SMALL_MATRIX_PERMIT(transa, transb, args.m, args.n, args.k, *(FLOAT *)(args.alpha), *(FLOAT *)(args.beta))){ +#endif if(*(FLOAT *)(args.beta) == 0.0){ (GEMM_SMALL_KERNEL_B0((transb << 2) | transa))(args.m, args.n, args.k, args.a, args.lda, *(FLOAT *)(args.alpha), args.b, args.ldb, args.c, args.ldc); }else{