pytorch

1
# This will define the following variables:
2
# SYCL_FOUND               : True if the system has the SYCL library.
3
# SYCL_INCLUDE_DIR         : Include directories needed to use SYCL.
4
# SYCL_LIBRARY_DIR         ：The path to the SYCL library.
5
# SYCL_LIBRARY             : SYCL library fullname.
6

7
include(FindPackageHandleStandardArgs)
8

9
set(SYCL_ROOT "")
10
if(DEFINED ENV{SYCL_ROOT})
11
  set(SYCL_ROOT $ENV{SYCL_ROOT})
12
elseif(DEFINED ENV{CMPLR_ROOT})
13
  set(SYCL_ROOT $ENV{CMPLR_ROOT})
14
endif()
15

16
string(COMPARE EQUAL "${SYCL_ROOT}" "" nosyclfound)
17
if(nosyclfound)
18
  set(SYCL_FOUND False)
19
  set(SYCL_REASON_FAILURE "SYCL library not set!!")
20
  set(SYCL_NOT_FOUND_MESSAGE "${SYCL_REASON_FAILURE}")
21
  return()
22
endif()
23

24
# Find include path from binary.
25
find_file(
26
  SYCL_INCLUDE_DIR
27
  NAMES include
28
  HINTS ${SYCL_ROOT}
29
  NO_DEFAULT_PATH
30
  )
31

32
# Find include/sycl path from include path.
33
find_file(
34
  SYCL_INCLUDE_SYCL_DIR
35
  NAMES sycl
36
  HINTS ${SYCL_ROOT}/include/
37
  NO_DEFAULT_PATH
38
  )
39

40
# Due to the unrecognized compilation option `-fsycl` in other compiler.
41
list(APPEND SYCL_INCLUDE_DIR ${SYCL_INCLUDE_SYCL_DIR})
42

43
# Find library directory from binary.
44
find_file(
45
  SYCL_LIBRARY_DIR
46
  NAMES lib lib64
47
  HINTS ${SYCL_ROOT}
48
  NO_DEFAULT_PATH
49
  )
50

51
# Find SYCL library fullname.
52
find_library(
53
  SYCL_LIBRARY
54
  NAMES sycl
55
  HINTS ${SYCL_LIBRARY_DIR}
56
  NO_DEFAULT_PATH
57
)
58

59
if((NOT SYCL_INCLUDE_DIR) OR (NOT SYCL_LIBRARY_DIR) OR (NOT SYCL_LIBRARY))
60
  set(SYCL_FOUND False)
61
  set(SYCL_REASON_FAILURE "SYCL library is incomplete!!")
62
  set(SYCL_NOT_FOUND_MESSAGE "${SYCL_REASON_FAILURE}")
63
  return()
64
endif()
65

66
find_package_handle_standard_args(
67
  SYCL
68
  FOUND_VAR SYCL_FOUND
69
  REQUIRED_VARS SYCL_INCLUDE_DIR SYCL_LIBRARY_DIR SYCL_LIBRARY
70
  REASON_FAILURE_MESSAGE "${SYCL_REASON_FAILURE}")
71