diff --git a/internal/core/thirdparty/jemalloc/CMakeLists.txt b/internal/core/thirdparty/jemalloc/CMakeLists.txt index 73d369adb1..f81edc8391 100644 --- a/internal/core/thirdparty/jemalloc/CMakeLists.txt +++ b/internal/core/thirdparty/jemalloc/CMakeLists.txt @@ -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