File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
example/ck_tile/17_grouped_gemm
test/ck_tile/grouped_gemm_quant Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff 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
186186template <>
@@ -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
204204using grouped_gemm_kargs = ck_tile::QuantGroupedGemmHostArgs;
Original file line number Diff line number Diff 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;
You can’t perform that action at this time.
0 commit comments