forked from huawei/mindspore2022
47 lines
1.4 KiB
CMake
47 lines
1.4 KiB
CMake
if(NOT PLATFORM_ARM64 AND NOT PLATFORM_ARM32)
|
|
add_definitions(-DUSE_GLOG)
|
|
endif()
|
|
|
|
set(3RD_DIR ${TOP_DIR}/third_party)
|
|
set(LITE_DIR ${TOP_DIR}/mindspore/lite)
|
|
set(MICRO_DIR ${LITE_DIR}/micro)
|
|
|
|
if(MSLITE_ENABLE_CONVERTER)
|
|
set(CODEGEN_PATH ${CMAKE_BINARY_DIR}/micro/coder/codegen)
|
|
else()
|
|
set(CODEGEN_PATH ${CMAKE_BINARY_DIR}/coder/codegen)
|
|
endif()
|
|
|
|
#include 3rd
|
|
include_directories(${3RD_DIR})
|
|
include_directories(${3RD_DIR}/flatbuffers/include)
|
|
|
|
#include ms
|
|
include_directories(${TOP_DIR}/)
|
|
include_directories(${TOP_DIR}/mindspore/core/)
|
|
include_directories(${NNACL_DIR}/../)
|
|
include_directories(${LITE_DIR})
|
|
include_directories(${MICRO_DIR})
|
|
|
|
#include coder
|
|
if(NOT MSVC)
|
|
include(${TOP_DIR}/cmake/external_libs/cmsis.cmake)
|
|
include(${MICRO_DIR}/cmake/package_wrapper.cmake)
|
|
add_subdirectory(wrapper)
|
|
endif()
|
|
|
|
if(MSLITE_ENABLE_CONVERTER)
|
|
include(${MICRO_DIR}/cmake/file_list.cmake)
|
|
add_executable(codegen main.cc ${FILE_SET})
|
|
add_dependencies(codegen fbs_src)
|
|
add_dependencies(codegen fbs_inner_src)
|
|
target_link_libraries(codegen PRIVATE ${SECUREC_LIBRARY} mindspore::glog wrapper_mid nnacl_mid cpu_ops_mid)
|
|
if(ENABLE_MODEL_OBF)
|
|
target_link_libraries(codegen PRIVATE
|
|
${OBF_LIB_DIR}/libmsdeobfuscator-lite.so)
|
|
endif()
|
|
if(NOT WIN32 AND "${CMAKE_BUILD_TYPE}" STREQUAL "Release")
|
|
add_custom_command(TARGET codegen POST_BUILD COMMAND strip ${CODEGEN_PATH})
|
|
endif()
|
|
endif()
|