AI导航

Google 用 Ironwood TPU 优化 Qwen 3.5,预填充性能提升 4.7 倍

AI资讯
3 min read
23 次阅读

2026 年 7 月 14 日,Google 发布 Qwen 3.5-397B-A17B 在 Ironwood(TPU v7x)上的推理优化报告:在其 512 并发测试档位下,4 月至 6 月间,偏解码工作负载性能提高约 3.1 倍,偏预填充工作负载提高约 4.7 倍。这个结果说明大型 MoE 模型可以通过硬件感知的并行策略和可复用内核显著提速,但它不是“所有 Qwen 任务都会快 4.7 倍”的通用承诺。

为什么 397B 模型仍有优化空间

Qwen 3.5-397B 总参数约 3970 亿,每个 token 只激活约 170 亿参数,激活比例约 4.3%。这种稀疏 MoE 结构降低了单次计算量,却把瓶颈转移到专家路由、跨芯片通信、KV 缓存和不规则张量处理。模型有 512 个路由专家,每次选择其中 10 个;同时只有 2 个 KV 头,传统的 8 路张量并行会遇到无法把两个头均匀切到八个设备的问题。

Google 改了并行方式而不是只换芯片

团队采用 8 路注意力数据并行与 8 路专家并行组合:注意力权重在各设备复制,每个核心本地处理完整 KV 头;512 个专家则平均分布到设备上。注意力和 MoE 层之间通过 All-Gather 与 Reduce-Scatter 交换数据,从而避免复制整套模型权重。

通信路径也被重新设计。团队把三次 All-Gather 合并为两次,并用分层 Reduce-Scatter 优先利用同一物理芯片内更快的连接,再跨芯片归并。针对推理内核,KV 页大小从 16 调到 256,使 512 并发下的解码步骤延迟从 428 微秒降到 283 微秒;Batched RPA、稀疏路由和融合 Gated DeltaNet 内核继续减少索引与内存搬运开销。

性能数字应该怎样理解

Google 的预填充测试使用 8K 输入、1K 输出,解码测试使用 1K 输入、8K 输出,并覆盖 64、128、256 和 512 四个并发档位。最终报告重点给出的 3.1 倍与 4.7 倍,是同一团队在特定硬件和软件栈上两个月优化前后的对比,不是与 GPU、其他 TPU 或其他云服务的横向结论。

在 64 并发基准下,优化栈的预填充吞吐达到每芯片每秒 3707 token,约为报告估算理论上限的 82.4%;解码吞吐为每芯片每秒 677 token。企业评估时还要结合请求长度、并发度、首 token 延迟、模型精度、编译时间和总体成本,不能只看峰值吞吐。

真正值得关注的是可复用性

这份报告的长期价值在于模块化方法。Google 把 Batched RPA、Grouped GEMM、SparseCore 处理和 Pallas 通信内核拆成可移植组件,并把部分优化接入 vLLM 与 SGLang。对于需要托管大型开放权重 MoE 模型的团队,这比一次性的跑分更重要:新模型若复用相近结构,可以少走一遍从零调优的过程。后续仍需观察这些优化在公开云实例、不同批量和真实混合请求中的稳定表现。

来源与说明

本文数据与技术细节来自 Google for Developers 于 2026 年 7 月 14 日发布的 Qwen 3.5 Ironwood TPU 优化报告。所有性能数字均保留其测试条件,不扩展为跨平台结论。

推荐工具

NVIDIA Chat with RTX AI聊天 Chat with RTX 是 NVIDIA 面向 RTX 电脑的本地 AI 聊天工具,可围绕本地文档和视频资料做问答,适合重视隐私、离线检索并具备硬件条件的用户更适合资料不便上传云 HuggingChat AI聊天 HuggingChat 是 Hugging Face 的开源模型聊天应用,支持 Omni 自动选模型,也可手动选社区开放模型对话。它适合体验开源模型、技术探索和问答,结果可能不稳定,重要内容需复核。 纳米AI搜索 AI搜索 纳米AI 是 360 旗下 AI 搜索和智能体入口,支持文字、语音、拍照提问、多模型协作与内容创作,适合中文用户做日常搜索、学习问答、移动查询、热点追踪、生活决策、知识整理和轻量创作。 Qwen Chat(通义千问) AI聊天 Qwen Chat 是Qwen 大模型聊天助手,支持Qwen 模型聊天、写作和编程问答,适合中文用户和开发者完成 AI 对话、资料问答和任务协作,适合上线前核对权限、成本和资料质量。 DuckDuckGo AI Chat AI聊天 DuckDuckGo AI Chat 适合重视隐私的普通用户使用,围绕隐私 AI 聊天、免账户使用和多模型问答提供对话式 AI 支持和任务协作能力,适合上线前核对权限、成本和资料质量。 Goody-2 AI聊天 Goody-2 是以极端安全和伦理对齐为特色的 AI 聊天助手,常用于观察模型拒答边界、内容安全策略、保守输出体验和 AI 风险讨论,更适合作为安全边界样本使用前应理解它偏实验性质,不适合。