enhance: jemalloc aarch64 platform use 64k pagesize. (#31114)

pr: https://github.com/milvus-io/milvus/pull/29522
enhance: jemalloc aarch64 platform use 64k pagesize.

Signed-off-by: ping.liu <ping.liu@zilliz.com>
This commit is contained in:
pingliu 2024-03-11 12:03:02 +08:00 committed by GitHub
parent 3c90475d55
commit 1dd4f4b4dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -27,6 +27,13 @@ message(STATUS "Building (vendored) jemalloc from source")
# installations.
# find_package(jemalloc)
include(CheckSymbolExists)
macro(detect_aarch64_target_arch)
check_symbol_exists(__aarch64__ "" __AARCH64)
endmacro()
detect_aarch64_target_arch()
set(JEMALLOC_PREFIX "${CMAKE_INSTALL_PREFIX}")
set(JEMALLOC_LIB_DIR "${JEMALLOC_PREFIX}/lib")
set(JEMALLOC_STATIC_LIB "${JEMALLOC_LIB_DIR}/libjemalloc_pic${CMAKE_STATIC_LIBRARY_SUFFIX}")
@ -37,10 +44,9 @@ if (CMAKE_OSX_SYSROOT)
list(APPEND JEMALLOC_CONFIGURE_COMMAND "SDKROOT=${CMAKE_OSX_SYSROOT}")
endif ()
if (DEFINED MILVUS_JEMALLOC_LG_PAGE)
# Used for arm64 manylinux wheels in order to make the wheel work on both
# 4k and 64k page arm64 systems.
list(APPEND JEMALLOC_CONFIGURE_COMMAND "--with-lg-page=${MILVUS_JEMALLOC_LG_PAGE}")
if (DEFINED __AARCH64)
#aarch64 platform use 64k pagesize.
list(APPEND JEMALLOC_CONFIGURE_COMMAND "--with-lg-page=16")
endif ()
list(APPEND