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

View File

@@ -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,
) )