Restore previous padd for 1x1 specialization

84ed31d3 · Bartlomiej Kocot · 04d762dc · 84ed31d3
Commit 84ed31d3 authored Aug 31, 2023 by Bartlomiej Kocot
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/operator_transform/transform_conv_bwd_data_to_gemm_v1.hpp ...operator_transform/transform_conv_bwd_data_to_gemm_v1.hpp +1 -1

No files found.
--- 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
@@ -546,7 +546,7 @@ struct TransformConvBwdDataToGemm_v1
            const auto wei_gemmbk0_gemmn_gemmbk1_grid_desc =
                ck::tensor_operation::device::PadTensorDescriptor(
                    wei_gemmbk0_gemmnraw_gemmbk1_grid_desc,
-                    make_tuple(GemmKPerBlock, GemmNPerBlock, BK1),
+                    make_tuple(BK0, GemmNPerBlock, BK1),
                    Sequence<false, DoPadGemmN, false>{});

            return wei_gemmbk0_gemmn_gemmbk1_grid_desc;