ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-13 10:37:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix passing the number of merged groups.

Browse Source
This commit is contained in:
Ville Pietilä
2025-10-08 11:23:27 +00:00
parent 36f75d061a
commit 595a9f01c1
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
example/ck_tile/20_grouped_convolution/grouped_convolution_backward_weight_invoker.hpp
View File

@@ -17,7 +17,7 @@ struct GroupedConvolutionBackwardWeightInvoker
typename InLayout,
typename WeiLayout,
typename OutLayout,
ck_tile::index_t NumGroupMerge = 1,
ck_tile::index_t NumGroupsToMerge = 1,
typename DsDataType = ck_tile::tuple<>,
typename DsLayout = ck_tile::tuple<>,
typename CDEElementWise = ck_tile::element_wise::PassThrough>
@@ -58,7 +58,8 @@ struct GroupedConvolutionBackwardWeightInvoker
OutLayout,
VectorSizeA,
VectorSizeB,
VectorSizeC>;
VectorSizeC,
NumGroupsToMerge>;
using CodegenPipelineProblem = ck_tile::GemmPipelineProblem<
InDataType,
WeiDataType,