test: just 1 tmem_store, no fence, no loop

2026-05-28 09:29:46 +00:00
parent 2b57f28968
commit a87f20a4ae
1 changed files with 2 additions and 5 deletions
--- 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();