Unverified Commit a4fdf240 authored by rasmith's avatar rasmith Committed by GitHub
Browse files

[CI/Build] Skip tests that require libcudart in test_lmcache_integration.py (#29228)


Signed-off-by: default avatarRandall Smith <ransmith@amd.com>
Co-authored-by: default avatarRandall Smith <ransmith@amd.com>
parent e6309acd
...@@ -9,6 +9,12 @@ ...@@ -9,6 +9,12 @@
# Assumption vs. Correctness Tests: # Assumption vs. Correctness Tests:
# these unit tests do *not* test correctness of LMCache-side or vLLM-side logic # these unit tests do *not* test correctness of LMCache-side or vLLM-side logic
# it is to ensure that assumptions LMCache makes about vLLM's interface are stable # it is to ensure that assumptions LMCache makes about vLLM's interface are stable
import pytest
from vllm.platforms import current_platform
def assumes(obj, attr, is_callable=False, is_instance_of=None): def assumes(obj, attr, is_callable=False, is_instance_of=None):
import inspect import inspect
from dataclasses import is_dataclass from dataclasses import is_dataclass
...@@ -48,6 +54,9 @@ def assumes(obj, attr, is_callable=False, is_instance_of=None): ...@@ -48,6 +54,9 @@ def assumes(obj, attr, is_callable=False, is_instance_of=None):
assert isinstance(attr_value, is_instance_of), assumption_msg assert isinstance(attr_value, is_instance_of), assumption_msg
@pytest.mark.skipif(
current_platform.is_rocm(), reason="Requires libcudart.so, not available on ROCm"
)
def test_multimodal_interface(): def test_multimodal_interface():
# protect against interface changes # protect against interface changes
from vllm.multimodal.inputs import PlaceholderRange from vllm.multimodal.inputs import PlaceholderRange
...@@ -72,6 +81,9 @@ def test_multimodal_interface(): ...@@ -72,6 +81,9 @@ def test_multimodal_interface():
assert token_ids.tolist() == [0, 0, 0, 0, 4, 4369, 4369, 4369, 4369, 9] assert token_ids.tolist() == [0, 0, 0, 0, 4, 4369, 4369, 4369, 4369, 9]
@pytest.mark.skipif(
current_platform.is_rocm(), reason="Requires libcudart.so, not available on ROCm"
)
def test_config_interface(): def test_config_interface():
# protect against interface changes # protect against interface changes
from vllm.config import VllmConfig from vllm.config import VllmConfig
...@@ -146,6 +158,9 @@ def test_config_interface(): ...@@ -146,6 +158,9 @@ def test_config_interface():
) )
@pytest.mark.skipif(
current_platform.is_rocm(), reason="Requires libcudart.so, not available on ROCm"
)
def test_request_interface(): def test_request_interface():
# protect against interface changes # protect against interface changes
from types import NoneType from types import NoneType
......
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