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

block_shape fixes

Browse Source
This commit is contained in:
Tianxing Wu
2025-11-12 14:00:51 +00:00
parent 3b56f0f0c3
commit 07bb33866b
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
include/ck_tile/ops/unified_attention/pipeline/unified_attention_pipeline.hpp
View File

@@ -850,11 +850,11 @@ struct UnifiedAttentionPipeline
{
gemm_1(o_acc,
get_slice_tile(sp(sp_reg_idx).p,
sequence<0, (k1_loops - 1) * HEAD_SIZE_PADDED>{},
sequence<BLOCK_M, k1_loops * HEAD_SIZE_PADDED>{}),
sequence<0, (k1_loops - 1) * BLOCK_SIZE>{},
sequence<BLOCK_M, k1_loops * BLOCK_SIZE>{}),
get_slice_tile(kv_tile.v_tile,
sequence<0, (k1_loops - 1) * HEAD_SIZE_PADDED>{},
sequence<BLOCK_SIZE, k1_loops * HEAD_SIZE_PADDED>{}));
sequence<0, (k1_loops - 1) * BLOCK_SIZE>{},
sequence<BLOCK_SIZE, k1_loops * BLOCK_SIZE>{}));
fmha_alu0(number<1>{} - sp_reg_idx);
}
};