From 22fb861447ff2720a7d0fe108996ab0557350fe0 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 09:30:37 +0000
Subject: [PATCH] test: 2 tmem_stores with syncwarp between

---
 tests/unit/test_umma_qk.cu | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/unit/test_umma_qk.cu b/tests/unit/test_umma_qk.cu
index 82416d16..0097f1ed 100644
--- a/tests/unit/test_umma_qk.cu
+++ b/tests/unit/test_umma_qk.cu
@@ -78,9 +78,12 @@ test_umma_qk_hd16(
     }
     __syncthreads();
 
-    // Zero TMEM — test just 1 store call, no fence
+    // Zero TMEM — test 2 stores with syncwarp between them
     if (wid == 0) {
         tmem_store(tmem_base, 0, 0, 0, 0);
+        __syncwarp();
+        tmem_store(tmem_base + 1, 0, 0, 0, 0);
+        __syncwarp();
     }
     __syncthreads();