Do not add add MM_MALLOC on arm

uav4geo · Mar 21, 2023 · ed93d06 · ed93d06
1 parent 33ad2f6
commit ed93d06
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -59,7 +59,10 @@ if (WITH_GBT)
  if (WIN32)
  set(LIGHTGBM_LIB_PATH "${LIGHTGBM_PREFIX}/lib/lib_lightgbm.lib")
  endif()
+
+ if( NOT ((${CMAKE_SYSTEM_PROCESSOR} STREQUAL "aarch64") OR (${CMAKE_SYSTEM_PROCESSOR} STREQUAL "arm64")) )
  add_definitions(-DMM_MALLOC)
+ endif()
 
  if (NOT EXISTS ${LIGHTGBM_INCLUDE_DIRECTORY})
  file(MAKE_DIRECTORY ${LIGHTGBM_INCLUDE_DIRECTORY})