Adjust A53 SGEMM parameters to reflect move to 8x8 kernel

This commit is contained in:
Martin Kroeker 2020-07-27 19:54:46 +00:00 committed by GitHub
parent d6b6e5ccd7
commit 921ec4e9e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions

View File

@ -195,8 +195,13 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS
"#define HAVE_VFP\n"
"#define HAVE_NEON\n"
"#define ARMV8\n")
if ("${TCORE}" STREQUAL "CORTEXA57")
set(SGEMM_UNROLL_M 16)
set(SGEMM_UNROLL_N 4)
else
set(SGEMM_UNROLL_M 8)
set(SGEMM_UNROLL_N 8)
endif
set(DGEMM_UNROLL_M 8)
set(DGEMM_UNROLL_N 4)
set(CGEMM_UNROLL_M 8)