Skip to content

Commit 8b19586

Browse files
authored
fix ort wrapper (#2560)
1 parent 2882c64 commit 8b19586

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

mmdeploy/backend/onnxruntime/init_plugins.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def get_lib_path() -> str:
2424
str: The library path to onnxruntime.
2525
"""
2626
candidates = [
27-
'../../lib/libonnxruntime_ops.so*',
27+
'../../lib/libonnxruntime.so*',
2828
'../../lib/onnxruntime.dll',
2929
]
3030
return get_file_path(os.path.dirname(__file__), candidates)

mmdeploy/backend/onnxruntime/wrapper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def __init__(self,
4848
# load ort lib before custom ops lib
4949
lib_path = get_lib_path()
5050
if osp.exists(lib_path):
51-
ctypes.DLL(lib_path)
51+
ctypes.CDLL(lib_path)
5252

5353
session_options.register_custom_ops_library(ort_custom_op_path)
5454
logger.info('Successfully loaded onnxruntime custom ops from '

0 commit comments

Comments
 (0)