Update prebuild.cmake
This commit is contained in:
parent
491e5d9646
commit
38b70e156b
|
@ -167,6 +167,16 @@ if (DEFINED CORE AND CMAKE_CROSSCOMPILING AND NOT (${HOST_OS} STREQUAL "WINDOWSS
|
||||||
set(HAVE_SSE2 1)
|
set(HAVE_SSE2 1)
|
||||||
set(HAVE_SSE3 1)
|
set(HAVE_SSE3 1)
|
||||||
set(HAVE_SSSE3 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_M 8)
|
||||||
set(SBGEMM_UNROLL_N 4)
|
set(SBGEMM_UNROLL_N 4)
|
||||||
set(SGEMM_UNROLL_M 8)
|
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(CGEMM_UNROLL_N 2)
|
||||||
set(ZGEMM_UNROLL_M 2)
|
set(ZGEMM_UNROLL_M 2)
|
||||||
set(ZGEMM_UNROLL_N 1)
|
set(ZGEMM_UNROLL_N 1)
|
||||||
|
endif()
|
||||||
set(CGEMM3M_UNROLL_M 8)
|
set(CGEMM3M_UNROLL_M 8)
|
||||||
set(CGEMM3M_UNROLL_N 4)
|
set(CGEMM3M_UNROLL_N 4)
|
||||||
set(ZGEMM3M_UNROLL_M 4)
|
set(ZGEMM3M_UNROLL_M 4)
|
||||||
|
|
Loading…
Reference in New Issue