27 lines
962 B
CMake
27 lines
962 B
CMake
|
# Run `EXPR` in python. The standard output of python is stored in `OUT` and
|
||
|
|
# has trailing whitespace stripped. If an error is encountered when running
|
||
|
|
# python, a fatal message `ERR_MSG` is issued.
|
||
|
|
#
|
||
|
|
function (run_python OUT EXPR ERR_MSG)
|
||
|
|
execute_process(
|
||
|
|
COMMAND
|
||
|
|
"${PYTHON_EXECUTABLE}" "-c" "${EXPR}"
|
||
|
|
OUTPUT_VARIABLE PYTHON_OUT
|
||
|
|
RESULT_VARIABLE PYTHON_ERROR_CODE
|
||
|
|
ERROR_VARIABLE PYTHON_STDERR
|
||
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE)
|
||
|
|
|
||
|
|
if(NOT PYTHON_ERROR_CODE EQUAL 0)
|
||
|
|
message(FATAL_ERROR "${ERR_MSG}: ${PYTHON_STDERR}")
|
||
|
|
endif()
|
||
|
|
set(${OUT} ${PYTHON_OUT} PARENT_SCOPE)
|
||
|
|
endfunction()
|
||
|
|
|
||
|
|
# Run `EXPR` in python after importing `PKG`. Use the result of this to extend
|
||
|
|
# `CMAKE_PREFIX_PATH` so the torch cmake configuration can be imported.
|
||
|
|
macro (append_cmake_prefix_path PKG EXPR)
|
||
|
|
run_python(_PREFIX_PATH
|
||
|
|
"import ${PKG}; print(${EXPR})" "Failed to locate ${PKG} path")
|
||
|
|
list(APPEND CMAKE_PREFIX_PATH ${_PREFIX_PATH})
|
||
|
|
endmacro()
|