From 89cc644fd681f98ab765b2413a2f7b778e646cd0 Mon Sep 17 00:00:00 2001
From: Wanglongzhi2001 <583087864@qq.com>
Date: Wed, 24 Dec 2025 16:58:45 +0800
Subject: [PATCH] [Model] support mode config for expert_dispatch

---
 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 79317afab48..213ca03e15f 100644
--- a/custom_ops/gpu_ops/moe/ep_moe_expert_dispatch.cu
+++ b/custom_ops/gpu_ops/moe/ep_moe_expert_dispatch.cu
@@ -53,6 +53,11 @@
       __VA_ARGS__                                                              \
       break;                                                                   \
     }                                                                          \
+    case 10: {                                                                 \
+      constexpr size_t NUM_EXPERTS_PER_RANK = 10;                              \
+      __VA_ARGS__                                                              \
+      break;                                                                   \
+    }                                                                          \
     case 16: {                                                                 \
       constexpr size_t NUM_EXPERTS_PER_RANK = 16;                              \
       __VA_ARGS__                                                              \