From 26db7e0b7c7cfdf5d09a6a3e861783b81ef9001a Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Tue, 22 Apr 2025 14:45:27 +0000 Subject: [PATCH] Use kN0=64 to save vgprs --- .../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 c842df341b..5d82a3680c 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, 128, 32, 128, 32, 128>; + using type = ck_tile::sequence<128, 64, 32, 128, 32, 128>; using gemm0_warps = ck_tile::sequence<4, 1, 1>; using gemm1_warps = ck_tile::sequence<4, 1, 1>; };