[Bugfix] Fix for the import error from #24588 (#25481)

Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
This commit is contained in:
Gregory Shtrasberg 2025-09-23 17:31:08 -04:00 committed by GitHub
parent 08275ec0a2
commit 690f948e4a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -23,7 +23,7 @@ if has_triton_kernels():
from triton_kernels.routing import (RoutingData, routing,
routing_from_bitmatrix)
from triton_kernels.tensor import Bitmatrix
except (ModuleNotFoundError, AttributeError) as e:
except (AttributeError, ImportError) as e:
logger.error(
"Failed to import Triton kernels. Please make sure your triton "
"version is compatible. Error: %s", e)