HPUPrecisionPlugin¶

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

Bases: lightning.pytorch.plugins.precision.precision_plugin.PrecisionPlugin

Plugin that enables bfloat/half support on HPUs.

Warning

This is an experimental feature.

Parameters

precision¶ (Literal[‘32-true’, ‘16-mixed’, ‘bf16-mixed’]) – 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.