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

update

Browse Source
This commit is contained in:
yadaish
2025-12-09 00:30:52 +00:00
parent 971ed7da51
commit aee38fdbf3
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
example/ck_tile/18_flatmm/mixed_prec/run_a16w4_moe_flatmm_example.inc
View File

@@ -72,7 +72,8 @@ int run_a16w4_moe_gemm_example_with_layouts(int argc,
using CDataType = PrecActType;
using AccDataType = float;
using ScaleType = ck_tile::e8m0_t;
static constexpr bool IsInt4 = std::is_same_v<BDataType, ck_tile::pk_int4_t>;
using ScaleType = std::conditional_t<IsInt4, ck_tile::bfloat16_t, ck_tile::e8m0_t>;
constexpr int ScaleGranularityN = 1;
constexpr int ScaleGranularityK = 32;