Handle triton kernel import exception (#25319)

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

Pull Request resolved: #25319

Handle triton kernel import exception in cases it is not available.

Test Plan: AMD DeepSeek test

Reviewed By: houseroad

Differential Revision:
D82899108

Privacy Context Container: L1370295

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,