alibaba--mnn
wehub-resource-sync
44 行
1.8 KiB
CMake
44 行
1.8 KiB
CMake
option(MNN_QNN_CONVERT_MODE "Enable the Convert mode of the QNN backend." OFF)
|
|
|
|
file(GLOB BACKEND_SRCS ${CMAKE_CURRENT_LIST_DIR}/backend/*)
|
|
file(GLOB EXECUTION_SRCS ${CMAKE_CURRENT_LIST_DIR}/execution/*)
|
|
set(MNN_QNN_SRCS ${BACKEND_SRCS} ${EXECUTION_SRCS})
|
|
|
|
# QNNBackend.cpp 中使用了 flatbuffers/flexbuffers，需要 flatbuffers util.cpp 中提供的
|
|
# ClassicLocale::instance_ 静态成员定义。MSVC 链接时不会自动从其它 target 获取，因此
|
|
# 显式将 util.cpp 一并编译进 MNN_QNN OBJECT 库。
|
|
if (MSVC OR WIN32)
|
|
list(APPEND MNN_QNN_SRCS ${CMAKE_SOURCE_DIR}/3rd_party/flatbuffers/src/util.cpp)
|
|
endif()
|
|
|
|
# Prefer CMake variable QNN_SDK_ROOT (passed via -DQNN_SDK_ROOT=...), fallback to environment
|
|
set(_QNN_ROOT "$ENV{QNN_SDK_ROOT}")
|
|
if (DEFINED QNN_SDK_ROOT AND NOT "${QNN_SDK_ROOT}" STREQUAL "")
|
|
set(_QNN_ROOT "${QNN_SDK_ROOT}")
|
|
endif()
|
|
message(STATUS "QNN Root: ${_QNN_ROOT}")
|
|
|
|
if (MNN_QNN_CONVERT_MODE)
|
|
file(GLOB CONVERTOR_SRCS ${CMAKE_CURRENT_LIST_DIR}/convertor/*.cpp)
|
|
list(APPEND MNN_QNN_SRCS ${CONVERTOR_SRCS})
|
|
endif()
|
|
|
|
add_library(MNN_QNN OBJECT ${MNN_QNN_SRCS})
|
|
|
|
if (MNN_ARM82)
|
|
target_compile_options(MNN_QNN PRIVATE -DENABLE_ARMV82)
|
|
endif()
|
|
|
|
if (MNN_QNN_CONVERT_MODE)
|
|
target_compile_options(MNN_QNN PRIVATE -DENABLE_QNN_CONVERT_MODE)
|
|
endif()
|
|
|
|
if (MNN_QNN_ONLINE_FINALIZE)
|
|
target_compile_options(MNN_QNN PRIVATE -DENABLE_QNN_ONLINE_FINALIZE)
|
|
endif()
|
|
|
|
target_include_directories(MNN_QNN PRIVATE ${CMAKE_CURRENT_LIST_DIR}/backend/)
|
|
target_include_directories(MNN_QNN PRIVATE ${CMAKE_CURRENT_LIST_DIR}/convertor/)
|
|
target_include_directories(MNN_QNN PRIVATE ${_QNN_ROOT}/include/QNN/)
|
|
# 在 MSVC / Windows 上 __fp16 不可用，需要使用 3rd_party/half 提供软件 fp16 转换。
|
|
target_include_directories(MNN_QNN PRIVATE ${CMAKE_SOURCE_DIR}/3rd_party/half) |