transformer_engine/pytorch/module/linear.py · db2aaa9e7765a3bcd1e652e3539d57e85cdbd5bc · OpenDAS / TransformerEngine

"git@developer.sourcefind.cn:liming6/sshd-tool.git" did not exist on "d198770e96a6eac4d7b6233e6f411e339b32ce3d"

Fix fp8_buf for Linear and LayerNormLinear (#1633) · e3e0375d

Kirthi Shankar Sivamani authored Apr 02, 2025



* Fix fp8_buf for Linear and LayerNormLinear
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

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

e3e0375d

linear.py 53.5 KB

Replace linear.py