Adjust A53 SGEMM parameters to reflect move to 8x8 kernel

2020-07-27 19:54:46 +00:00
parent d6b6e5ccd7
commit 921ec4e9e2
1 changed files with 5 additions and 0 deletions
--- a/cmake/prebuild.cmake
+++ b/cmake/prebuild.cmake
@@ -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)