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

Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Gregory Shtrasberg 2025-09-23 17:31:08 -04:00 committed by yewentao256
parent 0438c65376
commit c0f11557e1

View File

@ -23,7 +23,7 @@ if has_triton_kernels():
from triton_kernels.routing import (RoutingData, routing, from triton_kernels.routing import (RoutingData, routing,
routing_from_bitmatrix) routing_from_bitmatrix)
from triton_kernels.tensor import Bitmatrix from triton_kernels.tensor import Bitmatrix
except (ModuleNotFoundError, AttributeError) as e: except (AttributeError, ImportError) 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. Error: %s", e) "version is compatible. Error: %s", e)