[控场AI]
概念全归约 / AllReduce

all-reduce

一种集合通信原语,用于将分布在多张GPU上的部分结果汇总成完整结果并广播回所有参与方,是张量并行中每层前向计算的关键通信操作,对带宽和延迟高度敏感,环形算法下耗时约与参与卡数成正比。

时间轴 (近 90 天)

10月14日

抽象泄漏的典型场景包括分布式训练卡在all-reduce、混合精度导致梯度溢出、自定义算子不被支持。

待验证50%
10月12日

张量并行每一层前向计算几乎都要执行一次all-reduce等集合通信,通信频率随层数线性增长,对带宽和延迟极度敏感。

待验证50%
10月12日

all-reduce在环形算法下,其耗时约与参与卡数成正比。

待验证50%
10月7日

在all-reduce等集合通信操作中,一块GPU的网络异常足以拖慢整个训练作业

待验证50%
9月23日

在数据并行训练中,每块GPU独立计算mini-batch梯度,然后通过all-reduce将所有GPU梯度求和或取平均再同步回各卡以保持模型参数一致

待验证50%

全部知识事实 (5)

来源文章