From 38b70e156b3807e0fc857aa24e752328678018b9 Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Sat, 12 Oct 2024 21:16:39 +0200 Subject: [PATCH] Update prebuild.cmake --- cmake/prebuild.cmake | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/cmake/prebuild.cmake b/cmake/prebuild.cmake index 53a78d782..b6cab02aa 100644 --- a/cmake/prebuild.cmake +++ b/cmake/prebuild.cmake @@ -167,6 +167,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) + ifeq (${BINARY},32) + 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) @@ -177,6 +187,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 1) + endif() set(CGEMM3M_UNROLL_M 8) set(CGEMM3M_UNROLL_N 4) set(ZGEMM3M_UNROLL_M 4)