Skip to content

Commit 8985b70

Browse files
committed
chore: removed work-around for aquant bug that has been fixed
1 parent e78bcbb commit 8985b70

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

test/ck_tile/grouped_gemm_quant/test_grouped_gemm_util_quant.hpp

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff 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;

0 commit comments

Comments
 (0)