Skip to content

01. Why Parallel and Communication | 为什么需要并行与通信 ​

学习问题 ​

单卡走向多卡通常由容量、吞吐或模型结构约束触发。切分解决局部约束,也会把原本位于设备内部的数据依赖变成跨 rank 的通信、等待和调度问题。学习并行的起点不是记住框架参数,而是先判断“什么必须被切开”。

并行问题如何从单卡约束走向多卡证据

核心机制 ​

从约束选择切分对象 ​

同一种性能现象可能对应不同切分方式。先确认受限对象,再估计切分后的交换内容,才能避免用加卡掩盖真正瓶颈。

首要约束优先观察的切分对象通信随之出现在哪里典型机制
训练状态装不下参数、梯度、优化器状态参数收集、梯度归约、状态同步FSDP、ZeRO
模型层序列装不下连续 Block / stagestage 间 activation 与梯度Pipeline Parallel
单层矩阵或 Head 太大特征维、Head、权重矩阵层内 All-Reduce / Gather / ScatterTensor Parallel
长序列状态太大sequence / contextK/V 或中间状态的环式交换Context Parallel
稀疏专家规模扩大expert 与 tokendispatch / combine 的 All-to-AllExpert Parallel
Serving 请求规模扩大请求与模型副本路由、队列与缓存局部性Replica Serving

并行收益必须扣除协同代价 ​

并行方案的收益不能只看单卡显存下降。至少需要同时检查容量、关键路径和系统输出;其中任一项失败,都可能让“能够运行”无法转化为“值得采用”。

证据层需要回答的问题代表指标
容量切分后每个 rank 真正少保存了什么peak memory、可承载模型 / batch / context
通信交换了多少数据,发生多少次,是否落在关键路径payload、frequency、collective time、exposed communication
系统加卡后是否得到稳定的规模或速度收益throughput、step time、P95/P99、scaling efficiency

选择条件与验证证据 ​

目的入口
建立硬件和拓扑直觉Part 01 · 03 GPU 架构与显存、05 通信拓扑
比较切分策略Part 01 · 26 并行策略决策
进入机制实现Part 02 · 27–29、46–49
形成多卡结论07 Benchmark 与并行决策

相关阅读 ​

Released under the MIT License.