[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:
ad32e3e19c

Signed-off-by: MrZ20 <2609716663@qq.com>
Co-authored-by: Mengqing Cao <cmq0113@163.com>
This commit is contained in:
SILONG ZENG
2025-12-16 10:40:00 +08:00
committed by GitHub
parent 6063853ead
commit 237fad635c

View File

@@ -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,