ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-19 12:30:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '7f68f3c4fa5bf478313c2147610317b199f9e65b' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-12-24 17:14:26 +00:00
parent e1039a7eeb
commit a169d59e06
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)
{