From dafcb394967cbfb8200172cb6d8659b7ee9d92b1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ville=20Pietil=C3=A4?= Date: Mon, 22 Sep 2025 12:12:58 +0000 Subject: [PATCH] Tile example parameters changes to match depthwise conv. --- .../grouped_convolution_backward_data.cpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/example/ck_tile/20_grouped_convolution/grouped_convolution_backward_data.cpp b/example/ck_tile/20_grouped_convolution/grouped_convolution_backward_data.cpp index 52eaab9f94..2d0c6c9029 100644 --- a/example/ck_tile/20_grouped_convolution/grouped_convolution_backward_data.cpp +++ b/example/ck_tile/20_grouped_convolution/grouped_convolution_backward_data.cpp @@ -28,21 +28,21 @@ float grouped_conv_bwd_data(const ck_tile::GroupedConvBwdDataHostArgs& args, { constexpr int kBlockPerCu = 1; - constexpr ck_tile::index_t M_Tile = 64; - constexpr ck_tile::index_t N_Tile = 64; - constexpr ck_tile::index_t K_Tile = 32; + constexpr ck_tile::index_t M_Tile = 8; + constexpr ck_tile::index_t N_Tile = 128; + constexpr ck_tile::index_t K_Tile = 64; constexpr ck_tile::index_t M_Warp = 2; constexpr ck_tile::index_t N_Warp = 2; constexpr ck_tile::index_t K_Warp = 1; - constexpr ck_tile::index_t M_Warp_Tile = 32; - constexpr ck_tile::index_t N_Warp_Tile = 32; + constexpr ck_tile::index_t M_Warp_Tile = 4; + constexpr ck_tile::index_t N_Warp_Tile = 64; constexpr ck_tile::index_t K_Warp_Tile = 16; constexpr ck_tile::index_t VectorSizeA = 8; constexpr ck_tile::index_t VectorSizeB = 8; - constexpr ck_tile::index_t VectorSizeC = 8; + constexpr ck_tile::index_t VectorSizeC = 1; // Implicit GEMM Traits using CodegenShape =