diff --git a/cutedsl/bridge.py b/cutedsl/bridge.py index 2a0c2154..1c15d924 100644 --- a/cutedsl/bridge.py +++ b/cutedsl/bridge.py @@ -26,8 +26,8 @@ import cutlass.torch as cutlass_torch from cutedsl.kernel.moe.torch_scaled_grouped_mm import ( ScaledGroupedGemmKernel, utils, + ceil_div, ) -from cutedsl.kernel.moe.moe_utils import ceil_div # ── Constants ──────────────────────────────────────────────────────────