Skip to content

Commit bdbd417

Browse files
committed
Open pyhccl distributed e2e test.
Signed-off-by: leo-pony <nengjunma@outlook.com>
1 parent f5404dc commit bdbd417

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/e2e/multicard/test_pyhccl_distributed.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,8 @@ def worker_fn():
7676
assert torch.all(tensor == pynccl_comm.world_size).cpu().item()
7777

7878

79-
# def test_pyhccl():
80-
# distributed_run(worker_fn, 2)
79+
def test_pyhccl():
80+
distributed_run(worker_fn, 2)
8181

8282

8383
@worker_fn_wrapper
@@ -106,5 +106,5 @@ def broadcast_worker_fn():
106106
assert torch.all(recv_tensors[i] == i).cpu().item()
107107

108108

109-
# def test_pyhccl_broadcast():
110-
# distributed_run(broadcast_worker_fn, 4)
109+
def test_pyhccl_broadcast():
110+
distributed_run(broadcast_worker_fn, 2)

0 commit comments

Comments
 (0)