biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[misc] Rename Ray ADAG to Compiled Graph (#13928)

Browse Source
This commit is contained in:
Rui Qiao
2025-02-26 20:03:28 -08:00
committed by GitHub
parent ca377cf1b9
commit c9944acbf9
6 changed files with 22 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/basic_correctness/test_basic_correctness.py
View File

@@ -117,7 +117,7 @@ def test_models_distributed(
pytest.skip(f"Skip test for {test_suite}")
if model == "meta-llama/Llama-3.2-1B-Instruct" and distributed_executor_backend == "ray" and attention_backend == "" and test_suite == "L4": # noqa
# test ray adag
# test Ray Compiled Graph
os.environ['VLLM_USE_RAY_SPMD_WORKER'] = "1"
os.environ['VLLM_USE_RAY_COMPILED_DAG'] = "1"

2
tests/basic_correctness/test_chunked_prefill.py
View File

@@ -93,7 +93,7 @@ def test_models_distributed(
if (model == "meta-llama/Llama-3.2-1B-Instruct"
and distributed_executor_backend == "ray"):
# test ray adag
# test Ray Compiled Graph
os.environ['VLLM_USE_RAY_SPMD_WORKER'] = "1"
os.environ['VLLM_USE_RAY_COMPILED_DAG'] = "1"