update test case

Chao1Han · Nov 15, 2024 · 3e97e67 · 3e97e67
1 parent 65e0d9d
commit 3e97e67
Show file tree

Hide file tree

Showing 2 changed files with 1,399 additions and 29 deletions.
diff --git a/test/distributed/test_c10d_ops_xccl.py b/test/distributed/test_c10d_ops_xccl.py
@@ -44,6 +44,7 @@
 
 TEST_MULTIGPU = TEST_XPU and torch.xpu.device_count() >= 2
 
+
 class ProcessGroupXCCLOpTest(MultiProcContinousTest):
     @classmethod
     def backend_str(cls) -> str:
@@ -256,7 +257,6 @@ def reduce(xs, rootRank, rootTensor, op=None):
                 ):
                     reduce(tensors, self.rank, rt, op)
 
-
     @requires_xccl()
     @skip_but_pass_in_sandcastle_if(not TEST_MULTIGPU, "XCCL test requires 2+ GPUs")
     def test_allgather_ops(self):
@@ -710,7 +710,6 @@ def perm(n, k):
         expected = torch.tensor(prod_val)
         self.assertEqual(expected, output_tensor)
 
-
     @requires_xccl()
     @skip_but_pass_in_sandcastle_if(not TEST_MULTIGPU, "XCCL test requires 2+ GPUs")
     def test_reduce_scatter_base_ops(self):
@@ -830,4 +829,3 @@ def test_send_recv_object_list(self):
             nprocs=world_size,
             args=(world_size, rdvz_file),
         )
-