Allow Mac to use GPU Delegate

PiperOrigin-RevId: 575882254
2023-10-23 11:41:19 -07:00 · 2023-10-23 11:41:19 -07:00 · b904ade0cf
commit b904ade0cf
parent 6aa27d9aeb
1 changed files with 1 additions and 1 deletions
--- a/mediapipe/tasks/python/core/base_options.py
+++ b/mediapipe/tasks/python/core/base_options.py
@ -70,7 +70,7 @@ class BaseOptions:
    platform_name = platform.system()

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