microsoft/onnxruntime-extensions
Publicmirrored fromhttps://github.com/microsoft/onnxruntime-extensionsAvailable
cmake/ext_python.cmake
39lines · modecode
| 1 | set(Python3_FIND_REGISTRY NEVER CACHE STRING "...") |
| 2 | if(NOT "${Python3_FIND_REGISTRY}" STREQUAL "NEVER") |
| 3 | message(FATAL_ERROR "Python3_FIND_REGISTRY is not NEVER") |
| 4 | endif() |
| 5 | find_package(Python3 COMPONENTS Interpreter Development.Module NumPy) |
| 6 | if (NOT Python3_FOUND) |
| 7 | message(FATAL_ERROR "Python3 or NumPy not found!") |
| 8 | endif() |
| 9 | if (WIN32) |
| 10 | list(APPEND shared_TARGET_SRC "${PROJECT_SOURCE_DIR}/pyop/extensions_pydll.def") |
| 11 | endif() |
| 12 | |
| 13 | file(GLOB TARGET_SRC_PYOPS "pyop/*.cc" "pyop/*.h" "shared/*.cc") |
| 14 | add_library(extensions_pydll SHARED ${TARGET_SRC_PYOPS} ${shared_TARGET_LIB_SRC}) |
| 15 | standardize_output_folder(extensions_pydll) |
| 16 | list(APPEND OCOS_COMPILE_DEFINITIONS PYTHON_OP_SUPPORT) |
| 17 | target_compile_definitions(extensions_pydll PRIVATE ${OCOS_COMPILE_DEFINITIONS}) |
| 18 | |
| 19 | message(STATUS "Fetch pybind11") |
| 20 | include(pybind11) |
| 21 | target_include_directories(extensions_pydll PRIVATE |
| 22 | ${pybind11_INCLUDE_DIRS} |
| 23 | $<TARGET_PROPERTY:Python3::Module,INTERFACE_INCLUDE_DIRECTORIES> |
| 24 | $<TARGET_PROPERTY:Python3::NumPy,INTERFACE_INCLUDE_DIRECTORIES> |
| 25 | $<TARGET_PROPERTY:ocos_operators,INTERFACE_INCLUDE_DIRECTORIES>) |
| 26 | |
| 27 | target_compile_definitions(extensions_pydll PRIVATE |
| 28 | $<TARGET_PROPERTY:Python3::Module,INTERFACE_COMPILE_DEFINITIONS>) |
| 29 | |
| 30 | target_link_libraries(extensions_pydll PRIVATE Python3::Module ocos_operators) |
| 31 | |
| 32 | if(NOT "${OCOS_EXTENTION_NAME}" STREQUAL "") |
| 33 | if(NOT WIN32) |
| 34 | set_target_properties(extensions_pydll PROPERTIES |
| 35 | LIBRARY_OUTPUT_NAME ${OCOS_EXTENTION_NAME} |
| 36 | PREFIX "" |
| 37 | SUFFIX "") |
| 38 | endif() |
| 39 | endif() |
| 40 | |