A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.2k
stars
1.36k
forks
source link
Use `torch.testing.assert_close` in test_index_mul_2d.py #1693
Closed
crcrpar closed 1 year ago
hopefully this would give us some details
cc @eqy