From 655a22945573323a975cd786d8cef4224ea154bb Mon Sep 17 00:00:00 2001 From: jiangyunfan1 Date: Thu, 30 Oct 2025 18:25:54 +0800 Subject: [PATCH] [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: https://github.com/vllm-project/vllm/commit/83f478bb19489b41e9d208b47b4bb5a95ac171ac --------- Signed-off-by: jiangyunfan1 --- tests/e2e/nightly/models/test_deepseek_r1_0528_w8a8.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/e2e/nightly/models/test_deepseek_r1_0528_w8a8.py b/tests/e2e/nightly/models/test_deepseek_r1_0528_w8a8.py index 38f54a04..c9126577 100644 --- a/tests/e2e/nightly/models/test_deepseek_r1_0528_w8a8.py +++ b/tests/e2e/nightly/models/test_deepseek_r1_0528_w8a8.py @@ -32,6 +32,7 @@ MODES = [ "torchair", "single", "aclgraph", + "aclgraph_mlapo", "no_chunkprefill", ] @@ -108,6 +109,9 @@ async def test_models(model: str, mode: str) -> None: additional_config["torchair_graph_config"] = {"enabled": False} if mode == "aclgraph": additional_config["torchair_graph_config"] = {"enabled": False} + if mode == "aclgraph_mlapo": + env_dict["VLLM_ASCEND_ENABLE_MLAPO"] = "1" + additional_config["torchair_graph_config"] = {"enabled": False} if mode == "no_chunkprefill": additional_config["ascend_scheduler_config"] = {"enabled": True} i = server_args.index("--max-num-batched-tokens") + 1