PyTorch 编译陷阱:FSDP 导致 MFU 下降

SeunghyunSEO7 · x · 2026-08-24

开发者发现在使用 torch.compile 且设置 dynamic shape=True 训练 Flux.2 klein 图像模型时,系统会自动跳过对 FSDP(全分片数据并行)模块的 guard 安装。这一行为导致模型利用率(MFU)显著下降,建议在多分辨率训练场景下注意此性能陷阱。

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

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →