Megatron LM 心得
关于 1F1B 流水线并行调度图的讨论
在深入研究 Megatron-LM 源码后,我对官方文档的图示有了新的理解:
“感觉 NVIDIA 官方博客 对 1F1B 气泡图(图 1)有误,目前结合
forward_backward_pipelining_without_interleaving的源码实现来看,认为图 2 才是正确的。”
1F1B 调度对比
| 博客原图(图 1) | 源码推导修正图(图 2) |
|---|---|
![]() |
![]() |
| 官方博客中的 1F1B 示意图 | 基于源码逻辑绘制的实际调度图 |
Written on January 5, 2026

