Allow Mac to use GPU Delegate

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

View File

@ -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(