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

Fix the GetTileRangeAlongX() to align with the hstu masking definition when both causal=true and local=true

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-04-18 15:37:49 +00:00
parent efc786f6a3
commit ee259a8924
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -61,9 +61,10 @@ struct HstuBlockMaskWithLocal
}
else
{
index_t x_end = i_y + YTile + max_attn_len;
index_t x_start = max(0, i_y - max_attn_len);
index_t x_end = min(i_y + YTile, max_uih_len);
return ck_tile::make_tuple(0, x_end);
return ck_tile::make_tuple(x_start, x_end);
};
};
}