saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[pgnccl] skip restart test fro rocm (#139809)

Browse source 
Summary:
PG restart test is flaky in rocm: https://github.com/pytorch/pytorch/pull/139809, skip the AMD/ROCM test for now
Test Plan:
CI

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139809
Approved by: https://github.com/kwen2501
This commit is contained in:
Shuqiang Zhang 2024-11-05 15:16:35 -08:00 committed by PyTorch MergeBot
parent 96ca17fec4
commit d35a600b74
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
test/distributed/test_c10d_nccl.py
View file

@ -348,6 +348,7 @@ class ProcessGroupNCCLGroupTest(MultiProcessTestCase):
dist.all_reduce(t)
@requires_nccl()
@skip_if_rocm_multiprocess
@skip_but_pass_in_sandcastle_if(not TEST_MULTIGPU, "NCCL test requires 2+ GPUs")
def test_restart_pg(self):
# Note: restart test passes steadily only for blocking mode for now.