AGI Pulse
产品/应用AWS · agipulse

AWS 为 vLLM 分离式推理贡献推送式与双向 KV 缓存传输,自测首 token 时间最多快 3 倍

阅读 0

发布日期
2026-10-21
类型
产品/应用
相关机构
AWS
信息来源
agipulse

这件事意味着什么

【要点】

AWS 在机器学习博客发布两项为 vLLM「预填充/解码分离」推理做的 KV 缓存传输改进,并称都已作为与硬件无关的功能贡献给 vLLM。

一、推送式传输:vLLM 的 NIXL 连接器默认由解码端在预填充完成后再「拉」KV 块,协调步骤卡在首 token 的关键路径上。AWS 改为预填充端主动推送,把准备工作提前。在 EC2 P5en(GPU、EFA 互联)1P1D 部署上跑 Qwen3-32B,首 token 时间(TTFT)最多快 3 倍,每 token 输出时间(TPOT)最多快 30%。

二、双向传输:多轮智能体对话里,解码端生成的 KV 以前回不到预填充端,下一轮要重算。双向传输让两端复用已算好的上下文;多轮测试中 TTFT 对 Qwen3-32B 最多降 1.6 倍,对 GLM-5-FP8 最多降 3.3 倍。

三、两项功能先在 Trainium 集群上开发验证,博客结果来自 GPU 集群。

【可核验】

推送式 KV 传输:https://aws.amazon.com/blogs/machine-learning/push-based-kv-transfer-to-improve-ttft-and-tpot-for-disaggregated-inference-in-vllm/

双向 KV 传输:https://aws.amazon.com/blogs/machine-learning/improve-ttft-for-multi-turn-agentic-workloads-with-bi-directional-kv-transfer-in-vllm/

【我们的判断】

一、背景:预填充/解码分离已是大规模推理的主流做法,llm-d 1.0、DeepSeek 的推理栈和 NVIDIA Dynamo 都在围绕它做文章,瓶颈越来越落在两端之间搬 KV 缓存的那一步。

二、AWS 的两项改动都瞄准智能体负载:长上下文、多轮反复重发历史,正是 TTFT 最容易恶化的场景。和此前多家只在自家栈里优化不同,AWS 选择贡献给 vLLM 上游,且强调与加速器无关,GPU 用户同样受益,也有助于把 Trainium 拉进主流开源生态。

三、局限:数字均为 AWS 自测,只有 1P1D 小规模配置,且依赖 EFA 这类 RDMA 网络;博客发布已一天多,具体合并状态和适用版本请以 vLLM 仓库为准。

原文 →

同一天的短讯集中在 2026-10-21 短讯

继续了解相关进展

本页由 AGI Pulse 聚合,仅保留标题与链接,不转载正文。 编辑政策