fix: correct cutlass_torch import (cutlass.torch, not top-level)

2026-05-16 20:08:21 +00:00
parent 16c4fad025
commit 81632e2f21
1 changed files with 1 additions and 1 deletions
--- 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,