update moe_align_block_size

8e2b79b4 · zhuwenwen · 9f201bc1 · 8e2b79b4
Commit 8e2b79b4 authored Sep 15, 2025 by zhuwenwen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

vllm/model_executor/layers/fused_moe/moe_align_block_size.py vllm/model_executor/layers/fused_moe/moe_align_block_size.py +1 -1

No files found.
--- a/vllm/model_executor/layers/fused_moe/moe_align_block_size.py
+++ b/vllm/model_executor/layers/fused_moe/moe_align_block_size.py
@@ -97,7 +97,7 @@ def moe_align_block_size(

    if envs.VLLM_USE_LIGHT_OP:
        op.moe_align_block_size(topk_ids, num_experts, block_size, sorted_ids,
-                                expert_ids, num_tokens_post_pad)
+                                expert_ids, num_tokens_post_pad, None)
    else:
        ops.moe_align_block_size(topk_ids, num_experts, block_size, sorted_ids,
                                expert_ids, num_tokens_post_pad)