transformer_engine/pytorch/module.py · 40467fc24d99a5a2d13c7a50c1733bbaf76d1c55 · OpenDAS / TransformerEngine

"tests/vscode:/vscode.git/clone" did not exist on "cf5cb1e33eed16b2f0d5fe6268bf5705a4d0ea5a"

Indexing fix for bug in virtual interleaved pipelining configs (#52) · 40467fc2

Kirthi Shankar Sivamani authored Jan 13, 2023



Virtual interleaved pipelining fix
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

40467fc2

module.py 98 KB

Replace module.py