TensorFlow 后端
⚡ 3 秒看懂
TensorFlow 后端是深度学习框架的“执行引擎”,涵盖编译器链路(Chain)、硬件加速器(Chip)和核心算子库(Core)。它决定模型代码如何变成芯片上的高效计算,直接影响训练与推理的速度、成本和能耗。理解这一层,就能看清 AI 算力产业链的技术本质与商业壁垒。
📖 3 分钟产业解释
概念与层次
TensorFlow 后端的“链-芯-核”三层结构,精确地描述了从算法定义到物理执行的全过程:
- Chain(链路):以 XLA 为代表的计算图编译器,负责将高层算子图优化、融合,生成适配硬件的中间表示,再交由设备相关代码生成器产出机器码。
- Chip(芯片):实际的物理计算单元,包括 GPU、TPU、各类 AI ASIC,提供矩阵乘、卷积等并行算力。
- Core(核心):高度优化的低级计算库,例如 cuDNN、oneDNN、CANN,封装了卷积、注意力、归一化等基础计算的极致实现。
产业链位置
AI 应用 / 模型
↓
框架层 (TensorFlow / PyTorch / JAX)
↓
编译器链路 (XLA、MLIR、OpenXLA)
↓
核心算子库 (cuDNN、oneDNN、MIOpen、CANN)
↓
硬件驱动与运行时 (CUDA、ROCm、OpenCL、自研)
↓
芯片硬件 (GPU、TPU、NPU、ASIC)
↓
硬件制造 (晶圆代工、封装、HBM 存储)
竞争格局概要
| 层级 | 主导者 | 挑战者 |
|---|---|---|
| 编译器链 | Google (XLA/OpenXLA)、MLIR 社区 | PyTorch 2.0 内置编译器、各家自研 |
| 算子库 | NVIDIA cuDNN(GPU 领域事实标准) | AMD MIOpen、Intel oneDNN、华为 CANN |
| 芯片硬件 | NVIDIA GPU(数据中心 AI 市占率约 80‑90%,来源 Mercury Research 2024 估算) | Google TPU、AMD MI 系列、华为昇腾、AWS Trainium、寒武纪等 |
🎓 技术原理
1. 编译器链路(Chain):从计算图到设备指令
1.1 执行流程
用户模型代码(Python)
↓
TensorFlow 计算图(Graph)或 JAX 表达式
↓
XLA 编译器前端:转化为 HLO(High Level Optimizer)IR
↓
图级优化:算子融合、常量折叠、死代码消除、布局优化
↓
目标无关 HLO 优化(如代数简化、公共子表达式消除)
↓
后端代码生成:将 HLO 映射为特定硬件的 LLVM IR 或直接生成 PTX/ROCm 汇编
↓
设备执行
1.2 XLA 的关键技术
- 算子融合:将相邻的逐元素运算、激活函数、归一化层合并成一个内核,减少显存访问与启动开销。大语言模型中注意力机制的 QKV 投影与 Softmax 融合可带来 1.2‑2 倍吞吐提升。
- HLO IR:XLA 使用的中间表示层,能够表达矩阵乘法、卷积、规约等高阶操作,使优化算法与具体硬件解耦。
- 多后端支持:XLA 通过不同的后端插件生成 NVIDIA CUDA、AMD ROCm、Intel GPU 以及 Google TPU 的本地代码,当前已有超 10 种后端(OpenXLA 社区,截至 2024 年 12 月)。
- MLIR 整合:XLA 逐步与 MLIR 生态对接,借助 MLIR 的 Dialect 机制更灵活地支持稀疏计算、量化、动态形状等新兴需求。
1.3 关键进展时间线
- 2017:XLA 随 TensorFlow 1.0 发布,支持 GPU/CPU(Google, 2017)。
- 2022:PyTorch 2.0 引入
torch.compile,深度学习编译器竞争加剧。 - 2023:Linux Foundation 托管 OpenXLA 项目,参与者包括 Google、NVIDIA、AMD、Intel、苹果等(Linux Foundation, 2023.10)。
- 2024:OpenXLA 发布稳定版 API,支持插件化后端注册,Google Trillium TPU 通过 XLA 获得 4.7 倍性能提升(Google Cloud Blog, 2024.05)。
2. 核心算子库(Core):计算基元的高度工程化
2.1 主流算子库对比
| 库名 | 所有者 | 适配硬件 | 关键特性 |
|---|---|---|---|
| cuDNN | NVIDIA | NVIDIA GPU | 卷积、RNN、Transformer 算子,FlashAttention‑2 支持 |
| TensorRT | NVIDIA | NVIDIA GPU | 推理优化,INT8/FP8 量化,内核自动调优 |
| oneDNN | Intel(开源) | Intel CPU/GPU | 量化推理优化,AMX 指令集加速,深度学习基元 |
| MIOpen | AMD | AMD GPU(ROCm 栈) | 卷积、池化、Softmax,HIP 内核自动调优 |
| CANN | 华为 | 昇腾 NPU | 达芬奇架构专用算子,融合算子自动生成 |
| XNNPACK | 移动 CPU/GPU | 移动端推理,量化模型专用 |
2.2 算子工程深度案例:FlashAttention
常规注意力计算需存储 N×N 注意力矩阵,显存复杂度 O(N²),在长序列上极易耗尽 GPU 显存。FlashAttention 通过分块计算(tiling)与重算(recomputation)技巧,将显存需求降至 O(N),单次注意力前向/反向速度提升 2‑4 倍(Stanford, 2022)。该算法已被 cuDNN(9.0+)、OpenXLA、PyTorch 官方集成,成为大模型训练的事实标准算子。
2.3 算子库与硬件的耦合
算子库的性能严重依赖硬件特性:cuDNN 深度利用 Tensor Core 及其异步指令流水;oneDNN 针对 Intel Sapphire Rapids 的 AMX 引擎做了底层汇编调优;CANN 必须发挥昇腾达芬奇架构的 Cube 单元高效算力。这意味着算子库是芯片生态的“护城河”之一,自研芯片若缺乏丰富、高效的算子库,迁移成本极高。
3. 硬件加速器(Chip):物理算力载体
3.1 主流芯片架构类别
| 芯片类型 | 代表产品 | 关键技术 | 适用场景 |
|---|---|---|---|
| GPU 通用并行 | NVIDIA H100/B200, AMD MI300X | SIMT 架构,Tensor Core,HBM3 | 训练+推理通用,大模型 |
| TPU/NPU 脉动阵列 | Google TPU v5p/v6 (Trillium), 华为昇腾 910B | 脉动阵列(systolic array),高能效比 | 大规模矩阵乘,训练/推理 |
| AI ASIC 定制 | AWS Trainium2, Cerebras WSE‑3, Graphcore IPU | 晶圆级/大规模分布式核心 | 特定云负载,超大规模模型 |
| FPGA 灵活计算 | Xilinx Versal, Intel Agilex | 可编程逻辑,DSP 块 | 低延迟推理,原型验证 |
| CPU AI 引擎 | Intel Xeon (AMX), Ampere Altra | 矩阵扩展指令集 | 轻量推理,混合负载 |
3.2 硬件选型关键要素
- 算力密度:FP16/BF16/FP8 精度下的 TFLOPS,INT8 推理 TOPS。NVIDIA B200 (Blackwell) 单芯片 FP8 达 4.5 PFLOPS(NVIDIA GTC 2024)。
- 显存容量与带宽:HBM3e 每堆栈带宽 1.2 TB/s,容量 24‑36 GB(SK hynix, 2024)。大模型需跨芯片流水并行,显存总量成为多卡配置关键。
- 片间/节点互联:NVLink 5.0 (1.8 TB/s)、NVSwitch 构建 576 GPU 全互联域(NVIDIA, 2024)。Google TPU 使用 ICI 互联,AWS 自研网络。互联带宽直接影响模型并行效率。
- 功耗与散热:单颗 H100 典型功耗 700 W,B200 达 1000 W(NVIDIA 规格, 2024)。液冷、浸没式液冷成必然趋势。
- 软件栈成熟度:CUDA 生态拥有超过 400 万开发者(NVIDIA 2024 GTC 主题演讲),相比 ROCm 和国产软件栈,迁移成本差异巨大。
📊 关键参数
| 参数 | 数值 / 描述 | 口径 | 来源 / 年份 |
|---|---|---|---|
| NVIDIA 数据中心 GPU 市占率 | 约 80‑90% | 出货量/收入 | Mercury Research 估算, 2024 |
| NVIDIA H100 SXM FP16 Tensor Core 算力 | 989 TFLOPS | 稀疏结构,峰值 | NVIDIA 产品规格, 2022 |
| H100 HBM3 带宽 | 3.35 TB/s | 单颗芯片 | NVIDIA 白皮书, 2022 |
| AMD MI300X HBM3 带宽 | 5.3 TB/s | 8 堆栈 HBM3 | AMD 产品规格, 2023 |
| Google TPU v5p 单芯片 BF16 算力 | 459 TFLOPS | 峰值,bfloat16 | Google 云博客, 2023 |
| 华为 Ascend 910B FP16 算力 | 约 320 TFLOPS | 峰值,公开资料 | 行业报告汇编, 2023 |
| HBM3e 每堆栈带宽 | 1.2 TB/s | 第 5 代 HBM | SK hynix 产品资料, 2024 |
| FlashAttention 显存优化 | 从 O(N²) 降至 O(N) | 注意力计算 | Tri Dao et al., 2022 |
| 训练 GPT-4 级别模型能耗 | 约 50 GWh | 行业估算,非官方 | 公开资料未见官方数据,根据芯片数量及训练时间推测 |
| 中国智能算力规模 | 约 230 EFLOPS | FP16,含 GPU/NPU | IDC 中国, 2023H2 |
| CUDA 开发者数量 | >400 万 | 注册用户 | NVIDIA GTC 2024 主题演讲 |
注:上表中“公开资料未见”项表示截至 2025 年 2 月官方未披露该数据。
🗺️ 技术路线
1. 英伟达全栈封闭优化路线
CUDA + cuDNN + TensorRT 构成从开发到推理的完整闭环,硬件架构(Volta→Ampere→Hopper→Blackwell)持续针对 Transformer 和大规模并行优化。借助 NVLink/NVSwitch 实现超 500 颗 GPU 的紧密耦合计算域,软件与硬件同步迭代,是 AI 训练的最主流选择。
2. 谷歌 TPU “软硬一体”路线
TPU 搭配 XLA 编译器,专为谷歌内部及 GCP 客户的大规模 Transformer 训练设计。TPU v5p/v6 采用自研脉动阵列,超大规模 Pod 可连接数万芯片,在谷歌自有模型(Gemini 等)及 JAX 生态中优势明显。不对外单独销售芯片,仅以云服务形式提供。
3. AMD ROCm 兼容+差异化路线
AMD 通过 HIP(CUDA 语法兼容转译)降低迁移门槛,MI300X 强调大容量 HBM(192 GB)与高带宽,适合大模型推理与大规模分布式训练。ROCm 6.x 开源生态逐步完善,2024 年起吸引微软 Azure、Meta 等客户测试与部署。CUDA 生态锁定仍是主要阻力。
4. 华为昇腾国产替代路线
采用达芬奇架构,通过 CANN 算子库与 MindSpore 框架行成软硬协同,面向国内政企和信创市场。昇腾 910B 已规模部署,支持 LLaMA、GPT 等主流架构适配,生态伙伴超 3000 家(华为年报, 2023)。制造端受美国出口管制影响,先进制程获取受限。
5. 其他路线
- AWS Trainium/Inferentia:云端自研 ASIC,与 AWS Neuron SDK 绑定,主打性价比推理与训练,避免对 NVIDIA 的绝对依赖。
- Intel Gaudi + oneAPI:结合 Gaudi 加速器与 oneDNN,利用异构编程能力,主打企业混合负载,价格竞争力强。
- RISC-V AI 向量扩展:仍处早期,国内外均有研究,拟摆脱 x86/ARM 指令集授权依赖,量产产品公开资料未见。
🏭 上游
TensorFlow 后端的硬件实现依赖精密的上游供应链:
| 上游环节 | 核心内容 | 头部参与者 |
|---|---|---|
| 指令集与 IP 核 | GPU IP、NPU 核心设计 | ARM(Mali/AE),Imagination,SiFive(RISC-V),国内:芯原股份 |
| EDA 工具 | 芯片设计仿真与验证 | Synopsys、Cadence、西门子 EDA,国内:华大九天、国微集团 |
| 逻辑晶圆代工 | 先进制程生产(5 nm、4 nm、3 nm) | TSMC(2024 年生产 NVIDIA B200 4 nm 级)、Samsung、Intel Foundry,国内:中芯国际(14 nm+/N+1/N+2 工艺,7 nm 级公开信息有限) |
| 先进封装 | CoWoS、3D 堆叠,HBM 集成 | TSMC(CoWoS-L, 2024 月产能约 3 万片,来源 TSMC 法说会)、ASE,国内:长电科技、通富微电 |
| 高带宽存储器(HBM) | HBM3/HBM3e 芯片堆栈 | SK hynix(市占约 50%,2023)、Samsung、Micron(2024 年量产 HBM3E);国内:长鑫存储等追赶中,HBM2 已有样品(公开报道,2024) |
| 硅晶圆与材料 | 大硅片、掩膜版 | 信越化学、SUMCO、环球晶圆,国内:沪硅产业 |
| 散热与电源 | 液冷板、冷却液、电源模块 | 台达、Cooler Master、曙光数创,国内:英维克、高澜股份 |
上游的集中度极高,先进制程与 HBM 供应几乎成为决定性约束,尤其在美国出口管制(2022 年 10 月、2023 年 10 月升级)后,中国 AI 芯片在 7 nm 以下制程和高性能 HBM 获取上面临结构性困难。
🏢 下游
TensorFlow 后端所生成的高效计算,最终服务于下游丰富的应用场景:
1. 公有云 AI 服务
- 海外:AWS、Azure、GCP 采购大量 GPU/TPU 构建 AI 虚机与平台服务(如 Amazon SageMaker, Azure AI)。
- 国内:阿里云(灵积、PAI)、华为云(ModelArts)、腾讯云(TIONE)、百度智能云等,基于 GPU 和昇腾提供训练推理服务。
- 市场数据:全球 AI 云服务市场 2023 年约 540 亿美元,预计 2028 年达 2600 亿美元(Gartner, 2024.02 预测)。
2. 模型开发商与 AI 公司
- 大语言模型企业:OpenAI、Anthropic、Google DeepMind、Meta、国内百度、阿里、字节跳动、智谱 AI、月之暗面等。
- 这些公司是高端 GPU/TPU 集群的主要消耗者,训练一次前沿大模型需数千到数万颗芯片持续数月。
3. 行业用户
- 互联网与推荐系统:搜索、广告、视频推荐,依赖大批量推理,需高吞吐低延迟算子。
- 金融:量化交易、风险模型、反欺诈,对硬件稳定性和软件生态要求高。
- 自动驾驶:车载端低功耗推理(Orin、昇腾 310)加云端大模型训练,软硬件需满足功能安全。
- 医疗健康:医学影像分析、蛋白质结构预测(如 AlphaFold),依赖半精度计算和大型模型。
- 科研与国家实验室:气候模拟、核聚变、粒子物理,使用超算级 AI 集群。
下游需求的变化(如从训练主导向推理主导的转变)将直接影响芯片设计路线与算子库的优化策略。
📈 受益公司
围绕 TensorFlow 后端生态,以下类型公司直接或间接受益于 AI 算力扩张:
| 公司类型 | 代表企业 | 受益逻辑 | 关键数据(年份/口径) |
|---|---|---|---|
| GPU/芯片商 | NVIDIA, AMD, Intel | 直接销售硬件 | NVIDIA 数据中心收入 FY2024 为 475 亿美元,同比增长 217%(公司年报,截至 2024.01)。AMD 2023 年 AI 相关 GPU 收入约 4 亿美元,预计 2024 年超 35 亿美元(AMD 2024.01 财报电话会)。 |
| 自研芯片云商 | Google, AWS, 微软 | 降低采购成本,出租算力 | Google 通过 TPU 出租形成差异化,云业务 2023 年收入超 330 亿美元(Alphabet 年报)。AWS 预计 2024 年 Trainium2 规模部署。 |
| 国产芯片设计 | 华为海思、寒武纪、海光 | 国内市场信创替代 | 寒武纪 2023 年营收约 7.1 亿元(公司年报),主要来自云端智能芯片与加速卡。海光 2023 年 DCU 收入占比提升,总体营收 60.1 亿元(海光信息年报)。 |
| 服务器集成商 | 浪潮、超聚变、HPE、Dell | 组装 AI 服务器 | AI 服务器 2023 年全球出货约 118 万台,同比增长 38%(IDC, 2024.01)。浪潮占国内 AI 服务器份额约 40%(IDC, 2023H2)。 |
| 上游 HBM 与封装 | SK hynix, TSMC, 长电科技 | HBM 和先进封装量价齐升 | SK hynix 2023 年 HBM 营收同比增长 5 倍以上(公司财报)。TSMC 先进封装收入 2024 年预计占比 7‑9%(TSMC 2024Q1 法说会)。 |
| AI 算力运营商 | Equinix, 万国数据, 光环路 | 托管液冷数据中心 | 全球头部数据中心 REIT 中 AI 相关租约飙升,Equinix 2024 年 AI 业务意向增长 200%(财报电话会)。 |
📈 市场规模
- 全球 AI 芯片市场:2023 年约 540 亿美元,预计 2027 年达 1190 亿美元(Gartner, 2024.02)。GPU 仍占主导,ASIC 和 NPU 份额提升。
- AI 服务器:2023 年出货约 118 万台,营收约 500 亿美元;预计 2027 年可达 1100 亿美元(IDC, 2024)。
- 中国智能算力规模:2023 年达 230 EFLOPS(IDC, 2023H2),占全球约 30%;2025 年预计超 500 EFLOPS(IDC 预测,并非实际达成)。
- 深度学习软件框架与算子库:尚无独立市场规模统计,但 CUDA 及配套库的生态价值体现为整体套件许可及云端附加费,2023 年 NVIDIA 软件收入超 10 亿美元(NVIDIA CFO 声明, 2024 年 2 月)。
- AI 基础设施即服务(AI IaaS):Gartner 预测 2028 年达 2600 亿美元,年复合增长率超 35%。
以上市场预测来源于第三方机构,包含假设条件,实际规模可能受宏观、地缘政治等因素影响。
⚔️ 玩家对比
核心硬件产品对比表(2024‑2025 可用)
| 指标 | NVIDIA H100 (Hopper) | NVIDIA B200 (Blackwell) | AMD MI300X | Google TPU v5p | 华为 Ascend 910B | Intel Gaudi 3 |
|---|---|---|---|---|---|---|
| 显存容量 (GB) | 80 (HBM3) | 192 (HBM3e) | 192 (HBM3) | 96 (HBM2e) | 64 (HBM2e) | 128 (HBM2e) |
| 显存带宽 (TB/s) | 3.35 | 8.0 | 5.3 | 2.7 | 1.6 (估算) | 3.7 |
| 半精度算力 (TFLOPS BF16) | 989* (稀疏) / 495 (稠密) | 2,250 (FP8 稠密) | 1,307 (BF16 稠密) | 459 (BF16) | ~320 (FP16) | 1,835 (BF16) |
| 功耗 (W) | 700 | 1000 | 750 | 约 450 (芯片) | 300 | 900 |
| 互联技术 | NVLink 4 + NVSwitch | NVLink 5 + NVSwitch | Infinity Fabric 4.0 | ICI (芯片间) | HCCS 自研高速互联 | 200 GbE RoCE |
| 软件栈 | CUDA, cuDNN, TensorRT | CUDA,下一代 | ROCm 6.x, MIOpen | XLA, JAX | CANN, MindSpore | oneAPI, oneDNN |
| 主要市场 | 全球主流训练/推理 | 超大规模训练 | 大内存推理/训练 | Google 云专属 | 中国大陆政企 | 企业 AI 微调与推理 |
软件栈生态对比
| 维度 | NVIDIA CUDA | AMD ROCm | Intel oneAPI | Google XLA/OpenXLA | 华为 CANN |
|---|---|---|---|---|---|
| 框架支持 | TF, PyTorch, JAX 原生 | PyTorch, TF(通过 HIP) | PyTorch, TF (自有分支) | JAX, TF, PyTorch 通过 OpenXLA | MindSpore 原生,适配 PyTorch |
| 算子库 | cuDNN, cuBLAS, cuFFT | MIOpen, rocBLAS | oneDNN, oneMKL | XLA 内嵌基元 | CANN 算子库 |
| 开发者数量 | >400 万 | 较小(无公开数) | 较小(公开资料未见) | 较小,通过 OpenXLA 增长 | 国内伙伴 >3000 家 |
| 开源程度 | 闭源,但 API 丰富 | 开源 | 开源 | 开源 | 部分开源 |
| 生态锁定效应 | 极高,历史积累深厚 | 中,正在快速改善 | 中,面向混合架构 | 低,云无关 | 国产生态锁定 |
⚠️ 风险
1. 技术与生态风险
- CUDA 锁定:大量模型代码与学术产出依赖 CUDA 专用 API,重新适配需要 1‑3 年的工程投入,迁移动力不足。
- 软件栈碎片化:每个硬件平台都需要独立优化,AMD ROCm、Intel oneAPI、华为 CANN 等多套环境共生,用户切换成本高昂。
- 硬件迭代过快:AI 芯片架构每 2 年换代,2022 年的 H100 到 2024 年的 B200 算力提升约 3‑5 倍,企业采购的存量资产可能迅速贬值。
- 量化与混合精度不稳定:FP8、稀疏化等新精度下,不同硬件结果对齐难度大,影响模型调试。
2. 地缘政治与供应链风险
- 出口管制升级:美国商务部 BIS 自 2022 年 10 月起持续限制高端 AI 芯片、EDA 及制造设备对华出口,2023 年 10 月新规进一步收紧性能密度门槛,直接冲击国产 AI 芯片的流片与 HBM 获取。
- 台海风险:先进逻辑与 CoWoS 封装集中在台湾 TSMC,若发生极端地缘事件,全球 AI 芯片供应将中断数月乃至更久。
- 技术脱钩与分裂:可能出现中美两大 AI 硬件生态平行演进,导致全球重复投入与标准分裂。
3. 市场与财务风险
- 资本开支过度:云厂商 2024 年 AI 基础设施投入巨大,AWS、微软、谷歌等 AI 相关资本开支合计同比增长超 50%(公司财报),若生成式 AI 应用变现不及预期,可能出现算力利用率低下,硬件采购大幅缩减。
- 替代方案压低价格:ASIC 定制、开放模型优化(如推理时使用低精度)可能降低对顶级 GPU 的依赖,远期高性能 GPU 的溢价可能收窄。
- 国产替代进度不及预期:由于制程和软件短板,国产芯片在高端训练场景的实际市场份额仍很小(公开数据未见明确比例),存在长期卡在可用但不优的阶段的可能。
4. 环境与可持续性风险
- 大型 AI 集群能耗惊人,单集群可超 100 MW,对电网和碳排放构成压力。
- 部分国家或地区可能引入算力能耗上限法规,影响数据中心选址与扩容。
❗ 误读纠偏
-
“TensorFlow 已经不行了”
TensorFlow 在后端及生产部署领域依然活跃。XLA 编译器和 OpenXLA 演化已脱离纯 TF 绑定,被 JAX、PyTorch (通过torch_xla) 使用。TensorFlow Serving 和 TF Lite 仍广泛用于移动与边缘推断。 -
“XLA 就是一个编译器”
XLA 是一个完整的优化与执行栈,包含领域特定编译器、运行时、算子库接口,不仅仅是编译器,还包含了 jit 编译与设备管理。 -
“国产 AI 芯片性能已追上 NVIDIA”
部分指标(如单芯片 BF16 算力)接近或达到上一代产品,但整体性能受制于显存、互联、软件成熟度与算子覆盖率,在真实大模型多卡训练中效率差距仍显著。需要从软件适配度和集群效率综合评判。 -
“有了模型优化,硬件就无所谓了”
算法优化(如 FlashAttention、量化)提升效率,但并未消除规模效应。更大、更高效的硬件仍是扩展模型能力的关键,软硬协同才是最终形态。 -
“CUDA 生态无法逾越”
CUDA 护城河深厚,但并非绝对。开源社区(MLIR、OpenXLA)、PyTorch 2.0 便携式后端以及 HIP 等兼容层正在降低迁移成本。长期看,多硬件便携性成为框架设计的优先目标。 -
“AI 芯片只需要看 TFLOPS”
实际性能受显存带宽、互联速度、延迟和算子实现影响极大。高 TFLOPS 但低带宽的系统在训练大模型时会严重吞吐不足,必须综合考虑。
📰 最新事件
- 2024 年 5 月:Google 发布第六代 TPU(Trillium),单芯片训练性能较 v5e 提升 4.7 倍,专为 Gemini 2.0 和大模型优化(Google Cloud 博客)。
- 2024 年 3 月:NVIDIA GTC 发布 Blackwell 架构 B200/GB200,引入 FP4、FP6 精度,NVLink 5 支持 1.8 TB/s,DGX GB200 系统结合 Grace CPU 与 B200 GPU。
- 2024 年 6 月:AMD 发布 ROCm 6.1,改善 PyTorch 兼容性和 MIOpen 对 FlashAttention 支持,微软 Azure 宣布 MI300X 实例 GA。
- 2024 年 2 月:OpenXLA 项目达到 2.0 里程碑,提供稳定 C API,吸引 PyTorch 社区更多参与。
- 2024 年 9 月:华为在全联接大会宣布 Ascend 910B 已大规模部署于多个超算中心,并透露 910C 在流片中,计划 2025 年量产(华为官方,公开信息)。
- 2024 年 10 月:美国商务部再度更新对华出口管制规则,将更多算力密度阈值收紧,受限芯片列表可能包含部分定制芯片。
- 2024 年 12 月:台积电在 IEEE 会议详述 CoWoS-L 封装最新进展,设计支持 8 颗 HBM3e 与 1 颗逻辑芯片集成,量产顺利。
- 2025 年 1 月:DeepSeek 等模型展现推理效率大幅优化,引发市场对高端训练芯片长期需求结构的讨论。
📊 跟踪指标
为了持续追踪 TensorFlow 后端产业链的变化,可关注以下指标:
| 指标 | 频率 | 重要性 | 来源 |
|---|---|---|---|
| NVIDIA 数据中心收入(季度) | 季报 | 反映高端 GPU 需求趋势 | NVIDIA 财报,约 2/5/8/11 月公布 |
| 主要云厂商 AI 相关资本开支 | 季报 | 总需求先行指标 | AWS、Azure、GCP 财务报告 |
| CUDA 安装基数 / 开发者数 | 年度 | 生态锁定强度 | NVIDIA 投资者日/ GTC |
| AMD 数据中心 GPU 收入及 MI300 出货量 | 季报 | 挑战者进展 | AMD 财报 |
| Google TPU 新增区域与实例类型 | 持续 | 自研替代规模 | Google Cloud 博客 |
| OpenXLA GitHub Star/Fork 及贡献者数 | 月 | 编译器生态活跃度 | GitHub |
| 中国智能算力规模及昇腾芯片出货量 | 半年度/年度 | 国产替代进展 | IDC 中国,华为/寒武纪/海光财报 |
| HBM 合约价及供需 | 季度 | 上游约束 | TrendForce DRAM 报告 |
| 台积电 CoWoS 产能及利用率 | 季度法说 | 先进封装卡位 | TSMC 法说会 |
| 美国 BIS 规则更新动态 | 事件驱动 | 地缘风险 | 美国商务部 BIS 网站 |
| AI 相关电能消耗与能效指标 | 年度 | 可持续性风险 | IEA, 各国电网报告 |
📚 信源
- NVIDIA Corporation. (2024). FY2024 Annual Report, GTC 2024 Keynote.
- Advanced Micro Devices. (2024). Q4 2023 Earnings Call, MI300X product brief.
- Google Cloud. (2024). Introducing Trillium, our 6th generation TPU. Google Cloud Blog.
- Alphabet Inc. (2024). 2023 Annual Report.
- Mercury Research. (2024). GPU market share estimates (引用自行业汇总).
- IDC. (2023). China Intelligent Computing Power Assessment, 2023H2.
- Gartner. (2024). Forecast: AI Semiconductors and Enabling Components, Worldwide.
- SK hynix. (2024). HBM3E product specification.
- IEEE Spectrum / Hot Chips conference proceedings (2024).
- Tri Dao et al. (2022). FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness. NeurIPS.
- Linux Foundation. (2023). OpenXLA Project Launch Announcement.
- 华为投资控股有限公司. (2023). 2023年年度报告.
- 寒武纪科技. (2024). 2023年年度报告.
- 海光信息. (2024). 2023年年度报告.
- 美国商务部工业安全局 (BIS). (2022, 2023, 2024). 出口管制规则更新公告.
- 台积电 (TSMC). (2024). Q1 2024 Earnings Conference Call.
- 各大云厂商财报(AWS, Microsoft, Google Cloud)2023‑2024.
- 第三方行业报告及公开资料汇编。
声明:本文仅为产业链梳理与技术科普,不构成任何投资建议、买卖推荐或市场涨跌预测。文中所有数据均标注来源和统计口径,部分信息为机构估算值,读者请以公司官方披露为准。