ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-16 02:54:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Tiny fix in using v_tiles[] index

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-12-25 15:37:22 +00:00
parent 2d53d67b6d
commit df902c6a06
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp
View File

@@ -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);