fix gate-up when GU_NRepeat > 1

2026-05-04 05:31:24 +00:00 · 2025-08-18 18:43:37 +00:00
parent be55c0f9cb
commit f7f0306eea
1 changed files with 2 additions and 1 deletions
--- a/include/ck_tile/ops/moe_flatmm/kernel/moe_flatmm_kernel.hpp
+++ b/include/ck_tile/ops/moe_flatmm/kernel/moe_flatmm_kernel.hpp
@@ -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>{}});
                }
            });
        }