inference-engine/tests/unit/CMakeLists.txt

   1 # Copyright (C) 2018-2019 Intel Corporation
   2 # SPDX-License-Identifier: Apache-2.0
   3 #
   4
   5 cmake_policy(SET CMP0054 NEW)
   6
   7 set(TARGET_NAME InferenceEngineUnitTests)
   8
   9 #rpath enabled for unit tests only
  10 SET (CMAKE_SKIP_RPATH OFF)
  11
  12 # collect sources
  13
  14 file(GLOB
  15         TEST_SRC
  16         graph_tools/*.cpp
  17         inference_engine_tests/*.cpp
  18         inference_engine_tests/cpp_interfaces/*.cpp
  19         mem_solver/*.cpp
  20         cnn_network/*.cpp
  21         builders/*.cpp
  22         transformations/*.cpp
  23         shape_infer/*.cpp
  24         shape_infer/built-in/*.cpp
  25         topology_verification_tests/*.cpp
  26         stress_tests/*.cpp
  27         )
  28
  29 if (ENABLE_GNA)
  30     file(GLOB
  31             GNA_TESTS
  32             engines/gna/*cpp
  33             )
  34     list(APPEND TEST_SRC ${GNA_TESTS})
  35     source_group("gna" FILES ${GNA_TESTS})
  36
  37     find_package(libGNA)
  38     include_directories(${libGNA_INCLUDE_DIRS})
  39
  40     set (GNA_TEST_ENGINE GNAPlugin_test_static)
  41 endif()
  42
  43 if (ENABLE_MKL_DNN)
  44     if (GEMM STREQUAL "MKL")
  45         add_definitions(-DUSE_MKL)
  46     endif ()
  47     file(GLOB
  48             MKLDNN_TESTS
  49             engines/mkldnn/*.cpp
  50             engines/mkldnn/graph/layers/extensions/*.cpp
  51             engines/mkldnn/graph/layers/internal/*.cpp
  52             engines/mkldnn/graph/structure/*.cpp
  53             engines/mkldnn/graph/*.cpp)
  54     file(GLOB
  55             MKLDNN_TESTS_INCLUDE engines/mkldnn/graph/*.hpp)
  56
  57     include_directories(
  58             ${IE_MAIN_SOURCE_DIR}/thirdparty/mkl-dnn/include
  59             engines/mkldnn/graph
  60             ${CMAKE_BINARY_DIR}/include/)
  61
  62     source_group("mkldnn" FILES ${MKLDNN_TESTS} ${MKLDNN_TESTS_INCLUDE})
  63 endif ()
  64
  65 file(GLOB
  66         TEST_INCLUDE
  67         shape_infer/*.hpp)
  68
  69 source_group("src" FILES ${TEST_SRC})
  70 source_group("include" FILES ${TEST_INCLUDE})
  71
  72 # create target
  73
  74 add_executable(${TARGET_NAME} ${TEST_SRC} ${TEST_INCLUDE} ${MKLDNN_TESTS} ${MKLDNN_TESTS_INCLUDE} ${DLAI_TESTS} transformations/sub_test.cpp transformations/tranformations_test.hpp)
  75 set_ie_threading_interface_for(${TARGET_NAME})
  76
  77 target_include_directories(${TARGET_NAME} PRIVATE
  78         ${IE_MAIN_SOURCE_DIR}/src/mkldnn_plugin
  79         ${IE_MAIN_SOURCE_DIR}/src/gna_plugin
  80         ${IE_MAIN_SOURCE_DIR}/src/inference_engine
  81         ${IE_MAIN_SOURCE_DIR}/src/extension
  82         ${IE_MAIN_SOURCE_DIR}/src/extension/common
  83         "${CMAKE_CURRENT_SOURCE_DIR}/mocks")
  84
  85 set_target_properties(${TARGET_NAME} PROPERTIES COMPILE_PDB_NAME ${TARGET_NAME})
  86
  87 ## Mock macros doesn't use "override" specificator
  88 target_compile_options(${TARGET_NAME} PRIVATE $<$<CXX_COMPILER_ID:Clang>: -Wno-inconsistent-missing-override >)
  89 target_compile_options(${TARGET_NAME} PRIVATE $<$<CXX_COMPILER_ID:AppleClang>: -Wno-inconsistent-missing-override >)
  90
  91 target_link_libraries(${TARGET_NAME} PRIVATE
  92     gtest
  93     gtest_main
  94     gmock
  95     gflags
  96     inference_engine_s
  97     helpers
  98     ${CMAKE_DL_LIBS}
  99     ${GNA_TEST_ENGINE})
 100
 101 add_dependencies(${TARGET_NAME} ie_cpu_extension)
 102
 103 if (ENABLE_MKL_DNN)
 104     target_link_libraries(${TARGET_NAME} PRIVATE
 105             test_MKLDNNPlugin
 106             mkldnn)
 107 endif ()
 108
 109 add_test(NAME ${TARGET_NAME}
 110         COMMAND ${TARGET_NAME})
 111
 112 add_dependencies(${TARGET_NAME} mock_engine)
 113
 114 # GAPI unit tests
 115 add_subdirectory(opencv_test_gapi)