[ROCM][CUDA][NCCL] Disable test_lowering_one_shot_all_reduce on ROCM (#139414)

eqy · pytorchmergebot · commit 6fc63b4ef10b · 2024-11-01T18:39:47.000Z
I'm not sure this is expected to run if it requires buffer-registration support CC @yifuwang @huydhn @syed-ahmed #138029 Pull Request resolved: #139414 Approved by: https://github.com/huydhn, https://github.com/yifuwang
diff --git a/test/distributed/test_symmetric_memory.py b/test/distributed/test_symmetric_memory.py
@@ -766,6 +766,7 @@ def _init_process(self):
 
         torch._inductor.config._collective.auto_select = True
 
+    @skipIfRocm  # requires registered-buffer support
     @skip_if_lt_x_gpu(2)
     @fresh_inductor_cache()
     def test_lowering_one_shot_all_reduce(self):
