torch.compile 在 FSDP 动态形状训练下因跳过 Guard 导致 MFU 降低

SeunghyunSEO7 · x · 2026-08-24

用户发现 torch.compile 在处理 FSDP 包装的模块时会自动跳过安装 guards(尤其是 FSDP2 虽有遗留代码但行为仍在),这在 dynamic shape=True 的训练(如 Flux.2 Klein 图像模型的多分辨率训练)中会导致大量符号参数生成,使得 layernorm、rope 等逐元素操作难以融合,最终显著降低 MFU(模型算力利用率)。

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

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →