From da89540ee025ec0587416ca20f2473e4ddf562ed Mon Sep 17 00:00:00 2001
From: Qianfeng Zhang <Qianfeng.Zhang@amd.com>
Date: Wed, 30 Apr 2025 05:42:43 +0000
Subject: [PATCH] Use kN0=32

---
 .../ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp   | 1 -
 .../ck_tile/18_hstu_attention/hstu_attention_fwd_setting.hpp    | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

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<Problem>();
 
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>;
 };