[Hardware][openvino] is_openvino --> current_platform.is_openvino (#9716)

2024-10-26 18:59:06 +08:00
parent 067e77f9a8
commit 5cbdccd151
10 changed files with 69 additions and 38 deletions
--- a/tests/kernels/test_attention_selector.py
+++ b/tests/kernels/test_attention_selector.py
@@ -30,7 +30,8 @@ def test_env(name: str, device: str, monkeypatch):
                                        False)
        assert backend.name == "ROCM_FLASH"
    elif device == "openvino":
-        with patch("vllm.attention.selector.is_openvino", return_value=True):
+        with patch("vllm.attention.selector.current_platform.is_openvino",
+                   return_value=True):
            backend = which_attn_to_use(16, torch.float16, torch.float16, 16,
                                        False)
        assert backend.name == "OPENVINO"