diff --git a/tests/v1/kv_connector/unit/test_multi_connector.py b/tests/v1/kv_connector/unit/test_multi_connector.py index 1c1ac915c758..ffa7d884d276 100644 --- a/tests/v1/kv_connector/unit/test_multi_connector.py +++ b/tests/v1/kv_connector/unit/test_multi_connector.py @@ -20,6 +20,7 @@ from vllm.distributed.kv_transfer.kv_connector.v1.multi_connector import ( from vllm.distributed.kv_transfer.kv_connector.v1.nixl_connector import ( NixlKVConnectorStats, ) +from vllm.platforms import current_platform MODEL_NAME = "meta-llama/Llama-3.2-1B-Instruct" @@ -69,6 +70,13 @@ def _compare_directories(dir1: Path, dir2: Path) -> bool: return True +@pytest.mark.skipif( + current_platform.is_rocm(), + reason=( + "hipErrorLaunchFailure when running this test, see issue:" + "https://github.com/ROCm/pytorch/issues/2822" + ), +) def test_multi_shared_storage_connector_consistency(): """ Tests that MultiConnector with two SharedStorageConnectors saves