ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] FMHA BWD Fix Compilation with Bias (#2682)

Browse Source 
* [CK_TILE] FMHA BWD Fix Compilation with Bias

* Fix appendkv kApplyRoPE
This commit is contained in:
Yi DING
2025-08-22 10:01:10 +08:00
committed by GitHub
parent 49c6b05c72
commit 4cfa2c7158
5 changed files with 28 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
include/ck_tile/ops/fmha/kernel/fmha_fwd_appendkv_kernel.hpp
View File

@@ -649,8 +649,12 @@ struct FmhaFwdAppendKVKernel
{0, i_n0});
// If kApplyRoPe is false, we set the rotary_dim to 0
auto rotary_dim = kApplyRoPE ? kargs.rotary_dim : 0;
auto rotary_dim = [&]() {
if constexpr(kApplyRoPE)
return kargs.rotary_dim;
else
return 0;
}();
FmhaPipeline{}(q_dram_window,
k_dram_window,
i_page_block_k,