From bace12feac507e71c06c8a160b29a06a3c061200 Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Tue, 22 Apr 2025 13:42:03 +0000 Subject: [PATCH] Fix in GetTileRangeAlongX --- example/ck_tile/18_hstu_attention/hstu_block_masking.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/18_hstu_attention/hstu_block_masking.hpp b/example/ck_tile/18_hstu_attention/hstu_block_masking.hpp index 3c39b332a6..161b0cb5cb 100644 --- a/example/ck_tile/18_hstu_attention/hstu_block_masking.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_block_masking.hpp @@ -64,7 +64,7 @@ struct HstuBlockMaskWithLocal 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(x_start, x_end); + return ck_tile::make_tuple(x_start - x_start % XTile, x_end); }; }; }