Skip to content

GitLab

Unverified Commit 8cbdc7eb authored by Ryan Rock's avatar Ryan Rock Committed by GitHub
Browse files

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


Signed-off-by: default avatarRyan Rock <ryan.rock@amd.com>
parent b634e619
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
tests/v1/engine/test_engine_core_client.py
View file @ 8cbdc7eb
...@@ -37,8 +37,11 @@ from vllm.v1.executor.abstract import Executor ...@@ -37,8 +37,11 @@ from vllm.v1.executor.abstract import Executor
from ...distributed.conftest import MockSubscriber from ...distributed.conftest import MockSubscriber
from ...utils import create_new_process_for_each_test from ...utils import create_new_process_for_each_test
if not current_platform.is_cuda(): if not current_platform.is_cuda_alike():
pytest.skip(reason="V1 currently only supported on CUDA.", allow_module_level=True) pytest.skip(
reason="V1 currently only supported on CUDA-alike platforms.",
allow_module_level=True,
)
MODEL_NAME = "meta-llama/Llama-3.2-1B-Instruct" MODEL_NAME = "meta-llama/Llama-3.2-1B-Instruct"
TOKENIZER = AutoTokenizer.from_pretrained(MODEL_NAME) TOKENIZER = AutoTokenizer.from_pretrained(MODEL_NAME)
......
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