inference-engine/cmake/itt.cmake

   1 # Copyright (C) 2018-2019 Intel Corporation
   2 #
   3 # SPDX-License-Identifier: Apache-2.0
   4 #
   5
   6 unset(ITT_INCLUDE_DIR CACHE)
   7 unset(ITT_LIB CACHE)
   8
   9 if(NOT DEFINED INTEL_VTUNE_DIR AND DEFINED ENV{INTEL_VTUNE_DIR})
  10     set(INTEL_VTUNE_DIR "$ENV{INTEL_VTUNE_DIR}")
  11 endif()
  12 if(NOT DEFINED INTEL_VTUNE_DIR)
  13     if(EXISTS "/opt/intel/vtune_amplifier_xe/include")
  14         set(INTEL_VTUNE_DIR "/opt/intel/vtune_amplifier_xe")
  15     elseif(EXISTS "/opt/intel/vtune_amplifier/include")
  16         set(INTEL_VTUNE_DIR "/opt/intel/vtune_amplifier")
  17     elseif (EXISTS "C:/Program Files (x86)/IntelSWTools/VTune Amplifier XE")
  18         set(INTEL_VTUNE_DIR "C:/Program Files (x86)/IntelSWTools/VTune Amplifier XE")
  19     elseif (EXISTS "C:/Program Files (x86)/IntelSWTools/VTune Amplifier")
  20         set(INTEL_VTUNE_DIR "C:/Program Files (x86)/IntelSWTools/VTune Amplifier")
  21     endif()
  22 endif()
  23
  24 if(DEFINED INTEL_VTUNE_DIR)
  25     message(STATUS "INTEL_VTUNE_DIR = ${INTEL_VTUNE_DIR}")
  26
  27     find_path(ITT_INCLUDE_DIR
  28     FILES
  29         ittnotify.h
  30     PATHS "${INTEL_VTUNE_DIR}/include/")
  31
  32     find_library(ITT_LIB
  33     "libittnotify${CMAKE_STATIC_LIBRARY_SUFFIX}"
  34     PATHS ${INTEL_VTUNE_DIR}/lib64)
  35
  36     set(Located_ITT_LIBS ${ITT_LIB} ${CMAKE_DL_LIBS})
  37     set(Located_ITT_INCLUDE_DIRS ${ITT_INCLUDE_DIR})
  38 else()
  39     message(STATUS "INTEL_VTUNE_DIR is not defined")
  40 endif()
  41
  42 # Handle find_package() arguments, and set INTEL_ITT_FOUND
  43 include(FindPackageHandleStandardArgs)
  44 find_package_handle_standard_args(INTEL_ITT
  45     REQUIRED_VARS
  46         Located_ITT_INCLUDE_DIRS
  47         Located_ITT_LIBS)
  48
  49 if(ENABLE_PROFILING_ITT AND INTEL_ITT_FOUND)
  50     add_definitions(-DENABLE_PROFILING_ITT=1)
  51
  52     set(INTEL_ITT_LIBS ${Located_ITT_LIBS})
  53     set(INTEL_ITT_INCLUDE_DIRS ${Located_ITT_INCLUDE_DIRS})
  54
  55     message(STATUS "INTEL_ITT_INCLUDE_DIRS: ${INTEL_ITT_INCLUDE_DIRS}")
  56     include_directories(${INTEL_ITT_INCLUDE_DIRS})
  57     message(STATUS "INTEL_ITT_LIBS: ${INTEL_ITT_LIBS}")
  58 else()
  59     add_definitions(-DENABLE_PROFILING_ITT=0)
  60     message(STATUS "INTEL_ITT is disabled")
  61 endif()
  62