From fd23aae40530df028b96854c6bd2163e2881e5af Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Sun, 13 Oct 2024 13:20:24 +0200 Subject: [PATCH] Update prebuild.cmake --- cmake/prebuild.cmake | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/cmake/prebuild.cmake b/cmake/prebuild.cmake index 0f096f67e3..1f6ee03826 100644 --- a/cmake/prebuild.cmake +++ b/cmake/prebuild.cmake @@ -571,6 +571,18 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS set(HAVE_MISALIGNSSE 1) set(HAVE_128BITFPU 1) set(HAVE_FASTMOVU 1) + if (NOT BINARY64) + set(SBGEMM_UNROLL_M 4) + set(SBGEMM_UNROLL_N 4) + set(SGEMM_UNROLL_M 4) + set(SGEMM_UNROLL_N 4) + set(DGEMM_UNROLL_M 2) + set(DGEMM_UNROLL_N 4) + set(CGEMM_UNROLL_M 2) + set(CGEMM_UNROLL_N 2) + set(ZGEMM_UNROLL_M 1) + set(ZGEMM_UNROLL_N 2) + else() set(SBGEMM_UNROLL_M 8) set(SBGEMM_UNROLL_N 4) set(SGEMM_UNROLL_M 8) @@ -581,6 +593,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)