[CI/Build] Enable test_kv_cache_events_dp for AMD (#31834)

Signed-off-by: Ryan Rock <ryan.rock@amd.com>
This commit is contained in:
Ryan Rock
2026-01-08 03:00:24 -06:00
committed by GitHub
parent b634e619bb
commit 8cbdc7eb94

View File

@@ -37,8 +37,11 @@ from vllm.v1.executor.abstract import Executor
from ...distributed.conftest import MockSubscriber
from ...utils import create_new_process_for_each_test
if not current_platform.is_cuda():
pytest.skip(reason="V1 currently only supported on CUDA.", allow_module_level=True)
if not current_platform.is_cuda_alike():
pytest.skip(
reason="V1 currently only supported on CUDA-alike platforms.",
allow_module_level=True,
)
MODEL_NAME = "meta-llama/Llama-3.2-1B-Instruct"
TOKENIZER = AutoTokenizer.from_pretrained(MODEL_NAME)