add in macos handeling for `auto_annotation` (#524)

main
Ben Hoff 7 years ago committed by Nikita Manovich
parent 9eabd1738e
commit 41ee68f565

@ -6,6 +6,7 @@ from openvino.inference_engine import IENetwork, IEPlugin
import subprocess
import os
import platform
_IE_PLUGINS_PATH = os.getenv("IE_PLUGINS_PATH", None)
@ -27,6 +28,8 @@ def make_plugin():
plugin.add_cpu_extension(os.path.join(_IE_PLUGINS_PATH, 'libcpu_extension_avx2.so'))
elif (_check_instruction('sse4')):
plugin.add_cpu_extension(os.path.join(_IE_PLUGINS_PATH, 'libcpu_extension_sse4.so'))
elif platform.system() == 'Darwin':
plugin.add_cpu_extension(os.path.join(_IE_PLUGINS_PATH, 'libcpu_extension.dylib'))
else:
raise Exception('Inference engine requires a support of avx2 or sse4.')

Loading…
Cancel
Save