[Platforms] Add device_type in Platform (#10508)

Signed-off-by: MengqingCao <cmq0113@163.com>
2024-11-21 12:44:20 +08:00
parent 6c1208d083
commit 9d827170a3
10 changed files with 11 additions and 15 deletions
--- a/vllm/platforms/xpu.py
+++ b/vllm/platforms/xpu.py
@@ -16,6 +16,7 @@ logger = init_logger(__name__)

 class XPUPlatform(Platform):
    _enum = PlatformEnum.XPU
+    device_type: str = "xpu"

    @classmethod
    def get_default_attn_backend(cls, selected_backend: _Backend) -> _Backend: