Skip to content

Commit e9637ac

Browse files
authored
try inlining the small_matrix_permit
1 parent 3060ebe commit e9637ac

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

interface/gemm.c

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -566,7 +566,11 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
566566

567567
#if USE_SMALL_MATRIX_OPT
568568
#if !defined(COMPLEX)
569+
#if defined(ARCH_ARM64)
570+
if(args.m*args.n*args.k<=64*64*64){
571+
#else
569572
if(GEMM_SMALL_MATRIX_PERMIT(transa, transb, args.m, args.n, args.k, *(FLOAT *)(args.alpha), *(FLOAT *)(args.beta))){
573+
#endif
570574
if(*(FLOAT *)(args.beta) == 0.0){
571575
(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);
572576
}else{

0 commit comments

Comments
 (0)