BatchGen: An Architecture for Scalable and Efficient Batch Inference
BatchGen把每条推理序列建模为可yield、combine、partition、migrate的协程,使运行时能动态合并专家层批次、拆分长尾序列并迁移状态。事件驱动运行时统一调度prefill和decode;在128-GPU集群的MoE、测试时扩展和RL rollout实验中,批完成时间最高降低2.3倍,并已部署到生产集群。
INSIGHT离线伪标签、评测回放和搜索相关性模型蒸馏往往是批推理而非在线服务,固定batch很容易被长尾序列拖慢。协程级的combine与migrate提供了可复用抽象,工程上应把batch completion time、GPU利用率和straggler比例作为与吞吐同等重要的指标。