intel-extension-for-pytorch

1
# FindIPEX
2
# -------
3
#
4
# Finds the Torch IPEX library
5
#
6
# This will define the following variables:
7
#
8
#   IPEX_FOUND              -- True if the system has the Torch IPEX library
9
#   IPEX_INCLUDE_DIRS       -- The include directories for torch
10
#   IPEX_LIBRARIES          -- Libraries to link against
11
#   IPEX_CXX_FLAGS          -- Additional (required) compiler flags
12
#
13
#   TORCH_IPEX_INCLUDE_DIRS -- TORCH_INCLUDE_DIRS + IPEX_INCLUDE_DIRS
14
#   TORCH_IPEX_LIBRARIES    -- TORCH_LIBRARIES + IPEX_LIBRARIES
15
#   TORCH_IPEX_CXX_FLAGS    -- TORCH_CXX_FLAGS + IPEX_CXX_FLAGS
16
#
17
# and the following imported targets:
18
#
19
#   intel-ext-pt-cpu
20
#   intel-ext-pt-gpu
21

22
include(FindPackageHandleStandardArgs)
23

24
if(DEFINED ENV{IPEX_INSTALL_PREFIX})
25
  set(IPEX_INSTALL_PREFIX $ENV{IPEX_INSTALL_PREFIX})
26
else()
27
  # Assume we are in <install-prefix>/share/cmake/IPEX/IPEXConfig.cmake
28
  get_filename_component(CMAKE_CURRENT_LIST_DIR "${CMAKE_CURRENT_LIST_FILE}" PATH)
29
  get_filename_component(IPEX_INSTALL_PREFIX "${CMAKE_CURRENT_LIST_DIR}/../../../" ABSOLUTE)
30
endif()
31

32
# Include directories.
33
if(EXISTS "${IPEX_INSTALL_PREFIX}/include")
34
  list(APPEND IPEX_INCLUDE_DIRS ${IPEX_INSTALL_PREFIX}/include)
35
endif()
36

37
# Library dependencies.
38
if(@BUILD_WITH_CPU@)
39
  find_library(IPEX_CPU_CORE_LIBRARY intel-ext-pt-cpu PATHS "${IPEX_INSTALL_PREFIX}/lib")
40
  list(APPEND IPEX_LIBRARIES_PRINT ${IPEX_CPU_CORE_LIBRARY})
41
  list(APPEND IPEX_LIBRARIES intel-ext-pt-cpu)
42
endif()
43

44
if(@BUILD_WITH_XPU@)
45
  find_library(IPEX_GPU_CORE_LIBRARY intel-ext-pt-gpu PATHS "${IPEX_INSTALL_PREFIX}/lib")
46
  list(APPEND IPEX_LIBRARIES_PRINT ${IPEX_GPU_CORE_LIBRARY})
47
  list(APPEND IPEX_LIBRARIES intel-ext-pt-gpu)
48
endif()
49

50
# When we build ipex with the old GCC ABI, dependent libraries must too.
51
if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
52
  list(APPEND IPEX_CXX_FLAGS "-D_GLIBCXX_USE_CXX11_ABI=@GLIBCXX_USE_CXX11_ABI@")
53
endif()
54

55
if(@BUILD_WITH_CPU@)
56
  add_library(intel-ext-pt-cpu INTERFACE IMPORTED)
57
  set_target_properties(intel-ext-pt-cpu PROPERTIES
58
      IMPORTED_LOCATION "${IPEX_CPU_CORE_LIBRARY}"
59
      INTERFACE_INCLUDE_DIRECTORIES "${IPEX_INCLUDE_DIRS}"
60
      INTERFACE_LINK_LIBRARIES "-Wl,--no-as-needed,\"${IPEX_CPU_CORE_LIBRARY}\""
61
      CXX_STANDARD 14
62
  )
63
  if(IPEX_CXX_FLAGS)
64
    set_property(TARGET intel-ext-pt-cpu PROPERTY INTERFACE_COMPILE_OPTIONS "${IPEX_CXX_FLAGS}")
65
  endif()
66
  set(IPEX_WITH_CPU ON)
67
endif()
68

69
if(@BUILD_WITH_XPU@)
70
  add_library(intel-ext-pt-gpu INTERFACE IMPORTED)
71
  set_target_properties(intel-ext-pt-gpu PROPERTIES
72
      IMPORTED_LOCATION "${IPEX_GPU_CORE_LIBRARY}"
73
      INTERFACE_INCLUDE_DIRECTORIES "${IPEX_INCLUDE_DIRS}"
74
      INTERFACE_LINK_LIBRARIES "-Wl,--no-as-needed,\"${IPEX_GPU_CORE_LIBRARY}\""
75
      CXX_STANDARD 17
76
  )
77
  if(IPEX_CXX_FLAGS)
78
    set_property(TARGET intel-ext-pt-gpu PROPERTY INTERFACE_COMPILE_OPTIONS "${IPEX_CXX_FLAGS}")
79
  endif()
80
  set(IPEX_WITH_XPU ON)
81
endif()
82

83
# Import Torch to simplify usage
84
FIND_PACKAGE(Torch REQUIRED)
85
if(TORCH_LIBRARIES)
86
  list(APPEND TORCH_IPEX_LIBRARIES ${TORCH_LIBRARIES})
87
  if(IPEX_LIBRARIES)
88
      list(APPEND TORCH_IPEX_LIBRARIES ${IPEX_LIBRARIES})
89
  endif()
90
endif()
91

92
find_package_handle_standard_args(IPEX DEFAULT_MSG IPEX_LIBRARIES_PRINT IPEX_INCLUDE_DIRS)
93