From da5db1773d0359b8189ae3dc1627ae52a626e14e Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Tue, 12 Aug 2025 01:58:10 +0000 Subject: [PATCH] Tiny fix in HstuBlockMaskWithLocal::GetTileRangeAlongX() --- example/ck_tile/18_hstu_attention/hstu_block_masking.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 51cfd6ed1e..8e59995ff3 100644 --- a/example/ck_tile/18_hstu_attention/hstu_block_masking.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_block_masking.hpp @@ -88,7 +88,7 @@ struct HstuBlockMaskWithLocal // in [0, max_uih_len-min_full_attn_seqlen) if constexpr(!kUseCausal) { - if(i_y >= contextual_seqlen + max_attn_len) + if(i_y >= min(contextual_seqlen, 1) + max_attn_len) { // some row of the tile in [contextual_seqlen+max_attn_len, max_uih_len) if(i_y < max_uih_len) @@ -132,7 +132,7 @@ struct HstuBlockMaskWithLocal } else // kUseCausal && kUseLocal { - if(i_y >= contextual_seqlen + max_attn_len) + if(i_y >= min(contextual_seqlen, 1) + max_attn_len) { index_t x_end = min(i_y + YTile, seqlen);