Skip to content

Commit d003ad6

Browse files
authoredOct 31, 2023
Increase the default GEMM buffer size on modern ARM server cpus
1 parent 9d425a5 commit d003ad6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed
 

‎common_arm64.h

+4
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,11 @@ static inline int blas_quickdivide(blasint x, blasint y){
162162
#define HUGE_PAGESIZE ( 4 << 20)
163163

164164
#ifndef BUFFERSIZE
165+
if defined(NEOVERSEN1) || defined(NEOVERSEN2) || defined(NEOVERSEV1) || defined(A64FX) || defined(ARMV8SVE)
166+
#define BUFFER_SIZE (32 << 22)
167+
else
165168
#define BUFFER_SIZE (32 << 20)
169+
#endif
166170
#else
167171
#define BUFFER_SIZE (32 << BUFFERSIZE)
168172
#endif

0 commit comments

Comments
 (0)
Please sign in to comment.