AInfer-PD: Communication-Safe In-Place Prefill-Decode Multiplexing for Distributed MoE Rollouts
AInfer-PD 面向分布式 MoE 强化学习 rollout,把预填充与解码留在同一设备并共享权重和 KV。系统协调跨 rank 集体通信顺序,隔离 DeepEP 两条路径的协议状态;相对 SGLang,固定负载完成时间在单节点降低 24.8%–32.9%,双节点降低 18.3%–31.8%。
INSIGHT社区搜索若使用强化学习训练查询处理模型,多轮生成期间会反复出现预填充与解码混跑,通信顺序和协议状态需要一并设计。原地复用可避免另建资源池及 KV 迁移,但论文主要验证一至两节点 H20-3E;更大集群应先验证通信正确性、尾延迟和吞吐,不能直接外推加速幅度。