Skip to content

Commit d13491c

Browse files
smkarlapDu Li
authored andcommitted
Add OpenVINO EP shared lib to Py Wheel (#5920)
* Add OpenVINO EP shared lib to Py Wheel Include the libonnxruntime_providers_openvino.so/.dll to the wheel * Follow libs.extend pattern as other EPs (cherry picked from commit 4092686)
1 parent 4fb9e51 commit d13491c

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

setup.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -158,14 +158,13 @@ def run(self):
158158
# Additional binaries
159159
if platform.system() == 'Linux':
160160
libs = ['onnxruntime_pybind11_state.so', 'libdnnl.so.1', 'libmklml_intel.so', 'libmklml_gnu.so', 'libiomp5.so', 'mimalloc.so']
161-
# DNNL & TensorRT EPs are built as shared libs
161+
# DNNL, TensorRT & OpenVINO EPs are built as shared libs
162162
libs.extend(['libonnxruntime_providers_shared.so'])
163163
libs.extend(['libonnxruntime_providers_dnnl.so'])
164164
libs.extend(['libonnxruntime_providers_tensorrt.so'])
165-
# OpenVINO Libs
166-
if package_name == 'onnxruntime-openvino':
167-
if platform.system() == 'Linux':
168-
libs.extend(['libovep_ngraph.so'])
165+
libs.extend(['libonnxruntime_providers_openvino.so'])
166+
# OpenVINO libs
167+
libs.extend(['libovep_ngraph.so'])
169168
# Nuphar Libs
170169
libs.extend(['libtvm.so.0.5.1'])
171170
if nightly_build:
@@ -180,10 +179,11 @@ def run(self):
180179
libs.extend(['libonnxruntime_pywrapper.dylib'])
181180
else:
182181
libs = ['onnxruntime_pybind11_state.pyd', 'dnnl.dll', 'mklml.dll', 'libiomp5md.dll']
183-
# DNNL & TensorRT EPs are built as shared libs
182+
# DNNL, TensorRT & OpenVINO EPs are built as shared libs
184183
libs.extend(['onnxruntime_providers_shared.dll'])
185184
libs.extend(['onnxruntime_providers_dnnl.dll'])
186185
libs.extend(['onnxruntime_providers_tensorrt.dll'])
186+
libs.extend(['onnxruntime_providers_openvino.dll'])
187187
# DirectML Libs
188188
libs.extend(['directml.dll'])
189189
# Nuphar Libs

0 commit comments

Comments
 (0)