Support macos onnxruntime

open-mmlab · Nov 29, 2023 · aeac6b3 · aeac6b3
1 parent 8b19586
commit aeac6b3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/mmdeploy/backend/onnxruntime/init_plugins.py b/mmdeploy/backend/onnxruntime/init_plugins.py
@@ -13,6 +13,7 @@ def get_ops_path() -> str:
     candidates = [
         '../../lib/libmmdeploy_onnxruntime_ops.so',
         '../../lib/mmdeploy_onnxruntime_ops.dll',
+        '../../lib/libmmdeploy_onnxruntime_ops.dylib',
     ]
     return get_file_path(os.path.dirname(__file__), candidates)
 
@@ -26,5 +27,6 @@ def get_lib_path() -> str:
     candidates = [
         '../../lib/libonnxruntime.so*',
         '../../lib/onnxruntime.dll',
+        '../../lib/libmmdeploy_onnxruntime_ops.dylib',
     ]
     return get_file_path(os.path.dirname(__file__), candidates)
diff --git a/setup.py b/setup.py
@@ -138,7 +138,9 @@ def get_extensions():
         # environment, the compiler will choose the appropriate compiler
         # to compile those cpp files, so there is no need to add the
         # argument
-        if platform.system() != 'Windows':
+        if platform.system() == 'Darwin':
+            extra_compile_args['cxx'] = ['-std=c++17']
+        elif platform.system() != 'Windows':
             extra_compile_args['cxx'] = ['-std=c++14']
 
         include_dirs = []