Skip to content

Commit 3a088de

Browse files
authored
Merge pull request #5228 from martin-frbg/cmakecrossarm
Update and amend parameters for Neoverse cpus in CMake crossbuilds
2 parents 0241d51 + d9369bd commit 3a088de

File tree

1 file changed

+27
-6
lines changed

1 file changed

+27
-6
lines changed

cmake/prebuild.cmake

Lines changed: 27 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1006,15 +1006,15 @@ endif ()
10061006
"#define HAVE_SVE\n"
10071007
"#define ARMV8\n")
10081008
set(SGEMM_UNROLL_M 16)
1009-
set(SGEMM_UNROLL_N 4)
1010-
set(DGEMM_UNROLL_M 8)
1011-
set(DGEMM_UNROLL_N 4)
1012-
set(CGEMM_UNROLL_M 8)
1009+
set(SGEMM_UNROLL_N 8)
1010+
set(DGEMM_UNROLL_M 4)
1011+
set(DGEMM_UNROLL_N 8)
1012+
set(CGEMM_UNROLL_M 2)
10131013
set(CGEMM_UNROLL_N 4)
1014-
set(ZGEMM_UNROLL_M 4)
1014+
set(ZGEMM_UNROLL_M 2)
10151015
set(ZGEMM_UNROLL_N 4)
10161016
set(SYMV_P 16)
1017-
elseif ("${TCORE}" STREQUAL "NEOVERSEN2" or "${TCORE}" STREQUAL "ARMV9SME")
1017+
elseif ("${TCORE}" STREQUAL "NEOVERSEN2" OR "${TCORE}" STREQUAL "ARMV9SME")
10181018
file(APPEND ${TARGET_CONF_TEMP}
10191019
"#define L1_CODE_SIZE\t65536\n"
10201020
"#define L1_CODE_LINESIZE\t64\n"
@@ -1249,6 +1249,25 @@ endif ()
12491249
set(ZGEMM_UNROLL_M 2)
12501250
set(ZGEMM_UNROLL_N 4)
12511251
set(SYMV_P 16)
1252+
elseif ("${TCORE}" STREQUAL "ARMV8SVE" OR "${TCORE}" STREQUAL "CORTEXA510" OR "${TCORE}" STREQUAL "CORTEXX2" OR "${TCORE}" STREQUAL "ARMV9")
1253+
file(APPEND ${TARGET_CONF_TEMP}
1254+
"#define L1_DATA_SIZE\t32768\n"
1255+
"#define L1_DATA_LINESIZE\t64\n"
1256+
"#define L2_SIZE\t262144\n"
1257+
"#define L2_LINESIZE\t64\n"
1258+
"#define DTB_DEFAULT_ENTRIES\t64\n"
1259+
"#define DTB_SIZE\t4096\n"
1260+
"#define L2_ASSOCIATIVE\t32\n"
1261+
"#define ARMV8\n")
1262+
set(SGEMM_UNROLL_M 4)
1263+
set(SGEMM_UNROLL_N 8)
1264+
set(DGEMM_UNROLL_M 4)
1265+
set(DGEMM_UNROLL_N 8)
1266+
set(CGEMM_UNROLL_M 2)
1267+
set(CGEMM_UNROLL_N 4)
1268+
set(ZGEMM_UNROLL_M 2)
1269+
set(ZGEMM_UNROLL_N 4)
1270+
set(SYMV_P 16)
12521271
elseif ("${TCORE}" STREQUAL "P5600")
12531272
file(APPEND ${TARGET_CONF_TEMP}
12541273
"#define L2_SIZE 1048576\n"
@@ -1409,9 +1428,11 @@ endif ()
14091428
# GetArch_2nd
14101429
foreach(float_char S;D;Q;C;Z;X)
14111430
if (NOT DEFINED ${float_char}GEMM_UNROLL_M)
1431+
message(STATUS "setting unrollm=2")
14121432
set(${float_char}GEMM_UNROLL_M 2)
14131433
endif()
14141434
if (NOT DEFINED ${float_char}GEMM_UNROLL_N)
1435+
message(STATUS "setting unrolln=2")
14151436
set(${float_char}GEMM_UNROLL_N 2)
14161437
endif()
14171438
endforeach()

0 commit comments

Comments
 (0)