tests/tools/tflite_run/CMakeLists.txt

   1 if(NOT BUILD_TFLITE_RUN)
   2   return()
   3 endif(NOT BUILD_TFLITE_RUN)
   4
   5 list(APPEND TFLITE_RUN_SRCS "src/tflite_run.cc")
   6 list(APPEND TFLITE_RUN_SRCS "src/args.cc")
   7 list(APPEND TFLITE_RUN_SRCS "src/tensor_dumper.cc")
   8 list(APPEND TFLITE_RUN_SRCS "src/tensor_loader.cc")
   9
  10 nnfw_find_package(Boost REQUIRED program_options)
  11
  12 add_executable(tflite_run ${TFLITE_RUN_SRCS})
  13 target_include_directories(tflite_run PRIVATE src)
  14 target_include_directories(tflite_run PRIVATE ${Boost_INCLUDE_DIRS})
  15
  16 target_link_libraries(tflite_run tensorflow-lite ${LIB_PTHREAD} dl nnfw_lib_tflite)
  17 target_link_libraries(tflite_run ${Boost_PROGRAM_OPTIONS_LIBRARY})
  18
  19 target_link_libraries(tflite_run nnfw_lib_benchmark)
  20
  21 install(TARGETS tflite_run DESTINATION bin)
  22
  23 # TEST BUILD
  24 nnfw_find_package(GTest)
  25
  26 if(NOT GTest_FOUND)
  27   return()
  28 endif(NOT GTest_FOUND)
  29
  30 ## Add test cpp file
  31 add_executable(tflite_test src/tflite_test.cc)
  32 ## Link test executable against gtest & gtest_main
  33 target_link_libraries(tflite_test gtest gtest_main ${LIB_PTHREAD})
  34 ## install test binary for packaging
  35 install(TARGETS tflite_test DESTINATION unittest_standalone)