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

[Pytorch] Added missing assert_dim_for_fp8_exec for Linear · d668f18f

Paweł Gadziński authored Feb 24, 2025



* fix
Signed-off-by: Pawel Gadzinski <pgadzinski@nvidia.com>

* reshape inp
Signed-off-by: Pawel Gadzinski <pgadzinski@nvidia.com>

---------
Signed-off-by: Pawel Gadzinski <pgadzinski@nvidia.com>

d668f18f

linear.py 45.5 KB

Replace linear.py