如何解决国产 AI 芯片下的集合通信瓶颈:深度对比 HCCL 与 NCCL 在 Ring AllReduce 上的差异
如何解决国产 AI 芯片下的集合通信瓶颈:深度对比 HCCL 与 NCCL 在 Ring AllReduce 上的差异 在分布式深度学习中,集合通信(Collective Communication)是决定训练效率的关键。在 NVIDIA ...
如何解决国产 AI 芯片下的集合通信瓶颈:深度对比 HCCL 与 NCCL 在 Ring AllReduce 上的差异 在分布式深度学习中,集合通信(Collective Communication)是决定训练效率的关键。在 NVIDIA ...
处理百万级(1M)上下文长度是大型语言模型(LLM)面临的巨大挑战。传统的自注意力机制(Self-Attention)在序列长度$N$上具有$O(N^2)$的计算复杂度和内存占用,导致单个GPU无法容纳如此巨大的KV Cache和中间激活。...
在现代深度学习分布式训练中,NVIDIA Collective Communications Library (NCCL) 是实现高性能 GPU 间通信的核心工具。NCCL 提供了多种通信算法来优化 All-Reduce、Broadcast...
分布式训练是加速大型模型训练的关键技术。在多GPU或多机环境中,梯度同步(Gradient Synchronization)的效率决定了整体训练的性能上限。在TensorFlow和PyTorch等框架中,NVIDIA的集合通信库(NCCL)...