Handle triton kernel import exception (vllm-project#25319)

Signed-off-by: Ming Yang <[email protected]>

Author: minosfuture

vllm/model_executor/layers/fused_moe/config.py

@@ -14,11 +14,16 @@
 from vllm.utils import cdiv, has_triton_kernels
 from vllm.utils.flashinfer import has_flashinfer_cutlass_fused_moe
 
-if has_triton_kernels():
-    from triton_kernels.matmul_ogs import PrecisionConfig
-
 logger = init_logger(__name__)
 
+if has_triton_kernels():
+    try:
+        from triton_kernels.matmul_ogs import PrecisionConfig
+    except ImportError:
+        logger.error(
+            "Failed to import Triton kernels. Please make sure your triton "
+            "version is compatible.")
+
 
 def _get_config_dtype_str(
     dtype: torch.dtype,