• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1set(3RD_DIR ${TOP_DIR}/third_party)
2set(LITE_DIR ${TOP_DIR}/mindspore/lite)
3set(MICRO_DIR ${LITE_DIR}/micro)
4
5if(MSLITE_ENABLE_CONVERTER)
6    set(CODEGEN_PATH ${CMAKE_BINARY_DIR}/micro/coder/codegen)
7else()
8    set(CODEGEN_PATH ${CMAKE_BINARY_DIR}/coder/codegen)
9endif()
10
11#include 3rd
12include_directories(${3RD_DIR})
13include_directories(${3RD_DIR}/flatbuffers/include)
14
15#include ms
16include_directories(${TOP_DIR}/)
17include_directories(${TOP_DIR}/mindspore/core/)
18include_directories(${NNACL_DIR}/../)
19include_directories(${LITE_DIR})
20include_directories(${MICRO_DIR})
21
22#include coder
23if(NOT MSVC)
24    include(${TOP_DIR}/cmake/external_libs/cmsis.cmake)
25    include(${MICRO_DIR}/cmake/package_wrapper.cmake)
26    add_subdirectory(wrapper)
27endif()
28
29if(MSLITE_ENABLE_CONVERTER)
30    include(${MICRO_DIR}/cmake/file_list.cmake)
31    set_property(SOURCE ${FILE_SET} PROPERTY COMPILE_OPTIONS -Wno-error=stringop-overflow=)
32    add_executable(codegen main.cc ${FILE_SET})
33    add_dependencies(codegen fbs_src)
34    add_dependencies(codegen fbs_inner_src)
35    target_link_libraries(codegen PRIVATE ${SECUREC_LIBRARY} wrapper_mid nnacl_mid cpu_ops_mid)
36    if(ENABLE_MODEL_OBF)
37        target_link_libraries(codegen PRIVATE
38                ${OBF_LIB_DIR}/libmsdeobfuscator-lite.so)
39    endif()
40endif()
41