Unverified Commit 988ee66b authored by jinghanhu's avatar jinghanhu Committed by GitHub
Browse files

Handle triton kernel import exception (#29062)

parent ea38474a
......@@ -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,
)
......
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