为解决智算中心超大规模算力集群算力可用率低、国产技术成熟度低、大规模组网效率存在瓶颈、运营运维复杂等问题,提出了一种基于云计算技术构建智算中心万卡集群的系统。采用18 432块神经网络处理单元(neural processing unit,NPU)卡和优化后的基于以太网的远程直接内存访问(remote direct memory access,RDMA)网络构建云化的智算中心万卡集群,结合软件定义网络(software defined network,SDN)技术实现RDMA网络租户隔离,实现了链路负载均衡误差小于10%,集群All-Reduce带宽达35 GB/s以上。采用优化后的分布式存储协议,实现模型断点恢复时长缩短为原来的1/2。验证结果表明,经过软硬件协同优化,国产化的NPU万卡集群不仅能够满足千亿参数大模型训练的需求,未来更可以支撑万亿参数大模型训练任务。
Abstract
To address issues such as low availability of computing power in ultra-large scale computing clusters of intelligent computing centers
low maturity of domestically produced technologies
bottlenecks in large-scale networking efficiency
and complex operations and maintenance
a system based on cloud computing technology for constructing a ten-thousand card cluster in an intelligent computing center was proposed. A ten-thousand card cluster was constructed using 18 432 NPU units and an optimized RDMA network. A multi-plane network architecture was adopted
in conjunction with SDN technology to achieve RDMA network tenant isolation. The network load balancing strategy was optimized
resulting in a link load balancing error of less than 10% and an All-Reduce bandwidth of over 35 GB/s. By employing the optimized distributed storage protocol
the model’s breakpoint recovery time was reduced to half of its original duration. The validation results demonstrate that the domestic NPU ten-thousand card cluster
with the collaborative optimization of software and hardware
can not only meet the training needs of large models with hundreds of billions of parameters but also support the training tasks of large models with trillions of parameters.
IDC, Inspur Information, Tsinghua Institute of Global Industry . 2022-2023 global computing index evaluation report [R ] . 2023 .
中国软件评测中心 . 人工智能大语言模型技术发展研究报告(2024年) [R ] . 2024 .
China Software Testing Center . Research report on the development of artificial intelligence large language model technology (2024) [R ] . 2024 .
中国信息通信研究院 . 中国算力发展指数白皮书(2022年) [R ] . 2022 .
China Academy of Information and Communications Technology . White paper on China’s computing power development index (2022) [R ] . 2022 .
AN W , BI X , CHEN G T , et al . Fire-Flyer AI-HPC: a cost-effective software-hardware co-design for deep learning [J ] . arXiv preprint 2024: 2408 . 14158 v 1 .
KAPLAN J , MCCANDLISH S , HENIGHAN T , et al . Scaling laws for neural language models [J ] . arXiv preprint , 2020 : 2001 .08361.
Tencent cloud developer community . The application and evolution of large models in machine learning: from the emergence of deep learning of artificial intelligence to generate the type leap [EB ] . 2024 .
阿里云开发者社区 . 10万亿!达摩院发布全球最大AI预训练模型M6 [EB ] . 2021 .
Alibaba Cloud Developer Community . 10 trillion! damo academy releases the world’s largest AI pre-trained model M6 [EB ] . 2021 .
通信产业网 . 万卡集群: 为什么?是什么?怎么建? [EB ] . 2024 .
Communications Industry Network . Wan Ka Cluster: Why? What? How to Build? [EB] . 2024 .
鲍中帅 . 万卡级超大规模智算集群网络运维挑战及实战 [EB ] . 2024 .
BAO Z S . The operation and maintenance challenge and actual combat of super scale intelligent computing cluster network [EB ] . 2024 .
中国移动 . 面向超万卡集群的新型智算技术白皮书(2024年) [R ] . 2024 .
China Mobile . For a new type of wisdom is above all card cluster technology, the white paper (2024) [R ] . 2024 .
WEI Y Z , HU T S , LIANG C , et al . Communication optimization for distributed training: architecture, advances, and opportunities [J ] . arXiv preprint , 2024 : 2403 .07585.
SHOEYBI M , PATWARY M , PURI R , et al . Megatron-LM: training multi-billion parameter language models using model parallelism [J ] . arXiv preprint , 2019 : 1909 .08053.
DONG J B , WANG S C , FENG F , et al . ACCL: architecting highly scalable distributed training systems with highly efficient collective communication library [J ] . IEEE Micro , 2021 , 41 ( 5 ): 85 - 92 .