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 5d82a3680c..f8fdf681fc 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 @@ -48,7 +48,7 @@ struct HstuAttentionFwdBlockTile<256> using gemm1_warps = ck_tile::sequence<4, 1, 1>; }; -using HstuAttentionFwdWarpTile1 = ck_tile::sequence<32, 32, 16>; +using HstuAttentionFwdWarpTile1 = ck_tile::sequence<16, 16, 16>; template struct HstuAttentionFwdShape;