From 4fccb261b860e2977ef796def53db75cc627e9ab Mon Sep 17 00:00:00 2001
From: Clement Lin <clement.lin@amd.com>
Date: Wed, 23 Apr 2025 15:04:24 +0800
Subject: [PATCH] Add the warp gemm option

---
 .../03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp b/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp
index 23ca02946b..fc18640958 100644
--- a/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp
+++ b/example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp
@@ -36,7 +36,7 @@ struct BlockGemmARegBSmemCRegV1
         constexpr index_t NPerBlock = BlockGemmShape::kN;
         constexpr index_t KPerBlock = BlockGemmShape::kK;
 
-        constexpr auto config     = Policy::template GetWarpGemmMWarpNWarp<Problem>();
+        constexpr auto config     = Policy::template GetWarpGemmMWarpNWarp<Problem, MPerBlock>();
         using WG                  = remove_cvref_t<decltype(config.template get<0>())>;
         constexpr index_t MPerXDL = WG::kM;
         constexpr index_t NPerXDL = WG::kN;