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

enhance: jemalloc aarch64 platform use 64k pagesize.
issue: #28843

Signed-off-by: Yusheng.Ma <Yusheng.Ma@zilliz.com>
pull/30998/head
presburger 2024-03-07 21:01:01 +08:00 committed by GitHub
parent c411cb4a49
commit 19c64067af
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 4 deletions

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