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

Merge commit 'db79fad16fe9c9d52b72c592715adf51d25e525e' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-09-17 02:36:09 +00:00
parent 42cd6a5536
commit 8318ba72be
4 changed files with 33 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
include/ck_tile/ops/gemm/kernel/grouped_gemm_kernel.hpp
View File

@@ -292,34 +292,8 @@ struct GroupedGemmKernel
{
__shared__ char smem_ptr_1[GetSmemSize()];
if constexpr(UsePersistentKernel || GemmPipeline::Preshuffle)
{
RunGemmWithPipelineSelection2LDS(a_ptr,
b_ptr,
c_ptr,
smem_ptr_0,
smem_ptr_1,
kargs,
splitk_batch_offset,
i_m,
i_n);
return;
}
else
{
Base::RunGemm2LDS({a_ptr},
{b_ptr},
{/*ds_ptr*/},
c_ptr,
smem_ptr_0,
smem_ptr_1,
kargs,
splitk_batch_offset,
i_m,
i_n);
}
RunGemmWithPipelineSelection2LDS(
a_ptr, b_ptr, c_ptr, smem_ptr_0, smem_ptr_1, kargs, splitk_batch_offset, i_m, i_n);
}
else // SingleSmemBuffer
{