xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-30 12:37:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: Micah Williamson <micah.williamson@amd.com>
This commit is contained in:
Micah Williamson 2025-12-04 02:20:54 -06:00 committed by GitHub
parent 3f1b03739a
commit 5430e110c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/distributed/test_eplb_spec_decode.py
View File

@ -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(
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",
[ [