tests/pytorch/test_numerics.py · fceff07a59bacd517baaf6a0f9cb0fb087f117ea · OpenDAS / TransformerEngine

[PyTorch] Fix fuse_wgrad_accumulation for GroupedLinear (#1488) · fceff07a

Xin Yao authored Feb 20, 2025



* fix fuse_wgrad_accumulation for GroupedLinear
Signed-off-by: Xin Yao <xiny@nvidia.com>

* fix fuse_wgrad_accumulation for GroupedLinear
Signed-off-by: Xin Yao <xiny@nvidia.com>

* update tests
Signed-off-by: Xin Yao <xiny@nvidia.com>

---------
Signed-off-by: Xin Yao <xiny@nvidia.com>
Co-authored-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

fceff07a

test_numerics.py 72.8 KB

Replace test_numerics.py