Browse Source

Add sgemm_direct_performant for ARM64

pull/5423/head
Martin Kroeker GitHub 5 months ago
parent
commit
e76c39099a
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      common_param.h

+ 1
- 0
common_param.h View File

@@ -257,6 +257,7 @@ int (*shgemm_otcopy )(BLASLONG, BLASLONG, hfloat16 *, BLASLONG, hfloat16 *);
#ifdef ARCH_ARM64
void (*sgemm_direct) (BLASLONG, BLASLONG, BLASLONG, float *, BLASLONG , float *, BLASLONG , float * , BLASLONG);
void (*sgemm_direct_alpha_beta) (BLASLONG, BLASLONG, BLASLONG, float, float *, BLASLONG, float *, BLASLONG, float, float * , BLASLONG);
int (*sgemm_direct_performant) (BLASLONG M, BLASLONG N, BLASLONG K);
#endif



Loading…
Cancel
Save