From e587e26b06d4ac8d7dc76e5faaa2e786d40772a2 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 12:01:28 +0000
Subject: [PATCH] test: log canonical indices we write Q to

---
 tests/unit/test_umma_qk_hd64.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/unit/test_umma_qk_hd64.cu b/tests/unit/test_umma_qk_hd64.cu
index 1ead976f..d0f53eb9 100644
--- a/tests/unit/test_umma_qk_hd64.cu
+++ b/tests/unit/test_umma_qk_hd64.cu
@@ -58,6 +58,8 @@ test_umma_qk_hd64(const bf16_t* q, const bf16_t* k,
         int core_k = d / 8, local_c = d % 8;
         int idx = core_k * 16 * 64 + local_c;  // tile_mn=0, local_r=0
         sQ[idx] = q[d];
+        // Also write to flat offset for verification
+        if (d < 8) s_out[220 + d] = (float)idx;  // Log the indices we write to
     }
     // Write K (sk, hd) to sK in canonical layout
     for (int i = tid; i < sk * hd; i += N_WARPS * 32) {