test: zero 2 TMEM columns

2026-05-28 09:28:31 +00:00
parent 01c4097ccc
commit 25c9b70591
1 changed files with 1 additions and 1 deletions
--- 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();