mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
47b3ecb53b
This PR disable UTs in training's package pipelines for building packages with PythonOp (torch.autograd.Function).
13 lines
818 B
CMake
13 lines
818 B
CMake
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
file(GLOB onnxruntime_pyop_srcs "${ONNXRUNTIME_ROOT}/core/language_interop_ops/pyop/pyop.cc")
|
|
onnxruntime_add_static_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 onnx onnx_proto ${PROTOBUF_LIB} flatbuffers)
|
|
target_include_directories(onnxruntime_pyop PRIVATE ${ONNXRUNTIME_ROOT} ${eigen_INCLUDE_DIRS})
|
|
onnxruntime_add_include_to_target(onnxruntime_pyop Python::Module Python::NumPy)
|
|
if (TARGET Python::Python)
|
|
target_link_libraries(onnxruntime_pyop PRIVATE Python::Python)
|
|
else()
|
|
target_link_libraries(onnxruntime_pyop PRIVATE Python::Module)
|
|
endif()
|