Fire-Flyer AI-HPC: A Cost-Effective Software-Hardware Co-Design for Deep Learning

AI Systems and Hardware OLPP DPS ADCEEM
深度学习(DL)和大型语言模型(LLMs)的快速进展大大增加了计算能力和带宽的需求。这与更快的计算芯片和互连成本高昂的事实相结合,显著增加了高性能计算(HPC)的建设成本。为了应对这些挑战,我们介绍了Fire-Flyer AI-HPC架构,这是一个协同硬件-软件设计框架及其最佳实践。对于DL训练,我们使用了Fire-Flyer 2,配备了10,000个PCIe A100 GPU,实现了接近DGX-A100的性能,同时将成本降低了一半,能源消耗降低了40%。我们专门设计了HFReduce来加速allreduce通信,并实施了许多措施,以保持我们的计算存储集成网络无拥塞。通过我们的软件堆栈,包括HaiScale、3FS和HAI-Platform,我们通过重叠计算和通信实现了大规模可扩展性。我们在DL训练方面的系统经验为推动未来的AI-HPC进步提供了有价值的见解。
The rapid progress in Deep Learning (DL) and Large Language Models (LLMs) has exponentially increased demands of computational power and bandwidth. This, combined with the high costs of faster computing chips and interconnects, has significantly inflated High Performance Computing (HPC) construction costs. To address these challenges, we introduce the Fire-Flyer AI-HPC architecture, a synergistic hardware-software co-design framework and its best practices. For DL training, we deployed the Fire-Flyer 2 with 10,000 PCIe A100 GPUs, achieved performance approximating the DGX-A100 while reducing costs by half and energy consumption by 40%. We specifically engineered HFReduce to accelerate allreduce communication and implemented numerous measures to keep our Computation-Storage Integrated Network congestion-free. Through our software stack, including HaiScale, 3FS, and HAI-Platform, we achieved substantial scalability by overlapping computation and communication. Our system-oriented experience from DL training provides valuable insights to drive future advancements in AI-HPC.
许愿