[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:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user