pytorch/torch/testing
Rodrigo Kumpera 5b4a523583 Add all_reduce_coalesced to functional collectives (#98640)
This adds all_reduce_coalesced to MTPG to ease testing.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/98640
Approved by: https://github.com/wanchaol
2023-04-26 17:05:54 +00:00
..
_internal Add all_reduce_coalesced to functional collectives (#98640) 2023-04-26 17:05:54 +00:00
__init__.py document torch.testing.assert_allclose (#89526) 2022-12-01 11:22:50 +00:00
_comparison.py improve memory footprint of torch.testing.assert_close (#96131) 2023-03-29 23:49:56 +00:00
_creation.py disallow invalid value ranges in torch.testing.make_tensor (#96334) 2023-03-24 23:55:17 +00:00