transformer_engine/pytorch/attention.py · 487871e2abf05e06e83fbfb1390ff4ec1cad11cf · OpenDAS / TransformerEngine · GitLab

You need to sign in or sign up before continuing.

Find file Blame History Permalink

Revert "Fix BF16 ONNX export for successful ONNX Runtime Verification (#271)" (#275) · 487871e2
Tim Moon authored Jun 12, 2023
```
This reverts commit 914f3841

.
Signed-off-by: Tim Moon <tmoon@nvidia.com>
```
487871e2

attention.py 36 KB