We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9a9e4c0 commit 4e8f231Copy full SHA for 4e8f231
benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py
@@ -398,11 +398,6 @@ def main(args: argparse.Namespace):
398
topk = config.num_experts_per_tok
399
intermediate_size = config.moe_intermediate_size
400
shard_intermediate_size = 2 * intermediate_size // args.tp_size
401
- elif config.architectures[0] == "Qwen3MoeForCausalLM":
402
- E = config.num_experts
403
- topk = config.num_experts_per_tok
404
- intermediate_size = config.moe_intermediate_size
405
- shard_intermediate_size = 2 * intermediate_size // args.tp_size
406
elif config.architectures[0] in ["DeepseekV2ForCausalLM", "DeepseekV3ForCausalLM"]:
407
n_share_fusion_experts = args.n_share_experts_fusion
408
E = (
0 commit comments