#7291 [ckpt] feat: Node-local multi-sender broadcast in NCCL checkpoint engine
原始 PR · 作者 parinayc20 · 合并时间 2026-08-14 12:40
NCCL 引擎新增多发送者中继模式,权重同步提速约 3 倍
值得精读。核心看点:用 Ray 节点 id 作为 NVLink 可达性代理的拓扑决策、中继本地推导分块边界以避免集合通信死锁的设计、以及 root 等待时机与模式耦合的细节。建议重点关注 `_multi_sender_ranks` 与 `build_topology` 的成员判定逻辑、`split_weight_chunks(meta_only=True)` 的复用方案,以及 review 中关于"边界分叉会导致广播永久挂起"的讨论。合并后建议跟进两件事:修正 `multi_sender` 默认值与 docstring、PR 描述不一致的问题;补充多节点与死锁回归测试。