Skip to content

Commit 921ec4e

Browse files
authored
Adjust A53 SGEMM parameters to reflect move to 8x8 kernel
1 parent d6b6e5c commit 921ec4e

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

cmake/prebuild.cmake

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,8 +195,13 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS
195195
"#define HAVE_VFP\n"
196196
"#define HAVE_NEON\n"
197197
"#define ARMV8\n")
198+
if ("${TCORE}" STREQUAL "CORTEXA57")
198199
set(SGEMM_UNROLL_M 16)
199200
set(SGEMM_UNROLL_N 4)
201+
else
202+
set(SGEMM_UNROLL_M 8)
203+
set(SGEMM_UNROLL_N 8)
204+
endif
200205
set(DGEMM_UNROLL_M 8)
201206
set(DGEMM_UNROLL_N 4)
202207
set(CGEMM_UNROLL_M 8)

0 commit comments

Comments
 (0)