Skip to content

GitLab

".github/vscode:/vscode.git/clone" did not exist on "7cab92fd45ce6ad7fd78d705ef31b12d3beebb4b"
Unverified Commit 988ee66b authored by jinghanhu's avatar jinghanhu Committed by GitHub
Browse files

Handle triton kernel import exception (#29062)

parent ea38474a
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
vllm/model_executor/layers/fused_moe/config.py
View file @ 988ee66b
...@@ -28,10 +28,11 @@ logger = init_logger(__name__) ...@@ -28,10 +28,11 @@ logger = init_logger(__name__)
if has_triton_kernels(): if has_triton_kernels():
try: try:
from triton_kernels.matmul_ogs import PrecisionConfig from triton_kernels.matmul_ogs import PrecisionConfig
except ImportError: except (ImportError, AttributeError) as e:
logger.error( logger.error(
"Failed to import Triton kernels. Please make sure your triton " "Failed to import Triton kernels. Please make sure your triton "
"version is compatible." "version is compatible. Error: %s",
e,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment