transformer_engine/pytorch/transformer.py · 0c9c0ba1fe0f5f43b4bb68a690b9d8832496216b · OpenDAS / TransformerEngine

Enforce boolean attention mask type (#49) · 0c9c0ba1

Kirthi Shankar Sivamani authored Jan 04, 2023



* Enforce boolean attention mask type
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

0c9c0ba1

transformer.py 43.1 KB

Replace transformer.py