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

fix gate-up when GU_NRepeat > 1

Browse Source
This commit is contained in:
Feng Shijie
2025-08-18 18:43:37 +00:00
parent be55c0f9cb
commit f7f0306eea
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/moe_flatmm/kernel/moe_flatmm_kernel.hpp
View File

@@ -1117,7 +1117,8 @@ struct MoeFlatmmKernel
{
constexpr auto step = SFC::get_forward_step(iAccess);
// row_offset of out windows has been included in scatter offset
move_tile_window(c_block_window, {0, step.at(number<1>{})});
move_tile_window(c_block_window,
{0, step.at(number<1>{}) / number<IsGateUp ? 2 : 1>{}});
}
});
}