From bec35abafe52beaf36b2751827e4ff2fc2c53c60 Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Fri, 30 May 2025 08:49:33 +0000 Subject: [PATCH] Tune the settings for hdim-256 --- .../ck_tile/18_hstu_attention/hstu_attention_fwd_setting.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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>; };