Merge pull request #4279 from martin-frbg/issue4269

Increase the default GEMM buffer size on modern ARM server cpus
OpenMathLib · Oct 31, 2023 · 9f7c35a · 9f7c35a
2 parents 9d425a5 + 728788f
commit 9f7c35a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/common_arm64.h b/common_arm64.h
@@ -162,7 +162,11 @@ static inline int blas_quickdivide(blasint x, blasint y){
 #define HUGE_PAGESIZE   ( 4 << 20)
 
 #ifndef BUFFERSIZE
+#if defined(NEOVERSEN1) || defined(NEOVERSEN2) || defined(NEOVERSEV1) || defined(A64FX) || defined(ARMV8SVE)
+#define BUFFER_SIZE     (32 << 22)
+#else
 #define BUFFER_SIZE     (32 << 20)
+#endif
 #else
 #define BUFFER_SIZE	(32 << BUFFERSIZE)
 #endif