Skip to content

05. Dynamic Routing and Load | 动态路由与负载 ​

学习问题 ​

静态切分先确定张量或序列属于哪个 rank;动态路由则根据本批次状态决定工作落到哪里。EP 把 token 发给专家,Serving 路由把请求发给模型副本。两者的对象和通信不同,但都必须同时控制容量、局部性、负载偏斜和最慢执行单元。

专家路由如何形成 All-to-All 与负载热点

核心机制 ​

Router 到输出的完整路径 ​

阶段状态变化需要记录的机制证据
routetoken 选择 top-k expertsexpert counts、routing entropy
capacity每个 expert 接受有限 tokencapacity factor、overflow / dropped tokens
dispatchtoken 按目标 expert 跨 rank 重排send / receive counts、All-to-All bytes
expert compute各 rank 执行本地专家每 rank token 数、compute time、peak memory
combine专家输出返回原 token 顺序并加权reverse All-to-All、combine time、正确性

负载不均决定最慢 rank ​

平均 token 数不能解释 EP 性能。真正限制 step time 的是热点 rank;即使总 payload 不变,不均衡也可能让部分 rank 计算和通信拥塞、其他 rank 空等。

现象优先检查可调整方向
overflow 升高capacity、top-k、router 分布capacity factor、路由正则、专家数
All-to-All 时间抖动send / receive counts、拓扑expert placement、分层通信、路由约束
expert compute 不均max / mean load、最慢 rank负载均衡、shared expert、token 重分配
吞吐增加但质量下降dropped tokens、router 稳定性质量门槛和 overflow 策略

请求路由是另一类动态分工 ​

Serving Replica 不执行专家 All-to-All,也不需要训练梯度同步。调度器需要同时读取副本队列、预计可用时间、Prefix Cache 归属和 Session 状态。只按队列选择可能丢失缓存复用,只按缓存归属又可能把请求持续送入热点副本。

路由候选主要信号获得的收益主要风险应保留的证据
最短队列queue length、available time降低排队等待忽略 Prefix Cache 与迁移成本每副本队列、等待时间、P95/P99
缓存感知prefix owner、cache hit减少重复 Prefill热点前缀造成粘性拥塞命中率、复用 token、热点副本负载
Session 粘性session owner、会话状态避免状态搬运长会话长期占用同一副本会话长度、迁移次数、尾延迟
受约束的局部性cache gain 与等待容忍度在复用和均衡之间折中阈值失配后两边都不占优makespan、TTFT、命中收益与等待代价
路由对象容量信号局部性信号长尾来源机制入口
token → expertexpert capacity、overflowexpert placement热门专家与最慢 rank46 Expert Parallel
request → replicaqueue、可用时间prefix / session owner热点副本与粘性等待PAR-REPLICA

选择条件与验证证据 ​

目的入口
理解 Router 与均衡目标Part 02 · 06 MoE Router、07 Load Balancing
验证 dispatch / combinePart 02 · 46 Expert Parallel
比较请求路由与副本负载PAR-REPLICA
定位通信热点06 通信 Profiling 与 Overlap

相关阅读 ​

Released under the MIT License.