cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow Mac to use GPU Delegate

Browse Source 
PiperOrigin-RevId: 575882254
This commit is contained in:
Sebastian Schmidt 2023-10-23 11:41:19 -07:00 committed by Copybara-Service
parent 6aa27d9aeb
commit b904ade0cf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/tasks/python/core/base_options.py
View File

@ -70,7 +70,7 @@ class BaseOptions:
platform_name = platform.system() platform_name = platform.system()
if self.delegate == BaseOptions.Delegate.GPU: if self.delegate == BaseOptions.Delegate.GPU:
if platform_name == 'Linux': if platform_name in ['Linux', 'Darwin']:
acceleration_proto = _AccelerationProto(gpu=_DelegateProto.Gpu()) acceleration_proto = _AccelerationProto(gpu=_DelegateProto.Gpu())
else: else:
raise NotImplementedError( raise NotImplementedError(