Commit 4d08193a authored by Paul's avatar Paul
Browse files

Use default python if available

parent cdb0d24f
find_program(DEFAULT_PYTHON_EXE python)
if(DEFAULT_PYTHON_EXE)
set(PYTHON_EXECUTABLE ${DEFAULT_PYTHON_EXE} CACHE PATH "Path to python executable")
endif()
find_package(pybind11 REQUIRED) find_package(pybind11 REQUIRED)
pybind11_add_module(migraphx_py migraphx_py.cpp) pybind11_add_module(migraphx_py migraphx_py.cpp)
target_link_libraries(migraphx_py migraphx migraphx_onnx migraphx_cpu) target_link_libraries(migraphx_py migraphx migraphx_onnx migraphx_cpu)
set_target_properties(migraphx_py PROPERTIES set_target_properties(migraphx_py PROPERTIES
OUTPUT_NAME migraphx OUTPUT_NAME migraphx
) )
if(MIGRAPHX_ENABLE_GPU)
target_link_libraries(migraphx_py migraphx_gpu)
target_compile_definitions(migraphx_py PRIVATE -DHAVE_GPU)
endif()
...@@ -12,5 +12,7 @@ function(add_py_test NAME SCRIPT) ...@@ -12,5 +12,7 @@ function(add_py_test NAME SCRIPT)
COMMAND ${ENV_COMMAND} ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/${SCRIPT} ${ARGN} COMMAND ${ENV_COMMAND} ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/${SCRIPT} ${ARGN}
COMMENT "${PYTHON_EXECUTABLE} ${SCRIPT}") COMMENT "${PYTHON_EXECUTABLE} ${SCRIPT}")
endfunction() endfunction()
add_dependencies(tests migraphx_py)
add_dependencies(check migraphx_py)
add_py_test(test test.py WORKING_DIRECTORY ${TEST_ONNX_DIR}) add_py_test(test test.py WORKING_DIRECTORY ${TEST_ONNX_DIR})
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment