From a87f20a4aeabf258dc48c54d60d6637d5e5dfdd8 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 09:29:46 +0000
Subject: [PATCH] test: just 1 tmem_store, no fence, no loop

---
 tests/unit/test_umma_qk.cu | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/tests/unit/test_umma_qk.cu b/tests/unit/test_umma_qk.cu
index d92a6563..82416d16 100644
--- a/tests/unit/test_umma_qk.cu
+++ b/tests/unit/test_umma_qk.cu
@@ -78,12 +78,9 @@ test_umma_qk_hd16(
     }
     __syncthreads();
 
-    // Zero TMEM — test without fence
+    // Zero TMEM — test just 1 store call, no fence
     if (wid == 0) {
-        for (int col = 0; col < 128; col++) {
-            tmem_store(tmem_base + col, 0, 0, 0, 0);
-        }
-        // tmem_fence_store();  // skip fence
+        tmem_store(tmem_base, 0, 0, 0, 0);
     }
     __syncthreads();