From 60d8ffb6cfffdd53994029f2d7f8021defdf3c4c Mon Sep 17 00:00:00 2001
From: Qianfeng Zhang <Qianfeng.Zhang@amd.com>
Date: Thu, 26 Jun 2025 09:46:33 +0000
Subject: [PATCH] Use two work-groups per compute-unit for scheduling the
 kernel

---
 .../ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 05288b0d5d..d705f38d36 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
@@ -91,7 +91,7 @@ struct HstuAttentionFwdPipelineQRKSVS
                 if constexpr(kHasBias)
                     return 2;
                 else
-                    return 3;
+                    return 2;
             }
             else if constexpr(kQKHeaddim == 256)
             {