intel-extension-for-pytorch

1
# Try to find the pybind11 library and headers.
2
#  pybind11_FOUND        - system has pybind11
3
#  pybind11_INCLUDE_DIRS - the pybind11 include directory
4

5
set(pybind11_INCLUDE_DIRS)
6
set(torch_pybind11_root_hint)
7

8
if(TORCH_FOUND)
9
        set(torch_pybind11_root_hint ${TORCH_INCLUDE_DIRS})
10
elseif(DEFINED ENV{TORCH_ROOT})
11
        set(torch_pybind11_root_hint "$ENV{TORCH_ROOT}/include")
12
endif()
13

14
find_path(pybind11_INCLUDE_DIR
15
        NAMES pybind11/pybind11.h
16
        HINTS ${torch_pybind11_root_hint}
17
        DOC "The directory where pybind11 includes reside"
18
)
19

20
set(pybind11_INCLUDE_DIRS ${pybind11_INCLUDE_DIR})
21

22
include(FindPackageHandleStandardArgs)
23
find_package_handle_standard_args(pybind11
24
        FOUND_VAR pybind11_FOUND
25
        REQUIRED_VARS pybind11_INCLUDE_DIR
26
)
27

28
mark_as_advanced(pybind11_FOUND)
29