From b2e071dc52581ca6962e65c5920db4485656ad2b Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Sat, 4 Apr 2026 01:39:26 +0300
Subject: [PATCH] cleanup

---
 modules/sdnq/triton_mm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/sdnq/triton_mm.py b/modules/sdnq/triton_mm.py
index c8fb1a167..cd04f6631 100644
--- a/modules/sdnq/triton_mm.py
+++ b/modules/sdnq/triton_mm.py
@@ -117,7 +117,7 @@ def triton_mm_td_kernel(
 
     off_k = 0
     accumulator = tl.zeros((BLOCK_SIZE_M, BLOCK_SIZE_N), dtype=ACCUMULATOR_DTYPE)
-    for k in range(0, tl.cdiv(K, BLOCK_SIZE_K)):
+    for _ in range(0, K, BLOCK_SIZE_K):
         a = a_desc.load([pid_m * BLOCK_SIZE_M, off_k])
         b = b_desc.load([off_k, pid_n * BLOCK_SIZE_N])
         accumulator = tl.dot(a, b, accumulator, out_dtype=ACCUMULATOR_DTYPE)