diff --git a/vllm/triton_utils/importing.py b/vllm/triton_utils/importing.py index ef7ca149266b6..36315abcdfcda 100644 --- 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.")