Unverified Commit 9a0d2f0d authored by Zhewen Li's avatar Zhewen Li Committed by GitHub
Browse files

[CI/Build] Skip cpu offloading test on AMD (#27690)


Signed-off-by: default avatarzhewenli <zhewenli@meta.com>
parent ad3ec895
......@@ -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:
"""
......
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