From df902c6a065759f2f37402d5cde92dc78aceab8c Mon Sep 17 00:00:00 2001
From: Qianfeng Zhang <Qianfeng.Zhang@amd.com>
Date: Thu, 25 Dec 2025 15:37:22 +0000
Subject: [PATCH] Tiny fix in using v_tiles[] index

---
 .../hstu_attention_with_softmax_fwd_trload_pipeline.hpp         | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp
index ec409d7d0c..3230ad96df 100644
--- a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp
+++ b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp
@@ -552,7 +552,7 @@ struct HstuAttentionWithSoftmaxFwdPipelineQRKSVSTrLoad
                     __builtin_amdgcn_sched_barrier(0x00000001);
 
                     store_tile(v_lds_windows[number<(i_k1 + 3) % NumKVLdsBuffers>{}],
-                               v_tiles[number<i_k1 + 1>{}],
+                               v_tiles[number<(i_k1 + 1) % NumPrefetchV>{}],
                                partition_index);
 
                     __builtin_amdgcn_sched_barrier(0x00000001);