SemiAnalysis:谷歌TPU杀入英伟达腹地,Ironwood每美元性能最高领先B200 50%

华尔街见闻
Sep 08

谷歌自研芯片TPU正加速走出内部围墙,向外部客户开放。

半导体研究机构SemiAnalysis发布首份针对TPUv7 Ironwood的第三方推理测试结果,显示在同等条件对比下,Ironwood在每美元性能指标上最高领先英伟达B200和B300达50%,标志着谷歌在AI推理芯片市场的竞争格局中正式发起冲击。

SemiAnalysis的测试数据显示,在每用户每秒100个token的交互速度下,Ironwood每百万token推理成本约为0.181美元,低于B200的0.222美元和B300的0.276美元,分别便宜约19%和34%。

在更高并发场景下,若采用谷歌内部总拥有成本(TCO)计算,Ironwood相对B200的每美元性能优势可扩大至76.7%,相对B300则达130.2%。与此同时,谷歌正全力推进名为TorchTPU的新一代原生PyTorch后端,预计于10月开源,以取代此前存在诸多局限的TorchAX方案,进一步降低外部开发者使用TPU的门槛。

这一进展对英伟达的市场地位构成直接挑战。Anthropic已承诺采购逾百万颗TPU,成为谷歌TPU最大外部用户,预计到2029年其TPU使用量将超过DeepMind自身。SemiAnalysis认为,随着分离式预填充/解码(PD Disaggregation)、投机解码等关键优化陆续落地,TPUv7的竞争力将进一步提升,并有望在全性能曲线上与英伟达GB200/GB300 NVL72形成正面竞争。

每美元性能:Ironwood在多数场景领先Blackwell

SemiAnalysis的测试以Qwen3.5 397B FP8模型为基准,在聚合服务(Aggregated Serving)模式下对TPUv7 Ironwood与英伟达B200、B300进行了逐项对比。

在低并发、高交互性场景下,Ironwood的成本优势最为突出。在每秒20个token/用户的交互速度下,Ironwood每芯片总吞吐量达9,364 token/秒,高于B200的8,903和B300的8,925,吞吐量领先约5%。结合更低的每小时芯片成本,Ironwood每美元可产出的token数量比B200高出50.4%,比B300高出96.0%。

在端到端延迟维度,Ironwood同样保持竞争力。在20秒中位响应时间下,Ironwood每百万token成本约为0.098美元,低于B200的0.106美元和B300的0.132美元,分别便宜约8%和25%。

不过,测试结果并非全面领先。在约30秒中位响应时间附近的局部区间,B200在每美元性能上仍可超越Ironwood。此外,当英伟达GPU采用FP4精度时,由于Ironwood尚不支持原生FP4计算,英伟达在该精度下仍保持优势——这一差距预计将在下一代TPUv8i中弥补,后者将具备原生FP4加速能力。

在"非同等对比"场景下,即GB300 NVL72采用分离式服务、TPUv7仍使用聚合服务时,GB300在中等端到端延迟区间具备约30%的每美元性能优势。SemiAnalysis认为,一旦TPUv7的分离式服务优化完成,这一差距将消除。

TorchTPU:构建原生PyTorch生态,取代TorchAX

谷歌TPU外部化的核心瓶颈长期在于软件栈。此前,外部开发者需通过TorchAX将PyTorch模型转译为JAX执行,这一路径在低级优化、分页注意力机制以及与vLLM工作模型的适配上存在诸多问题。

新一代TorchTPU方案通过PyTorch的PrivateUse1后端扩展点,将TPU直接暴露为原生PyTorch设备(device="tpu"),开发者可使用.to("tpu")调用,并保留熟悉的分布式接口(包括DDP、FSDP2、DTensor等)。编译路径上,TorchDynamo和AOTAutograd生成FX计算图,TorchTPU将其降低为StableHLO,再由XLA生成TPU可执行代码,底层Pallas内核仍负责性能关键操作。

这一架构变化的实际意义在于:vLLM和SGLang可以复用更多上游模型代码、调度器和API逻辑,而无需跨越PyTorch到JAX的框架边界重新构建。SemiAnalysis指出,Inferact、RadixArk和Red Hat均在与谷歌合作,推动TorchTPU成为vLLM和SGLang的一级支持后端。

目前,TorchTPU仍处于私有测试阶段,预计于10月PyTorch大会期间开源。谷歌计划在Qwen3.5 397B完成初步适配后,进一步支持Kimi K3、GLM5.3以及谷歌自有开源模型Gemma4。SemiAnalysis预计,一旦少数模型完成优化,新增模型支持的边际成本将大幅下降,TPU有望进入vLLM和SGLang的"Day 0"支持名单。

内核优化:数百工时换来的吞吐量提升

为支撑上述性能数据,谷歌工程团队对TPU推理内核进行了大量针对性优化,涵盖注意力机制、MoE路由、混合模型状态管理等多个层面。

在注意力并行方面,针对Qwen3.5的GQA层(32个查询头、仅2个KV头)的不均匀分布问题,TPU后端新增了8路注意力数据并行(DP8)与8路专家并行(EP8)的组合支持,避免了不必要的All-to-All通信开销。

在通信优化方面,谷歌将专家ID和路由权重合并为单次All-Gather操作,在DeepSeek-V3测试中每层节省约80微秒;跨58层累计可节省约4.64毫秒每次前向传播。此外,ReduceScatter集合操作被迁移至SparseCore执行,并结合双缓冲技术实现计算与通信的流水线重叠,在8k1k工作负载下,并发64至512的吞吐量提升幅度为4.1%至14.2%。

在MoE路由内核方面,通过将token的不规则重排移至SparseCore处理、对专家权重实施三重缓冲、以及针对小批量场景引入独立排列路径,8k1k服务吞吐量在并发64时提升7.3%,在并发128时提升5.1%。

在混合模型状态管理方面,将循环状态(Recurrent State)存储精度从FP32降至BF16,在保持VMEM内FP32算术精度的同时,将HBM占用减半,1k8k吞吐量在并发512时提升15%。通过优化KV缓存页面布局(序列维度置于128-lane轴),可用KV页面数量从5,141增至10,283,在并发128的8k1k测试中吞吐量提升16.5%,中位TTFT下降95%。

Ironwood硬件架构:协同设计驱动成本优势

SemiAnalysis认为,谷歌在推理成本上的优势根源在于芯片、互联网络与编译器的协同设计,而非单纯的单芯片算力堆叠。

TPUv7 Ironwood打破了TPU v4和v5p时代的"MegaCore"设计惯例,改为在单颗芯片上集成两个独立计算裸片,通过高带宽裸片间链路连接,JAX等框架将其暴露为两个独立逻辑设备。每颗芯片配备2个TensorCore和4个第三代SparseCore,HBM容量约为上一代Trillium的6倍,并首次引入原生FP8硬件支持。

矩阵乘法单元(MXU)方面,Ironwood采用256×256的脉动阵列,每周期可执行65,536次乘加运算,是此前128×128设计的4倍。但这也带来了对模型形状的严格要求:矩阵维度需填充至256的倍数,否则将造成MXU利用率损失。以Llama 3 8B为例,其注意力头维度为128,在Ironwood上两个注意力矩阵乘法的MXU利用率上限仅为50%。

在芯片互联方面,Ironwood延续了3D环形拓扑(3D Torus),基本构建单元为64颗芯片组成的4×4×4立方体,通过光学电路交换机(OCS)可扩展至9,216颗芯片的超级Pod,总算力达42.5 FP8 exaflops。ICI网络绕过主机CPU,支持芯片间直接交换激活值和梯度,在整个Pod范围内提供接近NVLink级别的带宽。

前瞻:TPUv8i与软件路线图

谷歌新发布的第八代TPU首次将训练与推理拆分为两款独立芯片:TPU 8t面向训练,TPU 8i面向推理。

TPUv8i(代号Boardfly)以高基数交换机构成的扁平化层次网络取代3D环形拓扑,在1,024至1,152颗芯片规模下,网络直径从约16跳降至约7跳,降幅超过50%,有助于降低MoE路由和多轮智能体工作负载中的尾延迟。TPUv8i还配备19.2 Tb/s的ICI带宽(较上一代翻倍)和384 MB片上SRAM(较上一代增加3倍),专为在片上缓存推理和智能体模型的KV缓存而设计。此外,TPUv8i将支持原生FP4计算,SemiAnalysis认为其有望与英伟达Rubin NVL72形成正面竞争。

在软件路线图方面,谷歌当前优先推进的外部化工作包括:投机解码(Speculative Decoding/MTP)优化、预填充-解码分离式服务(PD Disaggregation)、KV缓存DRAM卸载,以及对多轮智能体工作负载(AgentX)的支持。谷歌已开源TPU-Sync(原TPU-raiden)KV缓存传输库,并计划支持业界标准的Mooncake Store卸载方案。

SemiAnalysis总结认为,与AMD相比,谷歌拥有数十年软件工程积累和成熟的质量驱动文化,预计外部TPU软件栈的成熟速度将显著快于竞争对手。"罗马不是一天建成的,但TPU外部化正在以极快的速度推进。"

Disclaimer: Investing carries risk. This is not financial advice. The above content should not be regarded as an offer, recommendation, or solicitation on acquiring or disposing of any financial products, any associated discussions, comments, or posts by author or other users should not be considered as such either. It is solely for general information purpose only, which does not consider your own investment objectives, financial situations or needs. TTM assumes no responsibility or warranty for the accuracy and completeness of the information, investors should do their own research and may seek professional advice before investing.

Most Discussed

  1. 1
     
     
     
     
  2. 2
     
     
     
     
  3. 3
     
     
     
     
  4. 4
     
     
     
     
  5. 5
     
     
     
     
  6. 6
     
     
     
     
  7. 7
     
     
     
     
  8. 8
     
     
     
     
  9. 9
     
     
     
     
  10. 10