From 794ed8592ac0acc1e9f6ebe82fada07404172f95 Mon Sep 17 00:00:00 2001 From: Bo Li <110066325+BLOrange-AMD@users.noreply.github.com> Date: Fri, 15 Nov 2024 15:04:58 -0600 Subject: [PATCH] Updated test_c10d_nccl.py to skip unit tests --- test/distributed/test_c10d_nccl.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/distributed/test_c10d_nccl.py b/test/distributed/test_c10d_nccl.py index e869178dd660dc..49141226215adc 100644 --- a/test/distributed/test_c10d_nccl.py +++ b/test/distributed/test_c10d_nccl.py @@ -4045,6 +4045,8 @@ class NCCLTraceTestDumpOnTimeout(NCCLTraceTestDumpOnTimeoutBase): @skip_but_pass_in_sandcastle_if(not TEST_MULTIGPU, "NCCL test requires 2+ GPUs") @parametrize("timing_enabled", [True, False]) def test_timeout_dumps(self, timing_enabled): + if TEST_WITH_ROCM and timing_enabled == False: + self.skipTest(f"Skipped on ROCm") # dump on heartbeatmonitor thread os.environ["TORCH_NCCL_COORD_CHECK_MILSEC"] = "1000" # need rank0 to crash before looking for its output file