pytorch/torch/testing/_internal/distributed
Prachi Gupta f2b44424a1 [ROCm] Skip *_stress_cuda and test_ddp_apply_optim_in_backward* (#155724)
These tests are flaky on ROCm and have been skipped via Github issues, but the bot keeps closing the issues after not observing the failures for these tests in the rerun_disabled_tests runs (not sure why they don't fail there), and we have to keep reopening them.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155724
Approved by: https://github.com/jeffdaily

Co-authored-by: Jithun Nair <37884920+jithunnair-amd@users.noreply.github.com>
2025-06-12 21:18:04 +00:00
..
_shard [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00
_tensor Add TEST_HPU flag to set device type (#153461) 2025-05-14 19:31:40 +00:00
nn [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00
rpc [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00
__init__.py
checkpoint_utils.py
common_state_dict.py [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
ddp_under_dist_autograd_test.py [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00
distributed_test.py [ROCm] Skip *_stress_cuda and test_ddp_apply_optim_in_backward* (#155724) 2025-06-12 21:18:04 +00:00
distributed_utils.py [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00
fake_pg.py [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00
multi_threaded_pg.py [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00
rpc_utils.py [BE][lint] fix PYFMT for PT-D code under torch.testing._internal, add them to the lint list (#153114) 2025-05-08 14:01:49 +00:00