|
1 | 1 | # The script detects Intel(R) Inference Engine installation |
2 | 2 | # |
3 | | -# Parameters: |
4 | | -# INTEL_CVSDK_DIR - Path to Inference Engine root folder |
5 | | -# IE_PLUGINS_PATH - Path to folder with Inference Engine plugins |
| 3 | +# Cache variables: |
| 4 | +# INF_ENGINE_OMP_DIR - directory with OpenMP library to link with (needed by some versions of IE) |
| 5 | +# INF_ENGINE_RELEASE - a number reflecting IE source interface (linked with OpenVINO release) |
6 | 6 | # |
7 | | -# On return this will define: |
| 7 | +# Detect parameters: |
| 8 | +# 1. Native cmake IE package: |
| 9 | +# - enironment variable InferenceEngine_DIR is set to location of cmake module |
| 10 | +# 2. Custom location: |
| 11 | +# - INF_ENGINE_INCLUDE_DIRS - headers search location |
| 12 | +# - INF_ENGINE_LIB_DIRS - library search location |
| 13 | +# 3. OpenVINO location: |
| 14 | +# - environment variable INTEL_CVSDK_DIR is set to location of OpenVINO installation dir |
| 15 | +# - INF_ENGINE_PLATFORM - part of name of library directory representing its platform (default ubuntu_16.04) |
8 | 16 | # |
9 | | -# HAVE_INF_ENGINE - True if Intel Inference Engine was found |
10 | | -# INF_ENGINE_INCLUDE_DIRS - Inference Engine include folder |
11 | | -# INF_ENGINE_LIBRARIES - Inference Engine libraries and it's dependencies |
| 17 | +# Result: |
| 18 | +# INF_ENGINE_TARGET - set to name of imported library target representing InferenceEngine |
12 | 19 | # |
13 | | -macro(ie_fail) |
14 | | - set(HAVE_INF_ENGINE FALSE) |
15 | | - return() |
16 | | -endmacro() |
17 | | - |
18 | 20 |
|
19 | | -find_package(InferenceEngine QUIET) |
20 | | -if(InferenceEngine_FOUND) |
21 | | - set(INF_ENGINE_LIBRARIES "${InferenceEngine_LIBRARIES}") |
22 | | - set(INF_ENGINE_INCLUDE_DIRS "${InferenceEngine_INCLUDE_DIRS}") |
23 | | - set(INF_ENGINE_VERSION "${InferenceEngine_VERSION}") |
24 | | - set(HAVE_INF_ENGINE TRUE) |
25 | | - return() |
| 21 | +if(NOT HAVE_CXX11) |
| 22 | + message(WARNING "DL Inference engine requires C++11. You can turn it on via ENABLE_CXX11=ON CMake flag.") |
| 23 | + return() |
26 | 24 | endif() |
27 | 25 |
|
28 | | -ocv_check_environment_variables(INTEL_CVSDK_DIR INF_ENGINE_ROOT_DIR IE_PLUGINS_PATH) |
| 26 | +# ======================= |
29 | 27 |
|
30 | | -if(NOT INF_ENGINE_ROOT_DIR OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}/include/inference_engine.hpp") |
31 | | - set(ie_root_paths "${INF_ENGINE_ROOT_DIR}") |
32 | | - if(DEFINED INTEL_CVSDK_DIR) |
33 | | - list(APPEND ie_root_paths "${INTEL_CVSDK_DIR}/") |
34 | | - list(APPEND ie_root_paths "${INTEL_CVSDK_DIR}/deployment_tools/inference_engine") |
35 | | - endif() |
| 28 | +function(add_custom_ie_build _inc _lib _lib_rel _lib_dbg _msg) |
| 29 | + if(NOT _inc OR NOT (_lib OR _lib_rel OR _lib_dbg)) |
| 30 | + return() |
| 31 | + endif() |
| 32 | + add_library(inference_engine UNKNOWN IMPORTED) |
| 33 | + set_target_properties(inference_engine PROPERTIES |
| 34 | + IMPORTED_LOCATION "${_lib}" |
| 35 | + IMPORTED_IMPLIB_RELEASE "${_lib_rel}" |
| 36 | + IMPORTED_IMPLIB_DEBUG "${_lib_dbg}" |
| 37 | + INTERFACE_INCLUDE_DIRECTORIES "${_inc}" |
| 38 | + ) |
| 39 | + find_library(omp_lib iomp5 PATHS "${INF_ENGINE_OMP_DIR}" NO_DEFAULT_PATH) |
| 40 | + if(NOT omp_lib) |
| 41 | + message(WARNING "OpenMP for IE have not been found. Set INF_ENGINE_OMP_DIR variable if you experience build errors.") |
| 42 | + else() |
| 43 | + set_target_properties(inference_engine PROPERTIES IMPORTED_LINK_INTERFACE_LIBRARIES "${omp_lib}") |
| 44 | + endif() |
| 45 | + set(INF_ENGINE_VERSION "Unknown" CACHE STRING "") |
| 46 | + set(INF_ENGINE_TARGET inference_engine PARENT_SCOPE) |
| 47 | + message(STATUS "Detected InferenceEngine: ${_msg}") |
| 48 | +endfunction() |
36 | 49 |
|
37 | | - if(NOT ie_root_paths) |
38 | | - list(APPEND ie_root_paths "/opt/intel/computer_vision_sdk/deployment_tools/inference_engine/") |
39 | | - endif() |
| 50 | +# ====================== |
40 | 51 |
|
41 | | - find_path(INF_ENGINE_ROOT_DIR include/inference_engine.hpp PATHS ${ie_root_paths}) |
42 | | - if(INF_ENGINE_ROOT_DIR MATCHES "-NOTFOUND$") |
43 | | - unset(INF_ENGINE_ROOT_DIR CACHE) |
44 | | - endif() |
| 52 | +find_package(InferenceEngine QUIET) |
| 53 | +if(InferenceEngine_FOUND) |
| 54 | + set(INF_ENGINE_TARGET IE::inference_engine) |
| 55 | + set(INF_ENGINE_VERSION "${InferenceEngine_VERSION}" CACHE STRING "") |
| 56 | + message(STATUS "Detected InferenceEngine: cmake package") |
45 | 57 | endif() |
46 | 58 |
|
47 | | -set(INF_ENGINE_INCLUDE_DIRS "${INF_ENGINE_ROOT_DIR}/include" CACHE PATH "Path to Inference Engine include directory") |
48 | | - |
49 | | -if(NOT INF_ENGINE_ROOT_DIR |
50 | | - OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}" |
51 | | - OR NOT EXISTS "${INF_ENGINE_ROOT_DIR}/include/inference_engine.hpp" |
52 | | -) |
53 | | - message(WARNING "DL IE: Can't detect INF_ENGINE_ROOT_DIR location.") |
54 | | - ie_fail() |
| 59 | +if(NOT INF_ENGINE_TARGET AND INF_ENGINE_LIB_DIRS AND INF_ENGINE_INCLUDE_DIRS) |
| 60 | + find_path(ie_custom_inc "inference_engine.hpp" PATHS "${INF_ENGINE_INCLUDE_DIRS}" NO_DEFAULT_PATH) |
| 61 | + find_library(ie_custom_lib "inference_engine" PATHS "${INF_ENGINE_LIB_DIRS}" NO_DEFAULT_PATH) |
| 62 | + find_library(ie_custom_lib_rel "inference_engine" PATHS "${INF_ENGINE_LIB_DIRS}/Release" NO_DEFAULT_PATH) |
| 63 | + find_library(ie_custom_lib_dbg "inference_engine" PATHS "${INF_ENGINE_LIB_DIRS}/Debug" NO_DEFAULT_PATH) |
| 64 | + add_custom_ie_build("${ie_custom_inc}" "${ie_custom_lib}" "${ie_custom_lib_rel}" "${ie_custom_lib_dbg}" "INF_ENGINE_{INCLUDE,LIB}_DIRS") |
55 | 65 | endif() |
56 | 66 |
|
57 | | -set(INF_ENGINE_LIBRARIES "") |
58 | | - |
59 | | -set(ie_lib_list inference_engine) |
60 | | - |
61 | | -if(NOT IS_ABSOLUTE "${IE_PLUGINS_PATH}") |
62 | | - set(IE_PLUGINS_PATH "${INF_ENGINE_ROOT_DIR}/${IE_PLUGINS_PATH}") |
| 67 | +set(_loc "$ENV{INTEL_CVSDK_DIR}") |
| 68 | +if(NOT INF_ENGINE_TARGET AND _loc) |
| 69 | + set(INF_ENGINE_PLATFORM "ubuntu_16.04" CACHE STRING "InferenceEngine platform (library dir)") |
| 70 | + find_path(ie_custom_env_inc "inference_engine.hpp" PATHS "${_loc}/deployment_tools/inference_engine/include" NO_DEFAULT_PATH) |
| 71 | + find_library(ie_custom_env_lib "inference_engine" PATHS "${_loc}/deployment_tools/inference_engine/lib/${INF_ENGINE_PLATFORM}/intel64" NO_DEFAULT_PATH) |
| 72 | + find_library(ie_custom_env_lib_rel "inference_engine" PATHS "${_loc}/deployment_tools/inference_engine/lib/intel64/Release" NO_DEFAULT_PATH) |
| 73 | + find_library(ie_custom_env_lib_dbg "inference_engine" PATHS "${_loc}/deployment_tools/inference_engine/lib/intel64/Debug" NO_DEFAULT_PATH) |
| 74 | + add_custom_ie_build("${ie_custom_env_inc}" "${ie_custom_env_lib}" "${ie_custom_env_lib_rel}" "${ie_custom_env_lib_dbg}" "OpenVINO (${_loc})") |
63 | 75 | endif() |
64 | 76 |
|
65 | | -link_directories( |
66 | | - ${INF_ENGINE_ROOT_DIR}/external/mkltiny_lnx/lib |
67 | | - ${INF_ENGINE_ROOT_DIR}/external/cldnn/lib |
68 | | -) |
69 | | - |
70 | | -foreach(lib ${ie_lib_list}) |
71 | | - find_library(${lib} NAMES ${lib} HINTS ${IE_PLUGINS_PATH}) |
72 | | - if(NOT ${lib}) |
73 | | - message(WARNING "DL IE: Can't find library: '${lib}'") |
74 | | - ie_fail() |
75 | | - endif() |
76 | | - list(APPEND INF_ENGINE_LIBRARIES ${${lib}}) |
77 | | -endforeach() |
| 77 | +# Add more features to the target |
78 | 78 |
|
79 | | -set(HAVE_INF_ENGINE TRUE) |
| 79 | +if(INF_ENGINE_TARGET) |
| 80 | + if(NOT INF_ENGINE_RELEASE) |
| 81 | + message(WARNING "InferenceEngine version have not been set, 2018R2 will be used by default. Set INF_ENGINE_RELEASE variable if you experience build errors.") |
| 82 | + endif() |
| 83 | + set(INF_ENGINE_RELEASE "2018020000" CACHE STRING "Force IE version, should be in form YYYYAABBCC (e.g. 2018R2.0.2 -> 2018020002)") |
| 84 | + set_target_properties(${INF_ENGINE_TARGET} PROPERTIES |
| 85 | + INTERFACE_COMPILE_DEFINITIONS "HAVE_INF_ENGINE=1;INF_ENGINE_RELEASE=${INF_ENGINE_RELEASE}" |
| 86 | + ) |
| 87 | +endif() |
0 commit comments