onnxruntime
34 строки · 1.8 Кб
1# Copyright (c) Microsoft Corporation. All rights reserved.
2# Licensed under the MIT License.
3
4set(PROVIDERS_SNPE onnxruntime_providers_snpe)
5add_compile_definitions(USE_SNPE=1)
6
7file(GLOB_RECURSE
8onnxruntime_providers_snpe_cc_srcs CONFIGURE_DEPENDS
9"${ONNXRUNTIME_ROOT}/core/providers/snpe/*.h"
10"${ONNXRUNTIME_ROOT}/core/providers/snpe/*.cc"
11)
12
13file(GLOB SNPE_SO_FILES LIST_DIRECTORIES false "${SNPE_CMAKE_DIR}/lib/${SNPE_ARCH_ABI}/*.so" "${SNPE_CMAKE_DIR}/lib/${SNPE_ARCH_ABI}/*.dll")
14# add dsp skel files to distribution
15file(GLOB SNPE_DSP_FILES LIST_DIRECTORIES false "${SNPE_CMAKE_DIR}/lib/dsp/*.so")
16list(APPEND SNPE_SO_FILES ${QCDK_FILES} ${SNPE_DSP_FILES})
17
18if(NOT SNPE OR NOT SNPE_SO_FILES)
19message(ERROR "Snpe not found in ${SNPE_CMAKE_DIR}/lib/${SNPE_ARCH_ABI} for platform ${CMAKE_GENERATOR_PLATFORM}")
20endif()
21
22source_group(TREE ${ONNXRUNTIME_ROOT}/core FILES ${onnxruntime_providers_snpe_cc_srcs})
23onnxruntime_add_static_library(onnxruntime_providers_snpe ${onnxruntime_providers_snpe_cc_srcs})
24onnxruntime_add_include_to_target(onnxruntime_providers_snpe onnxruntime_common onnxruntime_framework onnx onnx_proto protobuf::libprotobuf-lite flatbuffers::flatbuffers)
25link_directories(${SNPE_CMAKE_DIR}/lib/${SNPE_ARCH_ABI} ${SNPE_CMAKE_DIR}/lib/dsp)
26add_dependencies(onnxruntime_providers_snpe onnx ${onnxruntime_EXTERNAL_DEPENDENCIES})
27set_target_properties(onnxruntime_providers_snpe PROPERTIES CXX_STANDARD_REQUIRED ON)
28set_target_properties(onnxruntime_providers_snpe PROPERTIES FOLDER "ONNXRuntime")
29target_include_directories(onnxruntime_providers_snpe PRIVATE ${ONNXRUNTIME_ROOT} ${SNPE_ROOT}/include/zdl)
30set_target_properties(onnxruntime_providers_snpe PROPERTIES LINKER_LANGUAGE CXX)
31
32if(MSVC)
33target_compile_options(onnxruntime_providers_snpe PUBLIC /wd4244 /wd4505)
34endif()
35