diff --git a/include/ck/tensor_operation/operator_transform/transform_conv_bwd_data_to_gemm_v1.hpp b/include/ck/tensor_operation/operator_transform/transform_conv_bwd_data_to_gemm_v1.hpp index 52d5bebcdf..50a13e297f 100644 --- a/include/ck/tensor_operation/operator_transform/transform_conv_bwd_data_to_gemm_v1.hpp +++ b/include/ck/tensor_operation/operator_transform/transform_conv_bwd_data_to_gemm_v1.hpp @@ -42,6 +42,7 @@ struct TransformConvBwdDataToGemm_v1 * in convolution backward data operations. */ static constexpr bool CustomTensorTransformBwdData = true; + static constexpr bool UsePaddingOnKDim = std::is_same_v; template using NumberType = @@ -694,7 +695,7 @@ struct TransformConvBwdDataToGemm_v1 ck::tensor_operation::device::PadTensorDescriptor( out_gemmak0_gemmmraw_gemmak1_grid_desc, make_tuple(AK0 * batch_k_, GemmMPerBlock, AK1), - Sequence{}); + Sequence{}); return out_gemmak0_gemmm_gemmak1_grid_desc; } @@ -993,7 +994,7 @@ struct TransformConvBwdDataToGemm_v1 ck::tensor_operation::device::PadTensorDescriptor( wei_gemmbk0_gemmnraw_gemmbk1_grid_desc, make_tuple(BK0 * batch_k_, GemmNPerBlock, BK1), - Sequence{}); + Sequence{}); return wei_gemmbk0_gemmn_gemmbk1_grid_desc; } @@ -1573,7 +1574,7 @@ struct TransformConvBwdDataToGemm_v1 return ck::tensor_operation::device::PadTensorDescriptor( out_gemmk0_gemmm_gemmk1_grid_desc, make_tuple(Number{}, Number{}, Number{}), - Sequence{}); + Sequence{}); } else { @@ -1645,7 +1646,7 @@ struct TransformConvBwdDataToGemm_v1 return ck::tensor_operation::device::PadTensorDescriptor( out_gemmk0_gemmm_gemmk1_grid_desc, make_tuple(Number{}, Number{}, Number{}), - Sequence{}); + Sequence{}); } } @@ -1671,7 +1672,7 @@ struct TransformConvBwdDataToGemm_v1 return ck::tensor_operation::device::PadTensorDescriptor( wei_gemmk0_gemmn_gemmk1_grid_desc, make_tuple(Number{}, Number{}, Number{}), - Sequence{}); + Sequence{}); } else { @@ -1724,7 +1725,7 @@ struct TransformConvBwdDataToGemm_v1 return ck::tensor_operation::device::PadTensorDescriptor( wei_gemmk0_gemmn_gemmk1_grid_desc, make_tuple(Number{}, Number{}, Number{}), - Sequence{}); + Sequence{}); } }