diff --git a/cmake/prebuild.cmake b/cmake/prebuild.cmake index ee6f12f47..0f096f67e 100644 --- a/cmake/prebuild.cmake +++ b/cmake/prebuild.cmake @@ -131,6 +131,16 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS set(HAVE_SSE2 1) set(HAVE_SSE3 1) set(HAVE_SSSE3 1) + if (NOT BINARY64) + set(SGEMM_UNROLL_M 8) + set(SGEMM_UNROLL_N 2) + set(DGEMM_UNROLL_M 4) + set(DGEMM_UNROLL_N 2) + set(CGEMM_UNROLL_M 4) + set(CGEMM_UNROLL_N 1) + set(ZGEMM_UNROLL_M 2) + set(ZGEMM_UNROLL_N 1) + else() set(SBGEMM_UNROLL_M 8) set(SBGEMM_UNROLL_N 4) set(SGEMM_UNROLL_M 8) @@ -141,6 +151,7 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS set(CGEMM_UNROLL_N 2) set(ZGEMM_UNROLL_M 2) set(ZGEMM_UNROLL_N 2) + endif() set(CGEMM3M_UNROLL_M 8) set(CGEMM3M_UNROLL_N 4) set(ZGEMM3M_UNROLL_M 4) @@ -167,16 +178,6 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS set(HAVE_SSE2 1) set(HAVE_SSE3 1) set(HAVE_SSSE3 1) - if (NOT BINARY64) - set(SGEMM_UNROLL_M 8) - set(SGEMM_UNROLL_N 2) - set(DGEMM_UNROLL_M 4) - set(DGEMM_UNROLL_N 2) - set(CGEMM_UNROLL_M 2) - set(CGEMM_UNROLL_N 1) - set(ZGEMM_UNROLL_M 1) - set(ZGEMM_UNROLL_N 1) - else() set(SBGEMM_UNROLL_M 8) set(SBGEMM_UNROLL_N 4) set(SGEMM_UNROLL_M 8) @@ -187,7 +188,6 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS set(CGEMM_UNROLL_N 2) set(ZGEMM_UNROLL_M 2) set(ZGEMM_UNROLL_N 1) - endif() set(CGEMM3M_UNROLL_M 8) set(CGEMM3M_UNROLL_N 4) set(ZGEMM3M_UNROLL_M 4)