ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-18 09:38:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
yadaish
2025-12-04 18:28:17 +00:00
parent 7059613404
commit bd6897d432
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp
View File

@@ -639,7 +639,8 @@ struct MoeFlatmmKernel
index_t FlatScaleN = kargs.N / N_Pack / BlockGemmShape::WarpTile::at(I1);
// using ScaleType = std::conditional_t<MXFP4_Pipeline, e8m0_t, float>;
using ScaleType = std::conditional_t<MXFP4_Pipeline, bfloat16_t, float>;
static constexpr bool IsInt4 = std::is_same_v<BDataType, pk_int4_t>;
using ScaleType = std::conditional_t<MXFP4_Pipeline, std::conditional_t<IsInt4, bfloat16, e8m0_t>, float>;
const auto scale_b_flat_view = make_naive_tensor_view<address_space_enum::global>(
reinterpret_cast<const ScaleType*>(scale_n.ptr) + expert_id * kargs.N * scale_k,