Skip to content

Commit 6d8f465

Browse files
committed
fix: update base pipeline classes according to changes in develop branch
1 parent 6418393 commit 6d8f465

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

example/ck_tile/17_grouped_gemm/quant_grouped_gemm.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ struct GemmQuantConfig<ck_tile::QuantType::AQuantGrouped>
180180
using GemmPipeline = ck_tile::AQuantGemmPipelineAgBgCrCompV3<GemmProblem>;
181181

182182
template <typename GemmProblem, bool PreshuffleB = false>
183-
using BaseGemmPipeline = ck_tile::BaseAQuantGemmPipelineAgBgCrCompV3<GemmProblem>;
183+
using BaseGemmPipeline = ck_tile::BaseGemmPipelineAgBgCrCompV3<GemmProblem>;
184184
};
185185

186186
template <>
@@ -198,7 +198,7 @@ struct GemmQuantConfig<ck_tile::QuantType::BQuantGrouped>
198198
using BaseGemmPipeline =
199199
std::conditional_t<PreshuffleB == true,
200200
ck_tile::BaseWeightPreshufflePipelineAGmemBGmemCRegV2<GemmProblem>,
201-
ck_tile::BaseBQuantGemmPipelineAgBgCrCompV3<GemmProblem>>;
201+
ck_tile::BaseGemmPipelineAgBgCrCompV3<GemmProblem>>;
202202
};
203203

204204
using grouped_gemm_kargs = ck_tile::QuantGroupedGemmHostArgs;

test/ck_tile/grouped_gemm_quant/test_grouped_gemm_util_quant.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -183,11 +183,11 @@ class TestCkTileGroupedGemmQuant : public ::testing::Test
183183
UseGroupedQuant,
184184
std::conditional_t<
185185
QuantType == ck_tile::QuantType::AQuantGrouped,
186-
ck_tile::BaseAQuantGemmPipelineAgBgCrCompV3<GemmPipelineProblem>,
186+
ck_tile::BaseGemmPipelineAgBgCrCompV3<GemmPipelineProblem>,
187187
std::conditional_t<
188188
PreshuffleB == true,
189189
ck_tile::BaseWeightPreshufflePipelineAGmemBGmemCRegV2<GemmPipelineProblem>,
190-
ck_tile::BaseBQuantGemmPipelineAgBgCrCompV3<GemmPipelineProblem>>>,
190+
ck_tile::BaseGemmPipelineAgBgCrCompV3<GemmPipelineProblem>>>,
191191
ck_tile::BaseGemmPipelineAgBgCrCompV3<GemmPipelineProblem>>;
192192

193193
const ck_tile::index_t k_grain = gemm_descs[0].k_batch * GroupedGemKernelParam::K_Tile;

0 commit comments

Comments
 (0)