[CI] Fix test on pyhccl to 2 cards (#2094)
### What this PR does / why we need it?
Fix test on pyhccl to 2 cards
### Does this PR introduce _any_ user-facing change?
N/A
### How was this patch tested?
CI passed with existing test.
- vLLM version: v0.10.0
- vLLM main:
0d0cc9e150
Signed-off-by: MengqingCao <cmq0113@163.com>
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user