intel-extension-for-pytorch

1
# - Try to find torch-ccl
2
#
3
# The following are set after configuration is done:
4
#  TORCHCCL_FOUND          : set to true if oneCCL is found.
5
#  TORCHCCL_INCLUDE_DIR    : path to oneCCL include dir.
6
#  TORCHCCL_LIBRARIES      : list of libraries for oneCCL
7
#
8
# The following variables are used:
9
#  TORCHCCL_USE_NATIVE_ARCH : Whether native CPU instructions should be used in TORCHCCL. This should be turned off for
10
#  general packaging to avoid incompatible CPU instructions. Default: OFF.
11

12
IF (NOT TORCHCCL_FOUND)
13
SET(TORCHCCL_FOUND OFF)
14

15
SET(TORCHCCL_LIBRARIES)
16
SET(TORCHCCL_INCLUDE_DIR)
17

18
SET(TORCHCCL_ROOT "${PROJECT_SOURCE_DIR}/third_party/torch_ccl")
19

20
SET(CMAKE_INSTALL_PREFIX_SAVED "${CMAKE_INSTALL_PREFIX}")
21
SET(CMAKE_INSTALL_PREFIX "${CMAKE_INSTALL_PREFIX_SAVED}/../torch_ccl")
22
ADD_SUBDIRECTORY(${TORCHCCL_ROOT})
23
SET(CMAKE_INSTALL_PREFIX "${CMAKE_INSTALL_PREFIX_SAVED}")
24
IF(NOT TARGET torch_ccl)
25
    MESSAGE(FATAL_ERROR "Failed to include torch_ccl target")
26
ENDIF()
27
GET_TARGET_PROPERTY(INCLUDE_DIRS torch_ccl INCLUDE_DIRECTORIES)
28
SET(TORCHCCL_INCLUDE_DIR ${INCLUDE_DIRS})
29
SET(TORCHCCL_LIBRARIES torch_ccl)
30

31
ENDIF(NOT TORCHCCL_FOUND)
32