transformer_engine/pytorch/module/base.py · ca6fedcfd128ec6349a1fc481e007eafea531e1b · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[PyTorch] Unblock fused bgrad quantization path for nvfp4 (#2246) · 66f9b3cb
Kirthi Shankar Sivamani authored Oct 07, 2025
```
Unblock path for fusing NVFP4 quantize and bgrad
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
66f9b3cb

base.py 68.5 KB