Browse Source

try inlining the small_matrix_permit

pull/5012/head
Martin Kroeker GitHub 1 year ago
parent
commit
e9637ac7af
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      interface/gemm.c

+ 4
- 0
interface/gemm.c View File

@@ -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{


Loading…
Cancel
Save