diff --git a/include/ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_weight_kernel.hpp b/include/ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_weight_kernel.hpp index 1ca34f0169..557504c2c7 100644 --- a/include/ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_weight_kernel.hpp +++ b/include/ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_weight_kernel.hpp @@ -626,17 +626,17 @@ struct GroupedConvolutionBackwardWeightKernel return false; } - // if constexpr (GroupedConvTraitsType_::NumGroupsToMerge > 1) - // { - // const index_t ConvG = kargs.wei_g_k_c_xs_lengths[number<0>{}]; - // if (ConvG % GroupedConvTraitsType_::NumGroupsToMerge != 0) - // { - // CK_TILE_ERROR("ConvG must be a multiple of NumGroupsToMerge!"); - // return false; - // } + if constexpr(GroupedConvTraitsType_::NumGroupsToMerge > 1) + { + const index_t ConvG = kargs.wei_g_k_c_xs_lengths[number<0>{}]; + if(ConvG % GroupedConvTraitsType_::NumGroupsToMerge != 0) + { + CK_TILE_ERROR("ConvG must be a multiple of NumGroupsToMerge!"); + return false; + } - // // TODO: Should we also check that GemmM <= MPerBlock and GemmN <= NPerBlock? - // } + // TODO: Should we also check that GemmM <= MPerBlock and GemmN <= NPerBlock? + } return true; }