diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_setting.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_setting.hpp index d8256f5e04..65eeafbd0f 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_setting.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_setting.hpp @@ -43,7 +43,7 @@ struct HstuAttentionFwdBlockTile<128> template <> struct HstuAttentionFwdBlockTile<256> { - using type = ck_tile::sequence<128, 128, 32, 256, 32, 256>; + using type = ck_tile::sequence<128, 32, 16, 256, 16, 256>; using gemm0_warps = ck_tile::sequence<4, 1, 1>; using gemm1_warps = ck_tile::sequence<4, 1, 1>; };