[XPU] avoid triton import for xpu (vllm-project#9440)

Co-authored-by: Cyrus Leung <[email protected]> Co-authored-by: Cyrus Leung <[email protected]> Signed-off-by: qishuai <[email protected]>
FerdinandZhong · Oct 29, 2024 · 0b9bac0 · 0b9bac0
1 parent 9f4429c
commit 0b9bac0
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/vllm/triton_utils/importing.py b/vllm/triton_utils/importing.py
@@ -5,10 +5,12 @@
 
 logger = init_logger(__name__)
 
-# neuron has too old torch
-HAS_TRITON = find_spec(
-    "triton") is not None and not current_platform.is_neuron()
+HAS_TRITON = (
+    find_spec("triton") is not None
+    and not current_platform.is_xpu()  # Not compatible
+    and not current_platform.is_neuron()  # neuron has too old torch
+)
 
 if not HAS_TRITON:
-    logger.info("Triton not installed; certain GPU-related functions"
-                " will not be available.")
+    logger.info("Triton not installed or not compatible; certain GPU-related"
+                " functions will not be available.")