transformer_engine/pytorch/module/linear.py · f69e45be65f3b6c80987acc55905e891b0b96a14 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[PyTorch] Don't use autograd hook for bwd reduction (#781) · f69e45be
Kirthi Shankar Sivamani authored Apr 15, 2024
```
Don't use autograd hook for bwd reduction
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
f69e45be

linear.py 38.9 KB