diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index 7a1f38606062..f77111dca523 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -284,6 +284,7 @@ steps: - label: V1 Test others # 42min timeout_in_minutes: 60 + fast_check: true mirror_hardwares: [amdexperimental] source_file_dependencies: - vllm/ diff --git a/tests/v1/kv_connector/unit/test_shared_storage_connector.py b/tests/v1/kv_connector/unit/test_shared_storage_connector.py index 6be261e45cb0..3151c82a46e3 100644 --- a/tests/v1/kv_connector/unit/test_shared_storage_connector.py +++ b/tests/v1/kv_connector/unit/test_shared_storage_connector.py @@ -8,6 +8,8 @@ from vllm import LLM, EngineArgs, SamplingParams from vllm.assets.image import ImageAsset from vllm.config import KVTransferConfig +from vllm.distributed.kv_transfer.kv_transfer_state import ( + ensure_kv_transfer_shutdown) from vllm.multimodal.utils import encode_image_base64 MODEL_NAME = "RedHatAI/Qwen2.5-VL-3B-Instruct-quantized.w8a8" @@ -114,9 +116,12 @@ def process_prompt(processor, llm: LLM, question: str, def test_shared_storage_connector_hashes(tmp_path): """ Tests that SharedStorageConnector saves KV to the storage locations - with proper hashes; that are unique for inputs with identical text but + with proper hashes; that are unique for inputs with identical text but different images (same size), or same multiple images but different orders. """ + # Shutdown the KV transfer process if it is still running + ensure_kv_transfer_shutdown() + # Using tmp_path as the storage path to store KV print(f"KV storage path at: {str(tmp_path)}")