diff --git a/tests/e2e/multicard/test_pyhccl_distributed.py b/tests/e2e/multicard/test_pyhccl_distributed.py index e3d9aed..2300e0a 100644 --- a/tests/e2e/multicard/test_pyhccl_distributed.py +++ b/tests/e2e/multicard/test_pyhccl_distributed.py @@ -89,7 +89,7 @@ def worker_fn(): def test_pyhccl(): - distributed_run(worker_fn, 4) + distributed_run(worker_fn, 2) def broadcast_worker_fn(): @@ -118,4 +118,4 @@ def broadcast_worker_fn(): def test_pyhccl_broadcast(): - distributed_run(broadcast_worker_fn, 4) + distributed_run(broadcast_worker_fn, 2)