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,