XLAPrecision¶
- class lightning.fabric.plugins.precision.XLAPrecision(precision)[source]¶
Bases:
Precision
Plugin for training with XLA.
- Parameters:
precision¶ (
Literal
['32-true'
,'16-true'
,'bf16-true'
]) – Full precision (32-true) or half precision (16-true, bf16-true).- Raises:
ValueError – If unsupported
precision
is provided.