MPSAccelerator¶
- class lightning.fabric.accelerators.MPSAccelerator[source]¶
Bases:
Accelerator
Accelerator for Metal Apple Silicon GPU devices.
Warning
Use of this accelerator beyond import and instantiation is experimental.
- static is_available()[source]¶
MPS is only available on a machine with the ARM-based Apple Silicon processors.
- Return type:
- setup_device(device)[source]¶
- Raises:
ValueError – If the selected device is not MPS.
- Return type: