Handle triton kernel import exception (#29062)
This commit is contained in:
@@ -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,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user