[DNM] Test triton

Signed-off-by: wangxiyuan <[email protected]>
vllm-project · Dec 23, 2024 · e2f85d1 · e2f85d1
1 parent 048fc57
commit e2f85d1
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/vllm/triton_utils/importing.py b/vllm/triton_utils/importing.py
@@ -5,11 +5,7 @@
 
 logger = init_logger(__name__)
 
-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
-)
+HAS_TRITON = find_spec("triton") is not None
 
 if not HAS_TRITON:
     logger.info("Triton not installed or not compatible; certain GPU-related"