From c2e41a858ed9d33a56e9e8a75ef863ff9dd9649f Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 28 May 2026 12:09:45 +0000 Subject: [PATCH] test: force 2 K-tiles for debug --- tests/unit/test_umma_qk_hd64.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_umma_qk_hd64.cu b/tests/unit/test_umma_qk_hd64.cu index 6bd49721..74c3252f 100644 --- a/tests/unit/test_umma_qk_hd64.cu +++ b/tests/unit/test_umma_qk_hd64.cu @@ -67,7 +67,7 @@ test_umma_qk_hd64_1ktile(const bf16_t* q, const bf16_t* k, uint32_t idesc = make_idesc(128, 128); // K-tile loop with accumulate - for (int kt = 0; kt < hd / 16; kt++) { + for (int kt = 0; kt < 2; kt++) { // Force 2 K-tiles for debug // K-tile kt: columns [16*kt, 16*kt+16) // In canonical layout, columns start at core_k = 2*kt and 2*kt+1 // Offset = 2*kt * 2048 bytes from matrix base