File tree Expand file tree Collapse file tree 1 file changed +4
-9
lines changed
test/ck_tile/grouped_gemm_quant Expand file tree Collapse file tree 1 file changed +4
-9
lines changed Original file line number Diff line number Diff line change @@ -86,22 +86,17 @@ class TestCkTileGroupedGemmQuant : public ::testing::Test
8686
8787 struct GroupedGemKernelParam_Mfma
8888 {
89- // HACK: There's a bug in the AQuant pipeline that causes MRepeat > 1 to be incorrect
90- // For now we work-around like this
91- constexpr static bool DisableMRepeat =
92- QuantType == ck_tile::QuantType::AQuantGrouped && !TransposeC;
93-
9489 static const bool kPadM = false ;
9590 static const bool kPadN = false ;
9691 static const bool kPadK = false ;
9792
9893 static const int kBlockPerCu = 1 ;
99- static const ck_tile::index_t M_Tile = DisableMRepeat ? 128 : 256 ;
100- static const ck_tile::index_t N_Tile = DisableMRepeat ? 128 : 256 ;
94+ static const ck_tile::index_t M_Tile = 256 ;
95+ static const ck_tile::index_t N_Tile = 256 ;
10196 static const ck_tile::index_t K_Tile = 128 ;
10297
103- static const ck_tile::index_t M_Warp = DisableMRepeat ? 4 : 2 ;
104- static const ck_tile::index_t N_Warp = DisableMRepeat ? 1 : 2 ;
98+ static const ck_tile::index_t M_Warp = 2 ;
99+ static const ck_tile::index_t N_Warp = 2 ;
105100 static const ck_tile::index_t K_Warp = 1 ;
106101
107102 static const ck_tile::index_t M_Warp_Tile = 32 ;
You can’t perform that action at this time.
0 commit comments