From bee7cc5f8f077d871faf1e9f5a944c7381542b81 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Fri, 29 May 2026 18:32:21 +0000
Subject: [PATCH] fix: lane vs threadIdx.x in direct QK test

---
 tests/unit/test_qk_direct.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/test_qk_direct.cu b/tests/unit/test_qk_direct.cu
index edc27718..b205e370 100644
--- a/tests/unit/test_qk_direct.cu
+++ b/tests/unit/test_qk_direct.cu
@@ -92,7 +92,7 @@ test_qk_direct_kernel(float* __restrict__ out_s,
     __syncthreads();
 
     const bool my_warp_active = (wid == 0);
-    const int my_row = lane;
+    const int my_row = threadIdx.x % 32;
     const bool my_row_active = my_row < T;
     constexpr int NUM_READS = SK / 8;