diff --git a/cutedsl/bridge.py b/cutedsl/bridge.py index f8807c8f..2a0c2154 100644 --- a/cutedsl/bridge.py +++ b/cutedsl/bridge.py @@ -21,7 +21,7 @@ import threading import torch import cutlass import cutlass.cute as cute -import cutlass_torch +import cutlass.torch as cutlass_torch from cutedsl.kernel.moe.torch_scaled_grouped_mm import ( ScaledGroupedGemmKernel,