pytorch/torch/testing/_internal/distributed
2025-03-17 22:43:15 +00:00
..
_shard [BE][CI] bump ruff to 0.8.4 (#143753) 2024-12-24 12:24:10 +00:00
_tensor Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
nn Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
rpc [BE]: Enable ruff rule SIM113 (#147290) 2025-02-16 22:41:16 +00:00
__init__.py
checkpoint_utils.py [dcp] Add ZStandard transformer (#143360) 2025-01-25 00:14:07 +00:00
common_state_dict.py PEP585: More UP006 fixes (#146392) 2025-02-20 06:18:13 +00:00
ddp_under_dist_autograd_test.py
distributed_test.py Revert "[PGNCCL] Launch kernel on current stream & remove record_stream entirely (#148590)" 2025-03-17 22:43:15 +00:00
distributed_utils.py
fake_pg.py
multi_threaded_pg.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
rpc_utils.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00