A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.42k
stars
1.4k
forks
source link
Bump thresholds for `test_backward` in `test_fused_softmax.py` #1782
Closed
eqy closed 8 months ago
Workaround for higher mismatches seen on RTX 6000 (Ada)