From 9a0d2f0d92908c7fcc17eab63a3dfc998aae53a4 Mon Sep 17 00:00:00 2001 From: Zhewen Li Date: Wed, 29 Oct 2025 05:55:51 -0700 Subject: [PATCH] [CI/Build] Skip cpu offloading test on AMD (#27690) Signed-off-by: zhewenli --- tests/v1/kv_offload/test_cpu_offloading.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/v1/kv_offload/test_cpu_offloading.py b/tests/v1/kv_offload/test_cpu_offloading.py index b654ea4298dbb..a5cb23c4ef0f2 100644 --- a/tests/v1/kv_offload/test_cpu_offloading.py +++ b/tests/v1/kv_offload/test_cpu_offloading.py @@ -12,6 +12,7 @@ from tqdm import tqdm from vllm import LLM, SamplingParams, TokensPrompt from vllm.config import KVEventsConfig, KVTransferConfig from vllm.distributed.kv_events import BlockStored, KVEventBatch +from vllm.platforms import current_platform CPU_BLOCK_SIZES = [16, 48] @@ -63,6 +64,9 @@ class MockSubscriber: self.sub.close() +@pytest.mark.skipif( + not current_platform.is_cuda(), reason="CPU offloading only supported on CUDA" +) @pytest.mark.parametrize("cpu_block_size", CPU_BLOCK_SIZES) def test_cpu_offloading(cpu_block_size: int) -> None: """