ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enable padding blockscale for abquant (#3453)

Browse Source 
* Enable padding blockscale for abquant

* run clang-format

* Reduce unnecessary testing

* remove cout

[ROCm/composable_kernel commit: 7f68f3c4fa]
This commit is contained in:
kensclin
2025-12-25 01:12:40 +08:00
committed by GitHub
parent a2402950de
commit 0eb5d4a93f
8 changed files with 128 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
example/ck_tile/38_block_scale_gemm/run_gemm_quant_example.inc
View File

@@ -466,41 +466,6 @@ int run_gemm_example_with_layouts(const ck_tile::ArgParser& arg_parser,
ck_tile::index_t N = arg_parser.get_int("n");
ck_tile::index_t K = arg_parser.get_int("k");
if constexpr(QuantMode == ck_tile::QuantType::AQuantGrouped)
{
if(K % AQuantGroupSize::kK != 0)
{
throw std::runtime_error(
"K must be aligned with QuantGroupSize for AQuantGrouped mode");
}
}
if constexpr(QuantMode == ck_tile::QuantType::BQuantGrouped)
{
if(K % BQuantGroupSize::kK != 0)
{
throw std::runtime_error(
"K must be aligned with QuantGroupSize for BQuantGrouped mode");
}
}
if constexpr(QuantMode == ck_tile::QuantType::ABQuantGrouped)
{
if(K % AQuantGroupSize::kK != 0)
{
throw std::runtime_error(
"K must be aligned with QuantGroupSize for ABQuantGrouped mode");
}
if(K % BQuantGroupSize::kK != 0)
{
throw std::runtime_error(
"K must be aligned with QuantGroupSize for ABQuantGrouped mode");
}
if(K % BQuantGroupSize::kN != 0)
{
throw std::runtime_error(
"N must be aligned with QuantGroupSize for ABQuantGrouped mode");
}
}
ck_tile::index_t AQK, BQK, BQN = 0;
if constexpr(QuantMode == ck_tile::QuantType::AQuantGrouped)
{