transformer_engine/pytorch/fp8.py · 2215fa5c7557b66034068816020f9f611019e457 · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored Jun 13, 2024



* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

9416519d

fp8.py 25.7 KB

Replace fp8.py