Skip to content

Commit 655a229

Browse files
authored
[TEST]Add MALPO for aclgraph in nightly test (#3894)
### What this PR does / why we need it? This PR adds MALPO for deepseek aclgraph, we need to test it nightly ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? By running the test - vLLM version: v0.11.0 - vLLM main: vllm-project/vllm@83f478b --------- Signed-off-by: jiangyunfan1 <[email protected]>
1 parent 216fc0e commit 655a229

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/e2e/nightly/models/test_deepseek_r1_0528_w8a8.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
"torchair",
3333
"single",
3434
"aclgraph",
35+
"aclgraph_mlapo",
3536
"no_chunkprefill",
3637
]
3738

@@ -108,6 +109,9 @@ async def test_models(model: str, mode: str) -> None:
108109
additional_config["torchair_graph_config"] = {"enabled": False}
109110
if mode == "aclgraph":
110111
additional_config["torchair_graph_config"] = {"enabled": False}
112+
if mode == "aclgraph_mlapo":
113+
env_dict["VLLM_ASCEND_ENABLE_MLAPO"] = "1"
114+
additional_config["torchair_graph_config"] = {"enabled": False}
111115
if mode == "no_chunkprefill":
112116
additional_config["ascend_scheduler_config"] = {"enabled": True}
113117
i = server_args.index("--max-num-batched-tokens") + 1

0 commit comments

Comments
 (0)