thunder.plugins.FP8¶
- class thunder.plugins.FP8[source]¶
Bases:
Plugin
Plugin for enabling FP8 precision via NVIDIA Transformer Engine, enabling higher throughput of matrix operations in FP8.
See lightning-thunder/thunder/executors/transformer_engineex.py for implementation details.
Methods
__init__
()Imports the Transformer Engine executor.
setup_lookasides
()setup_transforms
()Attributes
policy