From 237fad635cc34799edd7530dc80ec19bc305e0f0 Mon Sep 17 00:00:00 2001 From: SILONG ZENG <2609716663@qq.com> Date: Tue, 16 Dec 2025 10:40:00 +0800 Subject: [PATCH] [Fix]Revert temporary skip on mtp1/mtp2 correctness tests (aclgraph fix) (#5039) ### What this PR does / why we need it? This Pull Request removes the @pytest.mark.skip decorators from test_mtp1_correctness_piecewise_graph and test_mtp2_correctness_piecewise_graph. These tests were temporarily skipped because of an issue with the MTP ACL Graph (as per the original TODO comment). Since the relevant bug/issue has been resolved, these tests are now re-enabled to ensure full correctness coverage for MTP functionality. - vLLM version: v0.12.0 - vLLM main: https://github.com/vllm-project/vllm/commit/ad32e3e19ccf0526cb6744a5fed09a138a5fb2f9 Signed-off-by: MrZ20 <2609716663@qq.com> Co-authored-by: Mengqing Cao --- tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py b/tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py index 99c7d51c..20f7d145 100644 --- a/tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py +++ b/tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py @@ -102,7 +102,6 @@ def test_mtp2_correctness_eager( mtp_correctness(sampling_config, model_name, 2, enforce_eager=True) -@pytest.mark.skip("TODO(cmq): Revert me when mtp aclgraph is fixed") def test_mtp1_correctness_piecewise_graph( sampling_config: SamplingParams, model_name: str, @@ -110,7 +109,6 @@ def test_mtp1_correctness_piecewise_graph( mtp_correctness(sampling_config, model_name, 1) -@pytest.mark.skip("TODO(cmq): Revert me when mtp aclgraph is fixed") def test_mtp2_correctness_piecewise_graph( sampling_config: SamplingParams, model_name: str,