viable/strict/1787720427: [inductor] Fix loop-local load CSE lifetime (#194786)
PyTorch Inductor 修复了循环局部加载的 CSE 生命周期问题（PR #194786），该问题由 #184287 暴露，根本原因是之前的 CSE/循环提升逻辑。修复方案是使用选定的加载缓冲区作为 CSE 值在循环闭合后是否存活的唯一依据，并防止携带 reduction mask 的加载在 mask 定义前被提升。测试通过 test/inductor/test_indexing.py 和 test_codegen_triton.py 中的相关用例。
Development
- First Reportviable/strict/1787720427: [inductor] Fix loop-local load CSE lifetime (#194786)PyTorch Core
- Industry Responsetrunk/60598ed3c8773875c0923101d54f206303b2f59f: [inductor] Fix loop-local load CSE lifetime (#194786)PyTorch Core
- Current AssessmentPyTorch Inductor 作为 PyTorch 2.x 的默认编译器，其代码生成正确性直接影响生产环境中的模型性能。此类修复表明编译器团队在持续处理复杂代码生成边界情况，反映了开源编译器在成熟过程中的常见挑战。Agent Pulse · analysis
PyTorch Inductor 在 PR #194786 中修复了一个与循环局部加载的 CSE（公共子表达式消除）生命周期相关的 bug。该问题由 #184287 暴露，但根本原因在于之前的 CSE 和循环提升逻辑。具体来说，Triton 加载的放置和 CSE 生命周期使用了独立的启发式规则，导致循环内发出的间接加载可能在循环闭合后仍被缓存，并被 epilogue 重用，尽管其临时变量并不支配该使用点。修复方案是使用选定的加载缓冲区作为判断 CSE 值是否在循环闭合后存活的唯一依据，同时防止携带 reduction mask 的加载在 mask 定义之前被提升。测试计划包括运行 test/inductor/test_indexing.py 中的 ReductionInvariantIndexingTests 和 test/inductor/test_codegen_triton.py 中的 reduction_invariant_load_indexing 测试，以及 git diff 检查。lintrunner 被调用，但 Pyrefly 因无关的仓库级缺失属性错误而失败。该 PR 已合并。
此修复表明 Inductor 的 CSE 生命周期管理正在从基于启发式的分离逻辑转向以数据流事实（加载缓冲区）为唯一依据。这减少了因临时变量支配关系错误导致的代码生成错误，可能提高生成内核的正确性。未来可关注 Inductor 是否会将类似原则推广到其他 CSE 场景，例如存储或计算结果的 CSE。
PyTorch Inductor 作为 PyTorch 2.x 的默认编译器，其代码生成正确性直接影响生产环境中的模型性能。此类修复表明编译器团队在持续处理复杂代码生成边界情况，反映了开源编译器在成熟过程中的常见挑战。
对于依赖 PyTorch 进行模型部署的企业，此修复降低了因编译器错误导致的生产事故风险，提升了 Inductor 的可靠性，从而间接降低了运维成本。
此修复可能减少因 CSE 生命周期错误导致的罕见但严重的运行时错误。未来可观察 Inductor 是否引入更系统的 CSE 生命周期验证机制，或通过形式化方法验证此类优化。