Allow Mac to use GPU Delegate
PiperOrigin-RevId: 575882254
This commit is contained in:
parent
6aa27d9aeb
commit
b904ade0cf
|
@ -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(
|
||||
|
|
Loading…
Reference in New Issue
Block a user