Skip to content

Commit 237fad6

Browse files
MrZ20MengqingCao
andauthored
[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: vllm-project/vllm@ad32e3e Signed-off-by: MrZ20 <[email protected]> Co-authored-by: Mengqing Cao <[email protected]>
1 parent 6063853 commit 237fad6

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,15 +102,13 @@ def test_mtp2_correctness_eager(
102102
mtp_correctness(sampling_config, model_name, 2, enforce_eager=True)
103103

104104

105-
@pytest.mark.skip("TODO(cmq): Revert me when mtp aclgraph is fixed")
106105
def test_mtp1_correctness_piecewise_graph(
107106
sampling_config: SamplingParams,
108107
model_name: str,
109108
):
110109
mtp_correctness(sampling_config, model_name, 1)
111110

112111

113-
@pytest.mark.skip("TODO(cmq): Revert me when mtp aclgraph is fixed")
114112
def test_mtp2_correctness_piecewise_graph(
115113
sampling_config: SamplingParams,
116114
model_name: str,

0 commit comments

Comments
 (0)