mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-01 23:30:35 +00:00
6d7081f33f
* Enable Interop Op test in ci pipelines * add doc * update doc * update doc * add supported types * format doc * update doc * update doc * update doc * update doc * update doc * update doc * update doc * remove comment * catch up with latest C API * fix cmake issue * update doc * add comments * update doc * update doc * fix compile err on mac * update doc * update doc * update doc * update doc * add sequence chart * format sequence chart * format sequence chart * update doc * update doc * update doc * update doc * update doc * update doc * fix cmake issue * fix cmake issue * fix cmake issue * fix cmake issue * fix cmake issue * fix cmake issue * fix cmake issue * add graph dependency * add graph dependency * add graph dependency * add links * enable test from yml * revert change in yml * renable test in yml * update doc
59 lines
2.3 KiB
CMake
59 lines
2.3 KiB
CMake
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
set(PYTHON_NOT_FOUND false)
|
|
exec_program("${PYTHON_EXECUTABLE}" # get python include path
|
|
ARGS "-c \"import distutils.sysconfig; print(distutils.sysconfig.get_python_inc())\""
|
|
OUTPUT_VARIABLE PYTHON_INCLUDE_DIR
|
|
RETURN_VALUE PYTHON_NOT_FOUND)
|
|
if(${PYTHON_NOT_FOUND})
|
|
message(FATAL_ERROR "Cannot get Python include directory. Is distutils installed?")
|
|
else()
|
|
exec_program("${PYTHON_EXECUTABLE}" # get python library path
|
|
ARGS "-c \"
|
|
import os
|
|
import re
|
|
import sys
|
|
major = sys.version_info.major
|
|
minor = sys.version_info.minor
|
|
regex = re.compile('^(libpython|python)'+str(major)+'\.?'+str(minor)+'m?\.(so|lib|dylib)$')
|
|
pydir = os.path.abspath(os.path.join(os.path.dirname(sys.executable),'..'))
|
|
for r,d,fs in os.walk(pydir):
|
|
for f in fs:
|
|
if regex.match(f):
|
|
print (os.path.join(r,f))
|
|
sys.exit()
|
|
\"
|
|
"
|
|
OUTPUT_VARIABLE PYTHON_LIBRARIES)
|
|
endif(${PYTHON_NOT_FOUND})
|
|
|
|
exec_program("${PYTHON_EXECUTABLE}" # get numpy include path
|
|
ARGS "-c \"import numpy; print(numpy.get_include())\""
|
|
OUTPUT_VARIABLE NUMPY_INCLUDE_DIR
|
|
RETURN_VALUE NUMPY_NOT_FOUND)
|
|
if(${NUMPY_NOT_FOUND})
|
|
message(FATAL_ERROR "Cannot get NumPy include directory: Is NumPy installed?")
|
|
endif(${NUMPY_NOT_FOUND})
|
|
|
|
message("PYTHON EXE: " ${PYTHON_EXECUTABLE})
|
|
message("PYTHON INC: " ${PYTHON_INCLUDE_DIR})
|
|
message("PYTHON LIB: " ${PYTHON_LIBRARIES})
|
|
message("NUMPY INC: " ${NUMPY_INCLUDE_DIR})
|
|
|
|
include_directories(${PYTHON_INCLUDE_DIR})
|
|
include_directories(${NUMPY_INCLUDE_DIR})
|
|
|
|
file(GLOB onnxruntime_pywrapper_srcs "${ONNXRUNTIME_ROOT}/core/language_interop_ops/pyop/pywrapper.cc")
|
|
add_library(onnxruntime_pywrapper SHARED ${onnxruntime_pywrapper_srcs})
|
|
if (WIN32)
|
|
set_target_properties(onnxruntime_pywrapper PROPERTIES LINK_FLAGS "/ignore:4199")
|
|
endif()
|
|
target_link_libraries(onnxruntime_pywrapper PUBLIC ${PYTHON_LIBRARIES})
|
|
|
|
file(GLOB onnxruntime_pyop_srcs "${ONNXRUNTIME_ROOT}/core/language_interop_ops/pyop/pyop.cc")
|
|
add_library(onnxruntime_pyop ${onnxruntime_pyop_srcs})
|
|
add_dependencies(onnxruntime_pyop onnxruntime_graph)
|
|
onnxruntime_add_include_to_target(onnxruntime_pyop onnxruntime_common onnxruntime_graph onnxruntime_framework gsl onnx onnx_proto protobuf::libprotobuf)
|
|
target_include_directories(onnxruntime_pyop PRIVATE ${ONNXRUNTIME_ROOT} ${eigen_INCLUDE_DIRS})
|
|
|