跳转至

02 3D 并行 (TP/PP/DP) 硬件映射与通信开销

1. 大模型 3D 并行与硬件拓扑最佳映射

graph TD
    subgraph IntraNode["节点内部 (NVLink 900GB/s+ 超大带宽)"]
        TP["张量并行 (Tensor Parallelism - Megatron)"]
    end

    subgraph InterNode["跨节点集群 (RoCEv2 / IB 400Gbps 网络)"]
        PP["流水线并行 (Pipeline Parallelism - 1F1B)"]
        DP["数据并行 (Data Parallelism - ZeRO-3 / FSDP)"]
    end

    TP -. 通信频次极高 (每层 2 次 AllReduce) .-> IntraNode
    PP -. 通信频次低 (仅阶段边界 P2P) .-> InterNode
    DP -. 阶段聚合通信 (AllGather / ReduceScatter) .-> InterNode