try inlining the small_matrix_permit

martin-frbg · Dec 10, 2024 · e9637ac · e9637ac
1 parent 3060ebe
commit e9637ac
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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{