From cea919aefb2d028349d74886cccfec28a7a70d3d Mon Sep 17 00:00:00 2001
From: Qianfeng Zhang <Qianfeng.Zhang@amd.com>
Date: Thu, 24 Apr 2025 07:59:28 +0000
Subject: [PATCH] Use 16x16x16 WarpGemm

---
 .../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 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 <ck_tile::index_t MaxK>
 struct HstuAttentionFwdShape;