Commit 90c5cc41 authored by zhuwenwen's avatar zhuwenwen
Browse files

update invoke_fused_moe_kernel() arguments

parent 40ff7bb2
...@@ -1699,7 +1699,6 @@ def fused_experts_impl( ...@@ -1699,7 +1699,6 @@ def fused_experts_impl(
w2_scale, w2_scale,
w2_zp, w2_zp,
curr_topk_weights, curr_topk_weights,
curr_topk_ids,
sorted_token_ids, sorted_token_ids,
expert_ids, expert_ids,
num_tokens_post_padded, num_tokens_post_padded,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment