XLAAccelerator
- class lightning.fabric.accelerators.XLAAccelerator(*args, **kwargs)
Bases:
AcceleratorAccelerator for XLA devices, normally TPUs.
Warning
Use of this accelerator beyond import and instantiation is experimental.
- static auto_device_count()
Get the devices when set to auto.
- Return type:
- static get_parallel_devices(devices)
Gets parallel devices for the Accelerator.
- static is_available()
Detect if the hardware is available.
- Return type:
- static name()
The name of the accelerator.
- Return type:
- classmethod register_accelerators(accelerator_registry)
Register the accelerator with the registry.
- Return type:
- setup_device(device)
Create and prepare the device for the current process.
- Return type:
- teardown()
Clean up any state created by the accelerator.
- Return type:
Was this page helpful?
Thanks for the feedback.