ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

tempsave, fmha_decode

Browse Source
This commit is contained in:
aska-0096
2025-07-08 08:37:20 +00:00
parent 47565f21a5
commit 18686cfe5b
13 changed files with 2562 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1140
include/ck_tile/ops/fmha/kernel/fmha_fwd_decode_kernel.hpp Normal file
View File

File diff suppressed because it is too large Load Diff

10
include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp
View File

@@ -1071,20 +1071,20 @@ struct FmhaFwdSplitKVKernel
{
return FmhaPipeline{}(q_dram_window,
k_dram_window_lengths,
k_page_block_navigator,
// k_page_block_navigator,
v_dram_window_lengths,
v_page_block_navigator,
// v_page_block_navigator,
bias_dram_window,
lse_acc_dram_window,
kargs.num_splits,
i_split_,
// kargs.num_splits,
// i_split_,
mask,
position_encoding,
kargs.scale_s,
variant,
variant_params,
block_indices,
kv_l2p_offset,
// kv_l2p_offset,
smem_ptr);
}
}();