Skip to content

Commit 5aa9b09

Browse files
rasmithRandall Smith
andauthored
[CI/Build][AMD] Skip test_shared_storage_connector_hashes in test_shared_storage_connector.py due to hipErrorLaunchFailure when calling .cpu() (vllm-project#29839)
Signed-off-by: Randall Smith <ransmith@amd.com> Co-authored-by: Randall Smith <ransmith@amd.com>
1 parent 1bb17ec commit 5aa9b09

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/v1/kv_connector/unit/test_shared_storage_connector.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,14 @@
33
from dataclasses import asdict
44
from typing import NamedTuple
55

6+
import pytest
67
from PIL import Image
78

89
from vllm import LLM, EngineArgs, SamplingParams
910
from vllm.assets.image import ImageAsset
1011
from vllm.config import KVTransferConfig
1112
from vllm.multimodal.utils import encode_image_base64
13+
from vllm.platforms import current_platform
1214

1315
MODEL_NAME = "RedHatAI/Qwen2.5-VL-3B-Instruct-quantized.w8a8"
1416

@@ -108,6 +110,13 @@ def process_prompt(processor, llm: LLM, question: str, image_urls: list[Image]):
108110
print("-" * 50)
109111

110112

113+
@pytest.mark.skipif(
114+
current_platform.is_rocm(),
115+
reason=(
116+
"hipErrorLaunchFailure when running this test, see issue:"
117+
"https://github.com/ROCm/pytorch/issues/2822"
118+
),
119+
)
111120
def test_shared_storage_connector_hashes(tmp_path):
112121
"""
113122
Tests that SharedStorageConnector saves KV to the storage locations

0 commit comments

Comments
 (0)