三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

PP pipeline并行算法总结

PP pipeline并行算法总结

ZBV思路有点类似1F1B-Interleaved, 上图说的chunk0是按模型切的不同的virtual pipeline stage(如layer0), chunk1是layer5. 所以pp通信量会增加vps倍。

1F1B-Interleaved 和virtual pipeline stage的原理:

DualPipe

上面蓝色框对应下的面这一部分。

DualPipeV

对比维度DualPipeDualPipeV关键解读
核心调度模式双向对称:数据从两端同时流入,设备两两镜像。V形调度:数据单向流入,在末端折返,形成一个“V”形路径。DualPipeV用更巧妙的单行道实现了双向流的效率。
参数内存2倍(2×)1倍(1×)这是DualPipeV最核心的优势!它直接消除了参数冗余,解决了之前讨论的显存翻倍问题。
所需设备数PPPP/2在达到相同流水线深度(PP)时,DualPipeV只需要一半的设备。
流水线气泡(PP/2-1)(F&B+B-3W)(PP/2-1)(F&B+B-3W)性能无妥协,两者的气泡大小公式完全相同,保持了极高的并行效率。
激活内存PP+1PP+1两者激活内存占用基本相同。
流水线通信量1倍 (1×)2倍 (2×)这是DualPipeV为了消除参数冗余而付出的新代价:设备间的通信量翻倍。
← 返回列表