Handle triton kernel import exception (#29062)

This commit is contained in:
jinghanhu 2025-11-22 18:07:50 +08:00 committed by GitHub
parent ea38474ac5
commit 988ee66b0d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -28,10 +28,11 @@ logger = init_logger(__name__)
if has_triton_kernels():
try:
from triton_kernels.matmul_ogs import PrecisionConfig
except ImportError:
except (ImportError, AttributeError) as e:
logger.error(
"Failed to import Triton kernels. Please make sure your triton "
"version is compatible."
"version is compatible. Error: %s",
e,
)