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 03c1945d95..b989d63e0e 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 @@ -814,7 +814,7 @@ struct TransformConvBwdDataToGemm_v1 IWTildeSliceBegin, GcdStrideDilationH_, GcdStrideDilationW_, - AK0, + AK0 * batch_k_, AK1, GemmMPerBlock, GemmKPerBlock)),