diff --git a/tests/v1/engine/test_engine_core_client.py b/tests/v1/engine/test_engine_core_client.py index 5067fc255..4b9c0e746 100644 --- a/tests/v1/engine/test_engine_core_client.py +++ b/tests/v1/engine/test_engine_core_client.py @@ -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)