This commit is contained in:
Jakub Piasecki
2025-10-30 13:17:06 +00:00
parent d8559798d5
commit e7696eeff1
32 changed files with 2189 additions and 93 deletions

View File

@@ -32,9 +32,9 @@ struct GroupedConvolutionBackwardWeightInvoker
GemmConfig::PermuteA,
GemmConfig::PermuteB>;
constexpr ck_tile::index_t VectorSizeA = 4;
constexpr ck_tile::index_t VectorSizeB = 8;
constexpr ck_tile::index_t VectorSizeC = 8;
constexpr ck_tile::index_t VectorSizeA = 1;
constexpr ck_tile::index_t VectorSizeB = 2;
constexpr ck_tile::index_t VectorSizeC = 2;
constexpr auto ConvSpec = ck_tile::ConvolutionSpecialization::Default;
using TilePartitioner =