SAGA: Workflow-Atomic Scheduling for AI Agent Inference on GPU Clusters
面向 Agent / 复杂 RAG 场景,论文指出现有 GPU 调度器把每次 LLM 调用当作独立请求,丢弃跨步骤累积的 KV cache 与中间状态,导致端到端延迟膨胀 3–8×。SAGA 提出 workflow-atomic 调度:把整个 agent 程序作为最小调度单元,把依赖紧密的调用共置同一个 KV/显存域,并按程序结构复用 KV cache 与 prefix。实验在真实 agent 工作负载上同时显著降低尾延迟和提升集群吞吐。
INSIGHTSAGA 揭示了一个被多数推理框架忽视的事实:调度粒度从 "单次 LLM 调用" 上升到 "整段 workflow" 后,KV cache 与 prefix 复用收益会成倍放大。社区搜索的 query 改写 → 多路召回 → 重排 → 摘要生成是典型多步链路,借鉴 SAGA 的 program-level 调度,可以把意图分类和重排共享 prefix,工程上一旦在 vLLM/SGLang 之上增加 workflow group 抽象,性价比远高于单 step 微优化。