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

Tiny fix and comments in HstuBlockMaskWithLocal::IsFullTimeInsideMask()

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-08-10 06:09:13 +00:00
parent 1404336200
commit 832ef5d1fa
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
example/ck_tile/18_hstu_attention/hstu_block_masking.hpp
View File

@@ -275,8 +275,11 @@ struct HstuBlockMaskWithLocal
index_t i_tile_right = i_tile_left + TileWidth;
index_t i_tile_bottom = i_tile_top + TileHeight;
if(!is_tile_in_first_split && i_tile_bottom <= max_uih_len &&
i_tile_right <= i_tile_top + 1)
// 1) tile is completely in [max_uih_len-min_full_attn_seqlen, max_uih_len]
// 2) some row of tile is in [max_uih_len, seqlen], requires i_tile_right <= max_uih_len
// to return true
if(!is_tile_in_first_split &&
(i_tile_bottom <= max_uih_len || i_tile_right <= max_uih_len))
return true;
};