From 25c9b70591cdbbb7ed31dfa5337c0503b8c24704 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 09:28:31 +0000
Subject: [PATCH] test: zero 2 TMEM columns

---
 tests/unit/test_umma_qk.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/test_umma_qk.cu b/tests/unit/test_umma_qk.cu
index d62f472b..648618dd 100644
--- a/tests/unit/test_umma_qk.cu
+++ b/tests/unit/test_umma_qk.cu
@@ -80,7 +80,7 @@ test_umma_qk_hd16(
 
     // Zero TMEM — test with 32 columns first
     if (wid == 0) {
-        for (int col = 0; col < 32; col++) {
+        for (int col = 0; col < 2; col++) {
             tmem_store(tmem_base + col, 0, 0, 0, 0);
         }
         tmem_fence_store();