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

Browse Source
This commit is contained in:
Juuso Korhonen
2025-10-13 10:30:11 +00:00
parent af94aaf1cb
commit b721f79f99
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -354,15 +354,13 @@ struct FmhaFwdV3Kernel
q_dram_pad,
make_tuple(
make_merge_transform(
make_tuple(seq_len, num_queries_per_kv)
make_tuple(seq_len_padded, num_queries_per_kv)
),
make_pass_through_transform(HEAD_SIZE_PADDED)
),
make_tuple(sequence<0, 1>{}, sequence<2>{}),
make_tuple(sequence<0>{}, sequence<1>{})
);
// TODO are we padding the tensor view or the block here?
); // flattens the first two dims, head dim is the fastest changing dim in the merged dim
return q_dram_merged;
}();