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

Remove necessary value-overwrite logic

Browse Source
This commit is contained in:
PoYen, Chen
2024-06-12 03:07:32 +00:00
parent 318b2d5c12
commit ba0bc1507c
1 changed files with 1 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_combine_pipeline.hpp
View File

@@ -127,14 +127,7 @@ struct BlockFmhaFwdSplitKVCombinePipeline
auto offset =
lse_acc_lds_ms_m0_for_write.calculate_offset(make_tuple(col, row));
if(col < num_splits)
{
lse_accum(distributed_indices) = lse_acc_lds_ptr[offset];
}
else
{
lse_accum(distributed_indices) = -numeric<LSEDataType>::infinity();
}
lse_accum(distributed_indices) = lse_acc_lds_ptr[offset];
});
});
}
@@ -298,7 +291,6 @@ struct BlockFmhaFwdSplitKVCombinePipeline
const OaccDramBlockWindow& o_acc_dram_block_window,
LSEDramBlockWindow& lse_dram_block_window,
index_t num_splits,
index_t seqlen_q,
index_t max_seqlen_q,
void* smem_ptr) const
{