Skip to content

Commit af31fec

Browse files
committed
add ray test
Signed-off-by: MengqingCao <cmq0113@163.com>
1 parent a08f3cf commit af31fec

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/e2e/multicard/test_pipeline_parallel.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424

2525
TENSOR_PARALLELS = [2]
2626
PIPELINE_PARALLELS = [2]
27+
DIST_EXECUTOR_BACKEND = ["mp", "ray"]
2728

2829
prompts = [
2930
"Hello, my name is",
@@ -34,10 +35,13 @@
3435
@pytest.mark.parametrize("model", MODELS)
3536
@pytest.mark.parametrize("tp_size", TENSOR_PARALLELS)
3637
@pytest.mark.parametrize("pp_size", PIPELINE_PARALLELS)
37-
def test_models(model: str, tp_size: int, pp_size: int) -> None:
38+
@pytest.mark.parametrize("distributed_executor_backend", DIST_EXECUTOR_BACKEND)
39+
def test_models(model: str, tp_size: int, pp_size: int,
40+
distributed_executor_backend: str) -> None:
3841
with VllmRunner(model,
3942
tensor_parallel_size=tp_size,
4043
pipeline_parallel_size=pp_size,
44+
distributed_executor_backend=distributed_executor_backend,
4145
enforce_eager=True,
4246
gpu_memory_utilization=0.7) as vllm_model:
4347
vllm_model.generate_greedy(prompts, 64)

0 commit comments

Comments
 (0)