xref: /aosp_15_r20/external/pytorch/cmake/public/xpu.cmake (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1# ---[ xpu
2
3# Poor man's include guard
4if(TARGET torch::xpurt)
5  return()
6endif()
7
8# Find SYCL library.
9find_package(SYCLToolkit REQUIRED)
10if(NOT SYCL_FOUND)
11  set(PYTORCH_FOUND_XPU FALSE)
12  return()
13endif()
14set(PYTORCH_FOUND_XPU TRUE)
15
16# SYCL library interface
17add_library(torch::sycl INTERFACE IMPORTED)
18
19set_property(
20    TARGET torch::sycl PROPERTY INTERFACE_INCLUDE_DIRECTORIES
21    ${SYCL_INCLUDE_DIR})
22set_property(
23    TARGET torch::sycl PROPERTY INTERFACE_LINK_LIBRARIES
24    ${SYCL_LIBRARY})
25
26# xpurt
27add_library(torch::xpurt INTERFACE IMPORTED)
28set_property(
29    TARGET torch::xpurt PROPERTY INTERFACE_LINK_LIBRARIES
30    torch::sycl)
31