[Bugfix] Handle PackageNotFoundError when checking for xpu version (#7398)

91294d56 · sasha0552 · GitHub · a046f863 · 91294d56
Unverified Commit 91294d56 authored Aug 12, 2024 by sasha0552 Committed by GitHub Aug 12, 2024
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

vllm/utils.py vllm/utils.py +5 -2

No files found.
--- a/vllm/utils.py
+++ b/vllm/utils.py
@@ -344,8 +344,11 @@ def is_tpu() -> bool:

 @lru_cache(maxsize=None)
 def is_xpu() -> bool:
-    from importlib.metadata import version
+    from importlib.metadata import PackageNotFoundError, version
+    try:
        is_xpu_flag = "xpu" in version("vllm")
+    except PackageNotFoundError:
+        return False
    # vllm is not build with xpu
    if not is_xpu_flag:
        return False