2020-11-21 09:57:46 +08:00
|
|
|
# Macros and functions related to detecting details of the Python environment.
|
|
|
|
|
|
|
|
# Detects a pybind11 package installed in the current python environment
|
|
|
|
# and sets variables to allow it to be found. This allows pybind11 to be
|
|
|
|
# installed via pip, which typically yields a much more recent version than
|
|
|
|
# the OS install, which will be available otherwise.
|
|
|
|
function(mlir_detect_pybind11_install)
|
|
|
|
if(pybind11_DIR)
|
|
|
|
message(STATUS "Using explicit pybind11 cmake directory: ${pybind11_DIR} (-Dpybind11_DIR to change)")
|
|
|
|
else()
|
2020-11-23 20:58:03 +08:00
|
|
|
message(STATUS "Checking for pybind11 in python path...")
|
2020-11-21 09:57:46 +08:00
|
|
|
execute_process(
|
2020-11-25 01:50:18 +08:00
|
|
|
COMMAND "${Python3_EXECUTABLE}"
|
2020-11-21 09:57:46 +08:00
|
|
|
-c "import pybind11;print(pybind11.get_cmake_dir(), end='')"
|
|
|
|
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
|
|
|
|
RESULT_VARIABLE STATUS
|
|
|
|
OUTPUT_VARIABLE PACKAGE_DIR
|
|
|
|
ERROR_QUIET)
|
|
|
|
if(NOT STATUS EQUAL "0")
|
2020-11-23 20:58:03 +08:00
|
|
|
message(STATUS "not found (install via 'pip install pybind11' or set pybind11_DIR)")
|
2020-11-21 09:57:46 +08:00
|
|
|
return()
|
|
|
|
endif()
|
2020-11-23 20:58:03 +08:00
|
|
|
message(STATUS "found (${PACKAGE_DIR})")
|
2020-11-21 09:57:46 +08:00
|
|
|
set(pybind11_DIR "${PACKAGE_DIR}" PARENT_SCOPE)
|
|
|
|
endif()
|
|
|
|
endfunction()
|