From 6a4f52cedc614cca26ec04173b7696a1faba9a6b Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Fri, 15 May 2026 05:31:13 +0000
Subject: [PATCH] god dam i just want the gemm in

---
 src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/kernel.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/kernel.py b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/kernel.py
index 5396adfe..f1262d77 100644
--- a/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/kernel.py
+++ b/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm/kernel.py
@@ -76,7 +76,7 @@ def cutlass_grouped_nvfp4_gemm(
         M_expert = token_indices.shape[0]
         
         # DEBUG: verify data going into GEMM
-        if e == 0:
+        if e < 3 and M_expert > 0:
             print(f"[GEMM-IN] expert={e} M={M_expert} N={N} K={K} "
                   f"w shape={expert_w.shape} w_sf shape={expert_w_sf.shape} "
                   f"w absmax={expert_w.view(torch.int8).abs().max().item()} "