Hari Krishna Sai Kodali 9d184bda2f add device generalization support for distributed tests (#156796) ... MOTIVATION To generalize Distributed test cases for non-CUDA devices CHANGES - test/distributed/checkpoint/test_fsspec.py - test/distributed/checkpoint/test_state_dict.py - test/distributed/test_multi_threaded_pg.py Replaced hard coded device names with torch.accelerator.current_accelerator - torch/testing/_internal/distributed/_shard/sharded_tensor/__init__.py support for hccl backend Pull Request resolved: https://github.com/pytorch/pytorch/pull/156796 Approved by: https://github.com/guangyey, https://github.com/ezyang		2025-07-16 09:37:03 +00:00
..
_internal	add device generalization support for distributed tests (#156796)	2025-07-16 09:37:03 +00:00
__init__.py
_comparison.py	[Torch] Fix error message formatting in fp8 comparison logic (#153647)	2025-05-27 02:51:05 +00:00
_creation.py	PEP585 update - torch/testing (#145200)	2025-01-20 22:42:42 +00:00
_utils.py	Fix code descriptions in the test package. (#148145)	2025-03-04 19:14:41 +00:00