From c7f3bcc81eebdc385386795456a1893eda0e3214 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bart=C5=82omiej=20Kocot?= Date: Fri, 10 Oct 2025 09:24:21 +0200 Subject: [PATCH] Fix splitK for grouped conv bwd data (#2991) [ROCm/composable_kernel commit: ad7a215aba80308e2ec49849a017f73da7bfa437] --- .../operator_transform/transform_conv_bwd_data_to_gemm_v1.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)),