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