From 13998da15a4a15989ddda179966b0d6ff5054954 Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Tue, 14 Jul 2026 23:58:13 +0800
Subject: [PATCH] fix: uninitialized strides in decode test and wrong stride
 helper in paged test
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- decode test main() missing set_default_strides → illegal memory access
- paged test used set_default_strides on PagedAttentionParams → compile error
---
 csrc/tests/attn_decode_test.cu       | 1 +
 csrc/tests/attn_paged_decode_test.cu | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/csrc/tests/attn_decode_test.cu b/csrc/tests/attn_decode_test.cu
index 780dcb3..221f505 100644
--- a/csrc/tests/attn_decode_test.cu
+++ b/csrc/tests/attn_decode_test.cu
@@ -163,6 +163,7 @@ int main() {
         p.batch=B; p.q_head=Hq; p.kv_head=Hk; p.q_len=1; p.kv_len=sl; p.head_dim=D;
         p.use_mask=0; p.causal_offset=-1;
         p.scale=1.0f/sqrtf((float)D);
+        set_default_strides(p);
         p.q=dQ; p.k=dK; p.v=dV; p.mask=nullptr; p.o=dO;
 
         // Split-K scratch (max 32 splits), sized for the production MMA path.
diff --git a/csrc/tests/attn_paged_decode_test.cu b/csrc/tests/attn_paged_decode_test.cu
index 33590f3..fd1ade9 100644
--- a/csrc/tests/attn_paged_decode_test.cu
+++ b/csrc/tests/attn_paged_decode_test.cu
@@ -145,7 +145,7 @@ static int run_test(int B, int Hq, int Hkv, int kv_len, int page_size, int seed)
     p.batch = B; p.q_head = Hq; p.kv_head = Hkv; p.q_len = 1;
     p.kv_len = kv_len; p.head_dim = HEAD_DIM;
     p.use_mask = 0; p.causal_offset = -1;
-    set_default_strides(p);
+    set_default_paged_strides(p);
     p.num_splits = 1; p.scale = scale_val;
     p.page_size = page_size; p.max_pages = max_pages;
     p.page_table = d_pt;