Intel Gaudi作为Habana Labs研发的AI加速器,以TPC张量处理核为计算单元,通过PyTorch和Hugging Face集成提供了高性价比的训练方案。文章介绍Gaudi的硬件架构、软件栈生态以及在主流大模型训练中的性能与成本表现,为AI算力选型提供Intel方案的参考依据。
需要学习Intel Gaudi:Habana Labs AI加速器、TPC张量处理核、PyTorch/Hugging Face Gaudi集成与性价比优化,Intel Gaudi作为Habana Labs研发的AI加速器,以TPC张量处理核为计算单元,通过PyTorch和Hugging Face集成提供了高性价比的训练方案。AI编译器基础 > 计算图中,Intel Gaudi:Habana Labs AI加速器、TPC张量处理核、PyTorch/Hugging Face Gaudi集成与性价比优化是理解整体架构不可缺失的一环。
目录
可视化演示
本节配有交互动效(进入视口后自动加载,约 3KB)。
Intel Gaudi 与 Habana Labs 概述
Habana Labs 是一家总部位于以色列的 AI 芯片创业公司,专注于高效深度学习加速器设计,于 2019 年被英特尔以约 20 亿美元收购,成为英特尔数据中心 AI 加速器产品线的核心力量,并主导 Gaudi 训练与 Goya 推理芯片的研发。
Intel Gaudi 是面向大模型训练与推理的专用 AI 加速器系列,定位为 NVIDIA GPU 的高性价比替代方案,强调开放架构、标准以太网互联与主流框架原生支持,广泛用于自然语言与计算机视觉任务。
Gaudi 产品线包括第一代 Gaudi、Gaudi2 与 Gaudi3,三代在显存容量、峰值算力与能效上持续跃升,同时保持 SynapseAI 软件栈与 PyTorch 集成的向后兼容性,使既有代码可平滑迁移到新一代硬件。
与封闭互联方案不同,Gaudi 采用 OAM 标准封装与以太网/RDMA 组网,使大规模训练集群可以基于通用交换机灵活构建与横向扩展,从而显著降低互联链路与日常运维成本,这也是 Gaudi 性价比优势的重要来源之一,使企业能以更低门槛构建自有算力。
hl-smi --query-gpu=index,name,memory.total --format=csvTPC 张量处理核架构
TPC(Tensor Processing Core,张量处理核)是 Habana 自研的可编程向量与张量计算单元,与专用的矩阵乘引擎 MME 协同工作,共同构成 Gaudi 的核心算力来源,兼顾峰值吞吐与算子灵活性。
MME(Matrix Multiply Engine)专门用于高密度矩阵乘(GEMM)运算,以获得峰值算力;而 TPC 负责元素级运算、归一化、激活等可编程算子,两者通过片上共享内存高效协作,减少数据搬运开销。
这种 MME 与 TPC 的异构设计让 Gaudi 既能高效执行线性代数,又能灵活支持非标准算子,避免了纯固定功能单元在新型模型上的兼容性问题,从而更好适应快速演进的深度学习网络结构与算子需求,降低新模型落地难度。
以 Gaudi2 为例,单芯片集成 24 个 TPC 与 2 个 MME,FP8 峰值算力较第一代大幅提升,并通过 SynapseAI 编译器将 PyTorch 算子自动映射到对应硬件单元,充分发挥异构架构的协同优势。
Gaudi 系列硬件演进
第一代 Gaudi 提供 32GB HBM2e 显存与片上以太网/RDMA 互联,主要面向中小规模训练,并率先在 ResNet 与 BERT 等任务上展现性价比优势,为后续两代产品的架构演进奠定了基础。
Gaudi2 于 2022 年发布,配备 96GB HBM2e 显存与 24 个 TPC,FP8 算力显著提升,能在单卡上训练更大参数模型,并支持更高速的片间互联带宽,显著增强了大模型训练能力与集群扩展性,是性价比提升的关键一代。
Gaudi3 于 2024 年发布,采用双 die 设计与 128GB HBM3e 显存,内存带宽达约 3.7TB/s,训练性能较 Gaudi2 提升约 4 倍,推理性能也有大幅增长,是三代中算力与显存最强的型号。
三代均采用 OAM(OCP Accelerator Module)标准形态与 Mezzanine 子卡设计,便于服务器厂商集成,并通过标准以太网交换机实现机柜级乃至集群级扩展,显著降低了大规模部署门槛,使数据中心能按需扩容。
export HABANA_LOGS=HL_MALLOC_TRACE:1,HL_GRAPH_COMPILATION:1SynapseAI 软件栈
SynapseAI 是 Habana 提供的完整软件栈,包含驱动、固件、Habana PyTorch 桥接库(habana_frameworks)以及图编译器,负责把高层模型映射到 TPC 与 MME,是连接框架与硬件的关键层。
SynapseAI 的图编译器会对计算图做算子融合、布局变换与内存规划,并支持 lazy 模式与 eager 模式,以兼顾训练吞吐与调试便利性,让开发者能在性能优化与可观测性之间灵活取舍,从而兼顾研发效率与运行性能。
通过 habana_frameworks.torch,开发者可以用原生 PyTorch API 编写模型,仅在关键位置插入少量标记即可在 Gaudi 上运行,迁移成本较低,便于复用既有训练代码与生态工具。
SynapseAI 还提供性能分析工具、recipe 缓存机制与混合精度支持,帮助在大规模训练中提升有效算力利用率(MFU),并通过缓存编译结果避免重复重编译带来的训练启动与切换开销,从而让模型迭代与集群扩容都更加高效顺畅。
import habana_frameworks.torch.core as htcorePyTorch 与 Habana 集成
Habana 通过 habana_frameworks 提供 PyTorch 后端插件,支持将张量与模型迁移到 hpu 设备,并保持与原生 PyTorch 训练循环一致的开发体验,使熟悉 PyTorch 的团队可以快速上手。
在分布式训练中,Habana 实现了基于 Gaudi 集合通信库(HCL)的 DDP 与 FSDP,底层走 RDMA 与以太网,无需依赖专用互联即可扩展到多卡多机,显著降低了大规模训练的网络门槛,便于横向扩展算力。
由于 Gaudi 采用标准以太网与通用交换机进行互联,开发者可以直接复用成熟的数据中心网络基础设施与运维体系,从而显著降低大规模训练集群的部署与扩容难度,同时也避免了被单一厂商的专用互联方案长期锁定,增强了整体供应链与运维的灵活性。
迁移时通常只需把 device 改为 hpu、启用 HPU Graph 与 bf16 或 fp8 混合精度,再结合 SynapseAI 的算子库即可获得接近峰值的训练吞吐,显著缩短从原型到生产的落地周期。
model = model.to("hpu")Hugging Face Optimum Habana 集成
Optimum Habana 是 Hugging Face 与 Habana 合作的加速库,让 Transformers 模型无需大幅改写即可在 Gaudi 上高效训练与推理,降低了自然语言与多模态模型在 Gaudi 上的部署门槛。
它提供 GaudiConfig 配置类与 GaudiTrainer 训练器,内置针对 Gaudi 调优的混合精度、梯度累积、HPU Graph 与分布式策略,开发者通过少量配置即可获得开箱即用的高吞吐训练。
通过官方维护的 Habana 模型仓(如 Habana/bert-base-uncased),用户可直接获取经过调优的 recipe,显著缩短从论文到生产的落地周期,并降低自行调参与适配的工程成本,加速 Gaudi 上的模型上线。
Optimum Habana 还支持 torch_compile 与 fp8 训练路径,在 Gaudi2 与 Gaudi3 上进一步提升大模型预训练与微调的吞吐,使开源大模型在 Gaudi 上的训练更具性价比。
from optimum.habana import GaudiConfig, GaudiTrainer性价比优化与集群部署
Gaudi 的核心性价比优势来自三方面:标准以太网降低互联成本、开放软件栈减少生态锁定、以及 TPC 与 MME 异构架构显著提升实际算力利用率,三者共同压低了单位算力的获取与运维成本,使其在高性价比训练中更具竞争力。
在集群层面,Habana 提供基于以太网与 RDMA 的参考拓扑(HLS),支持从 8 卡单机扩展到数千卡规模,整体 TCO 显著低于依赖专用互联的方案,便于按需弹性扩容与资源周转,适合持续增长的大模型训练需求。
性能优化需关注 batch size、梯度分桶、HPU Graph 重用、recipe 缓存与 fp8 量化,避免频繁重编译与通信气泡造成的算力浪费,从而把有效算力利用率 MFU 维持在较高水平,确保单位算力成本最优。
结合云厂商提供的 Gaudi 实例(如 AWS DL1)或自建机柜,用户可以按需选择训练与推理部署形态,实现单位成本最优的 AI 算力供给,灵活匹配不同业务规模与持续增长节奏,从而降低算力获取门槛并提升资源周转效率。
python run_mlm.py --use_habana --use_lazy_mode --bf16AI 应用场景
大语言模型预训练加速
阶段:模型训练
解决的问题:在有限预算下完成千亿参数级模型预训练,缓解专用 GPU 互联成本高、集群扩展受限的问题。
具体案例:某团队使用 64 张 Gaudi2 组建以太网/RDMA 集群,通过 Optimum Habana 与 Habana DDP 运行 BERT/LLaMA 类预训练,启用 bf16 与 HPU Graph 后单卡吞吐接近峰值,相较同等预算的专用 GPU 方案训练时长缩短约 30%,且集群扩容可直接复用现有以太网交换机。
低成本推理服务部署
阶段:模型推理与服务
解决的问题:在线推理需要高吞吐与低延迟,同时控制单 token 服务成本,避免专用加速器带来的高 TCO。
具体案例:在线客服平台将微调后的对话模型部署到 Gaudi3 推理卡,利用 FP8 量化与 SynapseAI 推理 recipe 把单次推理 batch 提升至原方案的两倍,整体 QPS 提升约 1.8 倍,而每千 token 服务成本下降约 40%,配合 hl-smi 监控实现稳定的多副本弹性扩缩容,在高峰流量下仍保持低延迟。
多机分布式微调与扩展
阶段:数据处理与分布式训练
解决的问题:跨多机多卡的微调任务中,专用互联难以横向扩展且价格昂贵,需要开放互联支撑弹性扩容。
具体案例:研究机构在 8 台服务器共 64 张 Gaudi 上使用 Habana HCL 与 PyTorch FSDP 微调百亿参数大模型,通过梯度分桶与 RDMA 通信压缩将跨机通信占比压到 15% 以内,多机扩展效率保持在 80% 以上,并可根据任务排队情况动态增减节点,显著提升资源周转率与集群利用率,整体 TCO 也更具优势。
Python 演示
import osimport torchimport habana_frameworks.torch.core as htcorefrom transformers import AutoModelForSequenceClassificationfrom optimum.habana import GaudiConfig, GaudiTrainer, GaudiTrainingArguments model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")model = model.to("hpu")gaudi_config = GaudiConfig.from_pretrained("Habana/bert-base-uncased")gaudi_config.use_habana = Truegaudi_config.use_lazy_mode = Truegaudi_config.use_fp16 = Truetraining_args = GaudiTrainingArguments(output_dir="./out", per_device_train_batch_size=32)trainer = GaudiTrainer(model=model, args=training_args, gaudi_config=gaudi_config, train_dataset=train_dataset)trainer.train()
import os导入标准库 os,用于后续设置环境变量或路径管理。import torch导入 PyTorch 主模块,张量与训练循环的基础依赖。import habana_frameworks.torch.core as htcore导入 Habana PyTorch 核心模块,注册 HPU 后端与算子库。from transformers import AutoModelForSequenceClassification从 Transformers 加载预训练序列分类模型,作为待训练网络。from optimum.habana import GaudiConfig, GaudiTrainer, GaudiTrainingArguments导入 Optimum Habana 的配置、训练器与训练参数类。空行,分隔导入区与逻辑区,保持代码可读性。model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")实例化 BERT 序列分类模型并加载预训练权重。model = model.to("hpu")将模型迁移到 Gaudi HPU 设备,类似 CUDA 的 to('cuda')。gaudi_config = GaudiConfig.from_pretrained("Habana/bert-base-uncased")加载针对 Gaudi 调优的官方 GaudiConfig recipe。gaudi_config.use_habana = True显式启用 Habana 执行路径,确保走 SynapseAI 编译。gaudi_config.use_lazy_mode = True启用 lazy 模式以触发图编译与算子融合,提升吞吐。gaudi_config.use_fp16 = True启用 fp16 混合精度训练,提升算力利用率并降低显存占用。training_args = GaudiTrainingArguments(output_dir="./out", per_device_train_batch_size=32)构造训练参数,设置输出目录与单卡 batch size。trainer = GaudiTrainer(model=model, args=training_args,实例化 GaudiTrainer,绑定模型与训练参数(续行)。gaudi_config=gaudi_config, train_dataset=train_dataset)向 GaudiTrainer 传入 GaudiConfig 与训练数据集完成构造。trainer.train()启动训练,SynapseAI 将自动编译并调度到 TPC/MME。
常见问答
Intel Gaudi 与 NVIDIA GPU 在互联方式上的核心区别是什么?
TPC 与 MME 在 Gaudi 架构中各自承担什么职责?
Optimum Habana 在 Hugging Face 生态中的作用是什么?
Gaudi3 相比 Gaudi2 在显存与带宽上有何提升?
在使用 SynapseAI 时,lazy 模式与 eager 模式有什么区别?
常见误区
误区Gaudi 必须使用专用互联才能组建训练集群。
正解Gaudi 基于标准以太网与 RDMA 进行互联,可直接复用通用交换机,无需专用互联链路。
为何常见:人们常把 AI 加速器与专用 GPU 互联默认绑定,误以为所有厂商都依赖私有高速互联。
误区TPC 就是固定功能的矩阵乘单元。
正解TPC 是可编程张量处理核,负责元素级与向量算子;矩阵乘由专用的 MME 承担,二者分工不同。
为何常见:TPC 名称中的 Processing Core 容易被误解为固定功能核,混淆了可编程核与专用引擎。
误区迁移到 Gaudi 需要重写整套 PyTorch 训练代码。
正解通过 habana_frameworks 与 Optimum Habana,通常只需把 device 改为 hpu 并启用相关配置即可,迁移成本较低。
为何常见:对异构加速器缺乏了解,误以为换硬件必然意味着换框架与重写算子。
误区Gaudi3 的 128GB 显存意味着单卡可装下任意大模型。
正解显存容量只是约束之一,还需考虑算力、带宽、通信与并行策略;超大模型仍需张量并行与流水并行跨卡拆分。
为何常见:把显存容量简单等同于可训练模型规模,忽略了算力与通信对训练可行性的影响。
参考来源
- official_documentationIntel Gaudi AI Accelerators英特尔官网关于 Gaudi 系列加速器的产品页,介绍 Gaudi2/Gaudi3 的硬件规格、OAM 形态与目标工作负载。
- official_documentationSynapseAI and Habana DocumentationHabana 官方文档,涵盖 SynapseAI 软件栈、habana_frameworks、图编译器、hl-smi 与分布式训练指南。
- official_documentationOptimum Habana DocumentationHugging Face Optimum Habana 文档,介绍 GaudiConfig、GaudiTrainer 与 Transformers 模型在 Gaudi 上的训练与推理。
- press_releaseIntel Unveils Gaudi 3 AI Accelerators英特尔发布 Gaudi3 的新闻稿,公布双 die 设计、128GB HBM3e、约 3.7TB/s 带宽与较 Gaudi2 约 4 倍训练性能提升。
- code_repositoryHabana Model ReferencesHabana 官方模型参考实现仓库,提供 BERT、LLaMA 等模型在 Gaudi 上的 SynapseAI/Optimum Habana 训练脚本与 recipe。