Fix `rtol` in `assert_close` cleanup

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

BSD 3-Clause "New" or "Revised" License

8.43k stars 1.4k forks source link

Closed eqy closed 1 year ago

eqy commented 1 year ago

Fix broken call in #1742 CC @crcrpar