transformer_engine/pytorch/module.py · e456110b56e9616e20bf5c9ea320e4ff3a729fcf · OpenDAS / TransformerEngine

Ensure contiguous inputs (#38) · 509bf877

Kirthi Shankar Sivamani authored Dec 06, 2022



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

509bf877

module.py 95 KB

Replace module.py