在 AI 算力需求持续爆发的背景下,如何有效降低大规模集群的互联成本,同时保证高性能和可扩展性,成为众多企业和研究机构面临的核心挑战。燧原科技与中兴通讯近期联合发布的云燧 ESL64-O 超节点,正是针对这一痛点推出的解决方案。该方案将燧原自研的 AI 训练芯片“邃思”与中兴通讯创新的 OEX(Open Express Architecture)架构相结合,旨在通过硬件和架构层面的协同设计,为千卡乃至万卡级别的 AI 训练集群提供一种更高效率、更低成本的互联路径。
对于从事大规模 AI 模型训练、高性能计算(HPC)或数据中心基础设施规划的工程师和架构师而言,理解 ESL64-O 超节点背后的技术逻辑、OEX 架构的设计要点,以及这种方案与传统 InfiniBand 或以太网方案的差异,是进行技术选型和架构设计的关键。本文将深入解析云燧 ESL64-O 超节点的核心组件、OEX 架构的工作原理、部署考量以及在实际场景中的潜在价值。
1. 云燧 ESL64-O 超节点的核心组件与技术定位
云燧 ESL64-O 并非一个单一的硬件产品,而是一个集成了计算、互联和软件栈的“超节点”系统。其核心价值在于通过软硬件协同优化,解决大规模 AI 训练中通信瓶颈这一关键问题。
1.1 自研 AI 芯片“邃思”的计算能力
燧原科技的“邃思”系列芯片是专为 AI 训练设计的高性能处理器。与通用 GPU 不同,邃思芯片在架构上针对张量计算和模型训练中的常见操作(如矩阵乘、卷积等)进行了深度优化。在 ESL64-O 超节点中,多颗邃思芯片通过高速互联接口直接相连,形成一个计算单元。这种设计减少了数据在芯片间传输的延迟,提升了单个节点内的计算效率。
关键的技术指标通常包括:
- 算力密度:单芯片提供的 FP16/BF16/TF32 等训练常用精度的峰值算力。
- 内存带宽与容量:芯片内置的 HBM(高带宽内存)的带宽和容量,直接决定了模型规模和训练速度。
- 片间互联带宽:节点内多芯片之间互联的带宽,如通过 NVLink-like 技术实现的高速直连。
在实际项目中,选择芯片时不仅要看峰值算力,更要关注其在目标模型(如 Transformer、Diffusion 模型)下的实际持续算力表现。
1.2 OEX 架构:打破传统互联瓶颈
OEX(Open Express Architecture)是中兴通讯提出的一种开放性、可扩展的数据中心互联架构。它是 ESL64-O 超节点降低互联成本的核心。传统大规模集群通常采用 InfiniBand 网络,虽然性能极高,但交换机、网卡和线缆的成本也相当昂贵。OEX 架构的创新之处在于:
- 基于以太网的增强:OEX 并非完全抛弃以太网,而是在标准以太网的基础上,通过协议优化和硬件加速,实现了接近 InfiniBand 的低延迟和高带宽,同时继承了以太网生态成熟、成本相对较低的优势。
- 拥塞控制与流量调度:大规模 AI 训练(尤其是 All-Reduce 等集合通信操作)会产生“Incast”流量,极易导致网络拥塞。OEX 架构内置了更智能的拥塞控制算法和流量调度机制,能够有效避免网络热点,保证通信效率。
- 开放性与兼容性:OEX 旨在打造一个开放的标准,避免厂商锁定,允许用户混合使用不同厂商的兼容设备,这在长期运维和成本控制上具有重要意义。
1.3 超节点形态:集成化与模块化设计
“超节点”的概念意味着 ESL64-O 是一个高度集成的系统。它可能将计算单元(多颗邃思芯片)、OEX 交换模块、电源和散热系统集成在一个机箱内。这种设计带来了两个主要好处:
- 降低部署复杂度:用户无需单独采购和组装服务器、交换机、网卡和线缆,以一个“节点”为单位进行机柜部署,极大简化了集群的搭建和扩容流程。
- 优化内部互联:节点内部的计算单元与网络单元之间通过背板或专用接口互联,带宽和延迟优于通过标准 PCIe 插卡的方式。
2. OEX 架构的技术深度解析
要理解 ESL64-O 的优势,必须深入理解 OEX 架构是如何工作的。其核心技术点可以归纳为以下几个方面。
2.1 通信协议栈优化
标准 TCP/IP 协议栈虽然通用,但其复杂的处理流程(如三次握手、拥塞控制、数据包确认)会引入较高的延迟和 CPU 开销。OEX 架构 likely 采用了以下一种或多种优化技术:
- RDMA over Converged Ethernet (RoCE):允许应用程序直接从一台计算机的内存读写数据到另一台计算机的内存,无需操作系统内核介入,大幅降低延迟和 CPU 占用。OEX 可能会对 RoCE 的底层实现进行增强,以更好地适应 AI 流量模式。
- 自定义传输协议:在特定场景下,完全绕开 TCP/UDP,设计一种更轻量、更专注的通信协议,专门为 AI 训练中的集合通信原语(如 All-Reduce, All-Gather)服务。
在软件层面,通常会提供优化的通信库(类似 NCCL),该库能够识别底层 OEX 硬件能力,并调用最优的通信路径。
2.2 网络拓扑与流量工程
对于万卡集群,网络拓扑设计至关重要。常见的拓扑有 Fat-Tree, Clos, Hypercube 等。OEX 架构会结合其硬件特性,推荐或强制使用某种优化的拓扑结构。
- 无阻塞或低阻塞设计:确保集群中任意两个节点之间都有充足的带宽,避免因拓扑限制导致通信瓶颈。
- 自适应路由:当网络中出现故障或拥塞时,数据包能够动态选择其他路径,保证通信的可靠性。
运维人员需要通过管理界面或命令行工具来监控网络流量,识别热点链路。OEX 架构应提供丰富的遥测数据,用于分析和优化网络性能。
2.3 与主流方案的对比:OEX vs. InfiniBand vs. 标准以太网
下表从几个关键维度对比了 OEX 与主流互联方案。
| 特性维度 | InfiniBand | 标准以太网 (RoCEv2) | OEX 架构 (目标) |
|---|---|---|---|
| 性能 | 极低延迟、高带宽,业界标杆 | 延迟和带宽取决于配置,可能因拥塞导致抖动 | 接近 IB 的性能,通过优化降低抖动 |
| 成本 | 交换机、网卡、线缆成本最高 | 成本相对较低,生态成熟 | 目标介于两者之间,追求更高性价比 |
| 成熟度与生态 | 非常成熟,广泛用于 HPC 和 AI | 最成熟,无处不在 | 较新,依赖合作伙伴生态建设 |
| 可扩展性 | 优秀,但大规模部署成本陡增 | 优秀,易于扩展 | 设计目标为极佳可扩展性 |
| 运维复杂度 | 需要专业知识和工具 | 运维人员熟悉,工具丰富 | 旨在通过集成化设计降低运维复杂度 |
从对比可以看出,OEX 架构的定位是试图在 InfiniBand 的高性能和标准以太网的低成本与开放性之间找到一个最佳平衡点。
3. 部署云燧 ESL64-O 超节点的实践考量
将 ESL64-O 超节点引入现有数据中心或构建新集群,需要从硬件、软件和运维多个层面进行规划。
3.1 硬件环境准备
- 机柜与电源:超节点通常密度较高,需要确认机柜的承重、供电功率(可能涉及高压直流)和散热能力(液冷或强力风冷)是否满足要求。
- 网络布线:虽然 OEX 降低了核心交换设备的成本,但节点之间以及与外界的连接仍然需要物理布线。需要规划好 Spine-Leaf 拓扑下的线缆连接。
- 存储集成:AI 训练需要高速存储(如 NVMe 阵列或分布式文件系统)来提供数据供给。需要确保存储网络(通常是另一个独立的以太网或 InfiniBand 网络)与计算网络之间的带宽和延迟满足需求。
3.2 软件栈与平台集成
ESL64-O 的成功运行离不开完整的软件栈支持。
- 驱动与固件:首先需要安装邃思芯片的驱动和 OEX 网络组件的固件。
- 容器化与调度:现代 AI 训练普遍采用容器化技术(如 Docker)和资源调度器(如 Kubernetes 配合 Kubeflow 或 Slurm)。需要确保有相应的 Device Plugin 或调度器插件能够识别和管理邃思芯片和 OEX 网络资源。
- 开发框架支持:主流的 AI 框架(如 PyTorch, TensorFlow, JAX)必须能够利用邃思芯片进行计算。这通常通过框架的扩展库(如
torch_dtu)来实现。同时,框架的分布式训练模块(如torch.distributed)需要能够调用优化后的 OEX 通信库。
一个典型的集群软件栈层次如下:
- 硬件层:ESL64-O 超节点
- 操作系统层:Linux (特定发行版)
- 容器运行时层:Docker/Containerd
- 调度层:Kubernetes + 自定义资源调度器
- AI 平台层:Kubeflow 或自定义训练平台
- 应用层:PyTorch/TensorFlow 训练脚本
3.3 性能调优与监控
部署完成后,性能调优是持续的过程。
- 计算侧调优:监控每颗芯片的利用率和温度。调整训练脚本的 Batch Size、模型并行策略,以最大化芯片计算效率。
- 网络侧调优:使用
ping、iperf3或厂商提供的专用工具测试节点间网络带宽和延迟。在运行分布式训练时,使用 Profiling 工具(如 PyTorch Profiler)分析通信操作在所有训练步骤中的耗时占比。如果通信成为瓶颈,可能需要调整模型并行的粒度或尝试不同的集合通信算法。 - 全局监控:建立集中的监控系统,收集所有节点的硬件指标(温度、功耗)、计算指标(算力利用率)和网络指标(端口流量、丢包率),以便快速定位故障和性能瓶颈。
4. 常见问题与排查思路
在部署和运行基于 ESL64-O 的集群时,可能会遇到以下几类典型问题。
4.1 节点无法发现或通信
- 现象:集群管理平台无法识别新加入的节点,或节点之间无法 Ping 通。
- 排查路径:
- 物理连接:检查网线是否插紧,交换机端口指示灯是否正常。
- 网络配置:检查每个节点的 IP 地址、子网掩码、网关是否配置正确,且没有冲突。确认 VLAN 等配置是否符合规划。
- 防火墙/SELinux:临时关闭防火墙和 SELinux,判断是否是安全策略阻断了通信。
- OEX 固件与驱动:确认所有节点的 OEX 网络组件固件和驱动版本一致且为推荐版本。
4.2 分布式训练性能不达预期
- 现象:训练速度远低于理论算力估算,或者增加卡数后加速比不明显。
- 排查路径:
- 通信库检查:确认训练程序是否正确链接了优化的 OEX 通信库,而非标准的 Gloo 或未优化的 NCCL。
- 网络拓扑检查:使用厂商提供的拓扑发现工具,确认物理连接是否符合预期的无阻塞拓扑(如 Clos)。是否存在多个节点争抢同一上行链路的情况。
- Profiling 分析:使用 Profiler 工具运行一个短的训练周期,重点关注
All-Reduce等通信操作的耗时。如果通信耗时占比过高,说明网络是瓶颈。 - 计算瓶颈检查:如果通信耗时占比很低,但整体速度慢,则可能是计算瓶颈。检查数据加载(I/O)是否太慢,或者模型本身在芯片上的计算效率不高。
4.3 硬件故障告警
- 现象:监控系统报告芯片温度过高、风扇故障或网络端口降速。
- 排查路径:
- 环境检查:检查机房温度、机柜风道是否通畅。对于液冷系统,检查冷却液流量和温度。
- 日志分析:登录到目标节点,查看内核日志(
dmesg)和硬件管理控制器(BMC)的日志,获取详细的错误信息。 - 负载检查:确认是否因为长时间 100% 满负载运行导致温度累积。可以考虑设置动态频率调整或调整任务调度策略,避免单一节点持续高负载。
5. 最佳实践与未来展望
采用云燧 ESL64-O 这类新型架构,需要在技术选型和运维管理上遵循一些最佳实践。
5.1 技术选型决策清单
在决定是否采用 ESL64-O 方案前,可以对照以下清单进行评估:
- [ ]业务需求:我们的模型是否需要千卡/万卡级别的规模?训练任务是否是长期的、核心的业务?
- [ ]总拥有成本(TCO):是否对 InfiniBand 的高成本感到压力?是否进行了详细的 TCO 对比分析(包括硬件采购、电费、运维人力)?
- [ ]技术团队能力:团队是否有能力学习和运维一套新的硬件和网络架构?能否获得厂商足够的技术支持?
- [ ]软件生态兼容性:确认所需的 AI 框架、库和业务软件是否已在邃思芯片和 OEX 架构上得到验证和支持。
- [ ]可扩展性规划:未来 2-3 年的算力增长计划是否与超节点的扩容方式匹配?
5.2 运维管理建议
- 基础设施即代码(IaC):使用 Ansible, Terraform 等工具自动化集群的部署和配置管理,确保环境的一致性。
- 渐进式部署:不要一次性替换整个集群。可以先部署一个小型试点集群,运行关键业务进行长时间稳定性测试,再逐步扩大规模。
- 与厂商紧密合作:对于 ESL64-O 这类集成系统,与燧原和中兴的技术支持团队建立畅通的沟通渠道至关重要,尤其是在遇到复杂问题时。
云燧 ESL64-O 超节点代表了 AI 基础设施发展的一个方向:通过软硬件垂直整合与开放架构,来应对算力规模扩张带来的成本和复杂性挑战。虽然其最终的市场表现和生态建设仍需时间检验,但它无疑为行业提供了除传统 IB 和以太网之外的一个重要选项。对于追求极致性价比和自主可控的大规模 AI 算力用户来说,密切关注并审慎评估此类技术,是保持竞争力的必要功课。下一步,可以深入测试其在不同模型(如超大规模语言模型、科学计算模型)下的实际表现,并评估其与异构计算资源(如其他品牌 GPU)混合调度的可行性。