FSDP 旧版跳过 guards 导致算子融合困难

SeunghyunSEO7 · x · 2026-08-24

指出 FSDP1 的某些遗留逻辑仍应用于 FSDP2,因其故意跳过安装 guards,导致生成了更多的符号参数,进而使得 LayerNorm、RoPE 等逐元素算子的融合变得困难,最终拖慢了训练速度。

所属事件:torch.compile 遇 FSDP 动态形状训练致 MFU 下降(3 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →