diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp index b5ebf701de..cca5f2dc48 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp @@ -156,7 +156,6 @@ struct HstuAttentionFwdPipelineQRKSVS "wrong!"); constexpr index_t k1_loops = kN0 / kK1; - static_assert(2 <= k1_loops); constexpr auto NumKVLdsBuffers = Policy::template GetNumKVLdsBuffers(); 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 f8fdf681fc..3c426fe526 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 @@ -35,7 +35,7 @@ struct HstuAttentionFwdBlockTile<64> template <> struct HstuAttentionFwdBlockTile<128> { - using type = ck_tile::sequence<128, 64, 32, 128, 32, 128>; + using type = ck_tile::sequence<128, 32, 32, 128, 32, 128>; using gemm0_warps = ck_tile::sequence<4, 1, 1>; using gemm1_warps = ck_tile::sequence<4, 1, 1>; };