update sgl-kernel for EP: kernel part (#8514)

Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Co-authored-by: Ke Bao <ispobaoke@gmail.com>
2025-07-30 22:19:55 -07:00
parent 59aab76f0a
commit a5f5ab4030
7 changed files with 12 additions and 32 deletions
--- a/sgl-kernel/python/sgl_kernel/moe.py
+++ b/sgl-kernel/python/sgl_kernel/moe.py
@@ -10,7 +10,6 @@ def moe_align_block_size(
    sorted_token_ids,
    experts_ids,
    num_tokens_post_pad,
-    token_cnts_buffer,
    cumsum_buffer,
    pad_sorted_token_ids=False,
 ):
@@ -21,7 +20,6 @@ def moe_align_block_size(
        sorted_token_ids,
        experts_ids,
        num_tokens_post_pad,
-        token_cnts_buffer,
        cumsum_buffer,
        pad_sorted_token_ids,
    )