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

Fix wrong LDS indexing logics

Browse Source
This commit is contained in:
PoYen, Chen
2024-06-09 15:08:28 +00:00
parent d74a1d6ed1
commit c36cad2e6c
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/fmha/pipeline/block_fmha_fwd_splitkv_combine_pipeline.hpp
View File

@@ -172,11 +172,11 @@ struct BlockFmhaFwdSplitKVCombinePipeline
#if defined(PRINT_LSE_ACCUM)
DEBUG_STMTS
{
printf("\n");
printf("(%2d, %2d)\n", num_splits, kM0);
for(index_t row = 0; row < num_splits; ++row)
{
printf("[POYENC][DEVICE] lse_acc[%2d] = ", row);
for(index_t col = 0; col < real_seqlen_q; ++col)
for(index_t col = 0; col < kM0; ++col)
{
printf("%11.7f", lse_acc_lds_ptr[row + col * kMaxSplits]);
}