Add endpoints to dump selected expert ids (#4435)

Co-authored-by: Cheng Wan <54331508+ch-wan@users.noreply.github.com>
2025-03-24 21:34:19 -07:00
parent 6b7038babd
commit 199bb01d00
10 changed files with 328 additions and 2 deletions
--- a/python/sglang/srt/managers/io_struct.py
+++ b/python/sglang/srt/managers/io_struct.py
@@ -658,6 +658,12 @@ class ProfileReqType(Enum):
    STOP_PROFILE = 2


+class ExpertDistributionReq(Enum):
+    START_RECORD = 1
+    STOP_RECORD = 2
+    DUMP_RECORD = 3
+
+
@dataclass
 class ProfileReq:
    type: ProfileReqType