mirror of
https://github.com/microsoft/vcpkg.git
synced 2024-11-30 13:39:01 +08:00
69 lines
2.9 KiB
Diff
69 lines
2.9 KiB
Diff
|
diff --git a/CMakeLists.txt b/CMakeLists.txt
|
||
|
|
index 628dcaa..300e4ea 100644
|
||
|
|
--- a/CMakeLists.txt
|
||
|
|
+++ b/CMakeLists.txt
|
||
|
|
@@ -111,8 +111,8 @@ endif()
|
||
|
|
# find_package Python has replaced PythonInterp and PythonLibs since cmake 3.12
|
||
|
|
# Use the following command in the future; now this is only compatible with the latest pybind11
|
||
|
|
# find_package(Python ${PY_VERSION} COMPONENTS Interpreter Development REQUIRED)
|
||
|
|
-find_package(PythonInterp ${PY_VERSION} REQUIRED)
|
||
|
|
-find_package(PythonLibs ${PY_VERSION})
|
||
|
|
+find_package(Python3 ${PY_VERSION} COMPONENTS Interpreter REQUIRED)
|
||
|
|
+set(PYTHON_EXECUTABLE ${Python3_EXECUTABLE})
|
||
|
|
|
||
|
|
if(CMAKE_SYSTEM_NAME STREQUAL "AIX")
|
||
|
|
set(CMAKE_NO_SYSTEM_FROM_IMPORTED 1)
|
||
|
|
@@ -422,6 +422,7 @@ target_link_libraries(onnx PUBLIC onnx_proto)
|
||
|
|
add_onnx_global_defines(onnx)
|
||
|
|
|
||
|
|
if(BUILD_ONNX_PYTHON)
|
||
|
|
+ find_package(Python3 ${PY_VERSION} COMPONENTS Development REQUIRED)
|
||
|
|
if("${PY_EXT_SUFFIX}" STREQUAL "")
|
||
|
|
if(MSVC)
|
||
|
|
set(PY_EXT_SUFFIX ".pyd")
|
||
|
|
@@ -441,10 +442,13 @@ if(BUILD_ONNX_PYTHON)
|
||
|
|
$<BUILD_INTERFACE:${ONNX_ROOT}>
|
||
|
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}>
|
||
|
|
$<INSTALL_INTERFACE:include>
|
||
|
|
- ${PYTHON_INCLUDE_DIR})
|
||
|
|
-
|
||
|
|
+ ${Python3_INCLUDE_DIRS})
|
||
|
|
+ target_link_directories(onnx_cpp2py_export PRIVATE
|
||
|
|
+ ${Python3_LIBRARY_DIRS})
|
||
|
|
+ target_link_libraries(onnx_cpp2py_export PRIVATE
|
||
|
|
+ ${Python3_LIBRARIES})
|
||
|
|
# pybind11 is a header only lib
|
||
|
|
- find_package(pybind11 2.2)
|
||
|
|
+ find_package(pybind11 2.2 CONFIG REQUIRED)
|
||
|
|
if(pybind11_FOUND)
|
||
|
|
target_include_directories(onnx_cpp2py_export PUBLIC
|
||
|
|
${pybind11_INCLUDE_DIRS})
|
||
|
|
@@ -687,6 +691,27 @@ endif()
|
||
|
|
|
||
|
|
include(GNUInstallDirs)
|
||
|
|
|
||
|
|
+# install protobuf files
|
||
|
|
+install(FILES ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx-data.proto
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx-data.proto3
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx-ml.proto
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx-ml.proto3
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx-operators-ml.proto
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx-operators-ml.proto3
|
||
|
|
+ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/onnx
|
||
|
|
+)
|
||
|
|
+# install python files
|
||
|
|
+if(BUILD_ONNX_PYTHON)
|
||
|
|
+ install(FILES ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx_data_pb.py
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx_data_pb2.py
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx_ml_pb2.py
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx_operators_ml_pb2.py
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx_operators_pb.py
|
||
|
|
+ ${CMAKE_CURRENT_BINARY_DIR}/onnx/onnx_pb.py
|
||
|
|
+ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/onnx
|
||
|
|
+ )
|
||
|
|
+endif()
|
||
|
|
+
|
||
|
|
install(DIRECTORY ${ONNX_ROOT}/onnx
|
||
|
|
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}
|
||
|
|
FILES_MATCHING
|