From 81632e2f21d29f4171166f19211e2b6b22ac8fbf Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 16 May 2026 20:08:21 +0000 Subject: [PATCH] fix: correct cutlass_torch import (cutlass.torch, not top-level) --- cutedsl/bridge.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,