From 065bd075c70e37119fd878fc87e715d02b2f466e Mon Sep 17 00:00:00 2001 From: yuanlehome Date: Tue, 6 Jan 2026 00:14:49 +0800 Subject: [PATCH] num_experts_per_rank dispatch 5 --- custom_ops/gpu_ops/moe/ep_moe_expert_dispatch.cu | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/custom_ops/gpu_ops/moe/ep_moe_expert_dispatch.cu b/custom_ops/gpu_ops/moe/ep_moe_expert_dispatch.cu index 9259ac7f45a..52e7abfc3bc 100644 --- a/custom_ops/gpu_ops/moe/ep_moe_expert_dispatch.cu +++ b/custom_ops/gpu_ops/moe/ep_moe_expert_dispatch.cu @@ -38,6 +38,11 @@ __VA_ARGS__ \ break; \ } \ + case 5: { \ + constexpr size_t NUM_EXPERTS_PER_RANK = 5; \ + __VA_ARGS__ \ + break; \ + } \ case 6: { \ constexpr size_t NUM_EXPERTS_PER_RANK = 6; \ __VA_ARGS__ \