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