pytorch

1
# ---[ xpu
2

3
# Poor man's include guard
4
if(TARGET torch::xpurt)
5
  return()
6
endif()
7

8
# Find SYCL library.
9
find_package(SYCLToolkit REQUIRED)
10
if(NOT SYCL_FOUND)
11
  set(PYTORCH_FOUND_XPU FALSE)
12
  return()
13
endif()
14
set(PYTORCH_FOUND_XPU TRUE)
15

16
# SYCL library interface
17
add_library(torch::sycl INTERFACE IMPORTED)
18

19
set_property(
20
    TARGET torch::sycl PROPERTY INTERFACE_INCLUDE_DIRECTORIES
21
    ${SYCL_INCLUDE_DIR})
22
set_property(
23
    TARGET torch::sycl PROPERTY INTERFACE_LINK_LIBRARIES
24
    ${SYCL_LIBRARY})
25

26
# xpurt
27
add_library(torch::xpurt INTERFACE IMPORTED)
28
set_property(
29
    TARGET torch::xpurt PROPERTY INTERFACE_LINK_LIBRARIES
30
    torch::sycl)
31