ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the warp gemm option

Browse Source
This commit is contained in:
Clement Lin
2025-04-23 15:04:24 +08:00
committed by Philip Maybank
parent ac5b7cbf63
commit 4fccb261b8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/ck_tile/99_toy_example/03_flash_attention_fwd/block_gemm_areg_bsmem_creg_v1.hpp
View File

@@ -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;