From 29fe485ad2ae6692b9be8f64a01cfae9888ba316 Mon Sep 17 00:00:00 2001 From: yuanlehome Date: Mon, 5 Jan 2026 23:39:41 +0800 Subject: [PATCH] dispatch num_experts_per_rank 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 213ca03e15f..7124b684d67 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__ \