wehub-resource-sync
48 lines
1.7 KiB
CMake
48 lines
1.7 KiB
CMake
# CPU
|
|
|
|
if(MNN_SUPPORT_RENDER)
|
|
FILE(GLOB MNN_CPU_SRC ${CMAKE_CURRENT_LIST_DIR}/*.cpp ${CMAKE_CURRENT_LIST_DIR}/*.h ${CMAKE_CURRENT_LIST_DIR}/*.hpp ${CMAKE_CURRENT_LIST_DIR}/compute/* ${CMAKE_CURRENT_LIST_DIR}/render/*)
|
|
else()
|
|
FILE(GLOB MNN_CPU_SRC ${CMAKE_CURRENT_LIST_DIR}/*.cpp ${CMAKE_CURRENT_LIST_DIR}/*.h ${CMAKE_CURRENT_LIST_DIR}/*.hpp ${CMAKE_CURRENT_LIST_DIR}/compute/*)
|
|
endif()
|
|
add_library(MNNCPU OBJECT ${MNN_CPU_SRC})
|
|
if (MNN_PIPELINE_PROFILE)
|
|
target_compile_options(MNNCPU PRIVATE -DMNN_PIPELINE_PROFILE)
|
|
endif()
|
|
if (MNN_SUPPORT_BF16)
|
|
include(${CMAKE_CURRENT_LIST_DIR}/bf16/CMakeLists.txt)
|
|
list(APPEND MNN_TARGETS MNN_BF16)
|
|
list(APPEND MNN_OBJECTS_TO_LINK $<TARGET_OBJECTS:MNN_BF16>)
|
|
target_compile_options(MNNCPU PRIVATE -DMNN_SUPPORT_BF16)
|
|
endif()
|
|
list(APPEND MNN_OBJECTS_TO_LINK $<TARGET_OBJECTS:MNNCPU>)
|
|
list(APPEND MNN_TARGETS MNNCPU)
|
|
|
|
|
|
if(MNN_USE_SPARSE_COMPUTE)
|
|
target_compile_options(MNNCPU PRIVATE -DMNN_USE_SPARSE_COMPUTE)
|
|
endif()
|
|
|
|
# X86_64 AVX/SSE
|
|
if (MNN_USE_SSE)
|
|
include(${CMAKE_CURRENT_LIST_DIR}/x86_x64/CMakeLists.txt)
|
|
endif()
|
|
|
|
# AArch32/64 Assemblies
|
|
include(${CMAKE_CURRENT_LIST_DIR}/arm/CMakeLists.txt)
|
|
if(MNN_USE_RVV)
|
|
include(${CMAKE_CURRENT_LIST_DIR}/riscv/CMakeLists.txt)
|
|
endif()
|
|
IF(NOT DEFINED IOS_ARCH)
|
|
set(IOS_ARCH "")
|
|
ENDIF()
|
|
|
|
# ARM82 Assemblies
|
|
IF(MNN_ARM82)
|
|
IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv7" OR ARCHS MATCHES "^armv7(;armv7s)?" OR CMAKE_SYSTEM_PROCESSOR MATCHES "^aarch64" OR ARCHS STREQUAL "arm64" OR ARCHS STREQUAL "ARM64")
|
|
target_compile_options(MNNCPU PRIVATE -DENABLE_ARMV82)
|
|
include(${CMAKE_CURRENT_LIST_DIR}/../arm82/CMakeLists.txt)
|
|
list(APPEND MNN_TARGETS MNN_Arm82)
|
|
list(APPEND MNN_OBJECTS_TO_LINK $<TARGET_OBJECTS:MNN_Arm82>)
|
|
ENDIF()
|
|
ENDIF() |