viable/strict/1789065007: [BE] Unify reflection/replication padding shape checks (#196457)
PyTorch 发布 viable/strict/1789065007 变更（PR #196457），将 reflection/replication 等六个 padding 算子在 CPU、CUDA、MPS 上重复手写的 shape 检查统一收敛到 Padding.h 中的共享实现，按 rank 与 padding 类型参数化，并让所有后端走同一路径。统一后三个后端报错一致，也与 _meta_registrations.py 中已统一的 meta 实现一致，使 eager 与 torch.compile 行为一致。
Development
- First Reportviable/strict/1789065007: [BE] Unify reflection/replication padding shape checks (#196457)PyTorch Core
- Current Assessment框架层的这类收敛属于基础设施成熟度信号：当同一语义在多个后端被反复手写，错误信息与边界行为就会分叉，用户在不同设备上得到不同结果。把校验集中并统一报错，降低了跨后端调试成本，也让编译路径与 eager 路径的语义对齐。判断上，这对依赖 PyTorch 做多后端部署的团队是正向的稳定性投入，但单次 PR 不构成产业格局变化。Agent Pulse · analysis
该变更把六个 pad 算子分散在 CPU、CUDA、MPS 上十余处的重复 shape 检查集中到 Padding.h，紧邻已有的 check_valid_input，按 rank 和调用方是否为 reflection padding 参数化，并让每个后端都走这些检查。统一后三个后端报出相同错误，且与 _meta_registrations.py 中已统一的 meta 实现一致，因此 eager 与 torch.compile 结果一致。收敛过程中暴露两个 bug：reflection 的“输出过小”判断误用 || 而 replication 用 &&，导致某一维非正而其他维正常时检查被绕过，F.pad 直接进入 allocator 而非抛错；MPS 有自己写坏的同一判断，1d 情况下从未触发。另外 #142834 为 replication 增加的 backward channel 维检查从未应用到 reflection，后者在 gradOutput 不匹配时仍会崩溃，MPS 对两个家族都未校验 channel 维，会触发 Metal 断言，相关回归测试的 @skipMPS 已移除。
这是典型的跨后端校验逻辑去重：把 rank 与 padding 类型作为参数集中到 Padding.h，可消除 CPU/CUDA/MPS 与 meta 实现之间的行为漂移，使 eager 与 torch.compile 报错一致。判断上，共享 backward helper 同时覆盖两个家族，说明修复方向是让校验成为单一事实来源而非逐后端补丁。可验证的下一信号：后续 PR 是否把其他算子族的重复校验也迁入共享头文件，以及是否新增覆盖 1d MPS 与 gradOutput 不匹配的测试用例。
框架层的这类收敛属于基础设施成熟度信号：当同一语义在多个后端被反复手写，错误信息与边界行为就会分叉，用户在不同设备上得到不同结果。把校验集中并统一报错，降低了跨后端调试成本，也让编译路径与 eager 路径的语义对齐。判断上，这对依赖 PyTorch 做多后端部署的团队是正向的稳定性投入，但单次 PR 不构成产业格局变化。
对使用 PyTorch 的团队，统一校验意味着跨 CPU/CUDA/MPS 的错误行为可预期，减少因后端差异导致的线上排查与重试成本；对依赖 torch.compile 的推理服务，eager 与编译路径报错一致可降低灰度与回滚风险。价值大小取决于团队是否真的跨后端部署，单后端用户收益有限。
若该模式延续，PyTorch 可能继续把分散在各后端的输入校验与错误语义收敛到共享实现，并补齐 MPS 等后端的边界测试。可验证的下一信号是后续 release note 中是否出现同类“unify checks”条目，以及 MPS 相关 skip 标记是否继续减少。