ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-01 04:07:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix q window

Browse Source
This commit is contained in:
Juuso Korhonen
2025-10-14 09:49:54 +00:00
parent b37c356090
commit c3d27abfb8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/ck_tile/ops/unified_attention/kernel/unified_attention_kernel.hpp
View File

@@ -376,8 +376,8 @@ struct FmhaFwdV3Kernel
// stride for dim 0 (num_queries_per_kv * head_dim, head_dim, 1)
auto q_dram_window = make_tile_window(
q_dram,
make_tuple(BLOCK_Q, HEAD_SIZE_PADDED),
{0, 0}
make_tuple(BLOCK_Q * num_queries_per_kv, HEAD_SIZE_PADDED),
{query_pos * num_queries_per_kv, 0}
);
const auto k_dram = [&]() {