Unverified Commit 690f948e authored by Gregory Shtrasberg's avatar Gregory Shtrasberg Committed by GitHub
Browse files

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


Signed-off-by: default avatarGregory Shtrasberg <Gregory.Shtrasberg@amd.com>
parent 08275ec0
...@@ -23,7 +23,7 @@ if has_triton_kernels(): ...@@ -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)
......
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