Unverified Commit 5430e110 authored by Micah Williamson's avatar Micah Williamson Committed by GitHub
Browse files

[CI][AMD] Match Main CI Behavior By Skipping test_eplb_spec_decode In AMD CI (#30006)


Signed-off-by: default avatarMicah Williamson <micah.williamson@amd.com>
parent 3f1b0373
...@@ -6,6 +6,7 @@ import lm_eval ...@@ -6,6 +6,7 @@ import lm_eval
import pytest import pytest
from tests.utils import large_gpu_mark from tests.utils import large_gpu_mark
from vllm.platforms import current_platform
def get_model_args( def get_model_args(
...@@ -45,6 +46,12 @@ def get_model_args( ...@@ -45,6 +46,12 @@ def get_model_args(
return model_args return model_args
pytestmark = pytest.mark.skipif(
current_platform.is_rocm(),
reason="EPLB with Spec Decode is a work in progress on ROCm.",
)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model_setup", "model_setup",
[ [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment