ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix wp gemm bug when permuteN is false (#2935)

Browse Source 
* fix wp gemm bug when permuteN is false

* code clean

---------

Co-authored-by: valarLip <340077269@qq.com>
This commit is contained in:
lalala-sh
2025-09-27 04:28:54 +08:00
committed by GitHub
parent a44bea45b2
commit ee9769616a
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
example/ck_tile/03_gemm/gemm_utils.hpp
View File

@@ -72,6 +72,7 @@ struct GemmConfigBase
static constexpr ck_tile::index_t Pipeline = CK_TILE_PIPELINE_COMPUTE_V3;
static constexpr ck_tile::index_t NumWaveGroups = 1;
static constexpr bool Preshuffle = false;
static constexpr bool TiledMMAPermuteN = false;
};
template <typename PrecType>

2
example/ck_tile/03_gemm/gemm_weight_preshuffle_invoker.hpp
View File

@@ -109,7 +109,7 @@ struct WeightPreshuffleInvoker
GemmConfig::NumWaveGroups,
false,
1,
true>>;
GemmConfig::TiledMMAPermuteN>>;
using Kernel = ck_tile::GemmKernel<TilePartitioner, GemmPipeline, GemmEpilogue>;
auto kargs = Kernel::MakeKernelArgs(args);