transformer_engine/pytorch/distributed.py · a6b5f96d7e065e33dd6718e3ab3744d44453dd23 · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored Oct 04, 2022


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

a6b5f96d

distributed.py 10.6 KB