#38020 [Diffusion] Port the Wan VAE decoder fast paths to the Qwen-Image VAE
原始 PR · 作者 Dayuxiaoshui · 合并时间 2026-09-05 18:02
性能优化
重要性 9.18
洞察度 8.00
将 Wan VAE 解码器快速路径（损失精确的融合卷积缓存、质量门控的 RMSNorm+SiLU 融合与通道最后布局优化）完整移植到 Qwen-Image VAE，消除大量低效算子启动，解码性能提升约 6.5%。
该 PR **值得精读**。理由：1) **设计质量高**：清晰地区分了损失精确（无条件安装）和质量门控两条路径，权衡了性能与精度。将 Wan 特定逻辑重构为通用安装器是良好的工程实践。2) **实现严谨**：修复了一个隐蔽的布局问题（退化步长导致 NCHW 路径），并提供了详尽的精度/性能基准测试数据。3) **测试充分**：6 个新测试用例覆盖了关键功能点，且与源码变更紧密对应。值得关注的设计决策包括：使用 `as_strided` 修复布局而非复制数据（纯视图操作），以及将快速路径安装逻辑集中到一个通用函数以支持多模型。风险点主要在于质量门控路径引入的微小数值差异，但这在扩散模型视觉输出的可接受范围内。