We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2882c64 commit 8b19586Copy full SHA for 8b19586
mmdeploy/backend/onnxruntime/init_plugins.py
@@ -24,7 +24,7 @@ def get_lib_path() -> str:
24
str: The library path to onnxruntime.
25
"""
26
candidates = [
27
- '../../lib/libonnxruntime_ops.so*',
+ '../../lib/libonnxruntime.so*',
28
'../../lib/onnxruntime.dll',
29
]
30
return get_file_path(os.path.dirname(__file__), candidates)
mmdeploy/backend/onnxruntime/wrapper.py
@@ -48,7 +48,7 @@ def __init__(self,
48
# load ort lib before custom ops lib
49
lib_path = get_lib_path()
50
if osp.exists(lib_path):
51
- ctypes.DLL(lib_path)
+ ctypes.CDLL(lib_path)
52
53
session_options.register_custom_ops_library(ort_custom_op_path)
54
logger.info('Successfully loaded onnxruntime custom ops from '
0 commit comments