transformer_engine/pytorch/module.py · 9f7b025531a3bdd5e792d95c1704dad8c35e2db4 · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored Nov 15, 2022



addressed LayerNormMLP bias issue #26
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

9f7b0255

module.py 92.3 KB

Replace module.py