From 7e6f4e1d8e44fd518bd0bf194fd6a84fc75a849b Mon Sep 17 00:00:00 2001 From: Sy03 <1370724210@qq.com> Date: Sat, 3 Oct 2026 02:46:08 +0800 Subject: [PATCH] [Test] Align CosyVoice3 hardware mocks across platform layers Signed-off-by: Sy03 <1370724210@qq.com> --- tests/config/test_config_factory.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/config/test_config_factory.py b/tests/config/test_config_factory.py index fe870a7c1..2d0de7baa 100644 --- a/tests/config/test_config_factory.py +++ b/tests/config/test_config_factory.py @@ -589,7 +589,10 @@ class TestPipelineDiscovery: from vllm.platforms import current_platform from vllm.platforms.interface import DeviceCapability + from vllm_omni.platforms import current_omni_platform + monkeypatch.setattr(current_platform, "is_cuda", lambda: cuda) + monkeypatch.setattr(current_omni_platform, "device_name", "cuda" if cuda else "cpu") capability = DeviceCapability(major, 0) if major is not None else None monkeypatch.setattr(current_platform, "get_device_capability", lambda: capability) monkeypatch.setattr(current_platform, "get_device_total_memory", lambda: memory_gib * 1024**3)