[BugFix] fix wint2 (#6109)

* fix

* fix

* fix
This commit is contained in:
lizexu123
2026-01-20 21:46:21 +08:00
committed by GitHub
parent 5385d51808
commit f4902fe42d
4 changed files with 17 additions and 12 deletions
@@ -278,6 +278,7 @@ class CutlassWint2FusedMoeMethod(Wint2MoeMethod):
topk_idx,
expert_idx_per_token,
dequant_scale,
_,
) = moe_expert_dispatch(
x,
gate_out,