transformer_engine/pytorch/module/linear.py · 14c1ecd09c86a2bbab4702e15ea19fc0da0e05b4 · OpenDAS / TransformerEngine

[PyTorch] Misc fixes for release_v1.6 (#784) · cd54a8cd

Kirthi Shankar Sivamani authored Apr 17, 2024



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

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

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

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

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

cd54a8cd

linear.py 43.1 KB

Replace linear.py