PyTorch PR 修复 FSDP 模块静态特化问题

ezyang · x · 2026-08-26

该 PR 旨在修复 FSDP2 下模型编译时产生过多符号参数导致 Inductor 融合失败和 MFU 降低的问题。方案是在 skipfsdpguards 下对 FSDP 管理模块的 int/tuple 属性进行静态特化。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →