Fix positional argument (#7093)

This commit is contained in:
Quanfeng Li
2025-06-12 09:31:13 +08:00
committed by GitHub
parent 021f76e4f4
commit ef32677444
2 changed files with 2 additions and 0 deletions

View File

@@ -84,6 +84,7 @@ def ep_moe(
top_k,
hidden_states.shape[1],
BLOCK_SIZE=512,
use_per_token_if_dynamic=True,
)
seg_indptr_cur_rank = seg_indptr[start_expert_id : end_expert_id + 2]