HPUPrecisionPlugin

class pytorch_lightning.plugins.precision.HPUPrecisionPlugin(precision, opt_level='O2', bf16_file_path=None, fp32_file_path=None, verbose=False)[source]

Bases: pytorch_lightning.plugins.precision.precision_plugin.PrecisionPlugin

Plugin that enables bfloat/half support on HPUs.

Parameters

• precision (Union[Literal[32, 16], Literal[‘32’, ‘16’, ‘bf16’]]) – The precision to use.

• opt_level (str) – Choose optimization level for hmp.

• bf16_file_path (Optional[str]) – Path to bf16 ops list in hmp O1 mode.

• fp32_file_path (Optional[str]) – Path to fp32 ops list in hmp O1 mode.

• verbose (bool) – Enable verbose mode for hmp.