OpenBLAS/kernel/arm64/sgemm_direct_performant.c

16 lines
380 B
C

#include "common.h"
/* helper for the direct sgemm code adapted from Arjan van der Ven's x86_64 version */
int CNAME(BLASLONG M, BLASLONG N, BLASLONG K)
{
if (M<3) return 0;
unsigned long long mnk = M * N * K;
/* benchmark performance on M4 peaks around 512 and crosses the graph of the NEON SGEMM at about 3100 */
if (mnk >= 3100 * 3100 * 3100)
return 0;
return 1;
}