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__ \