transformer_engine/pytorch/module.py · 677de00de2881ebaa0c1978a06dd39be5e76f27a · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored Jan 25, 2023



Stats and dgrad bug fix
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

677de00d

module.py 109 KB