viable/strict/1787817381: Fix _maybe_view_chunk_cat view shape for negative gather_dim (#194865)
PyTorch 在 2026-08-27 的 viable/strict 和 trunk 发布中修复了 _maybe_view_chunk_cat 对负 gather_dim 的视图形状处理问题(PR #194865)。修复前,当 gather_dim == -1 时,视图目标形状计算 shape[gather_dim + 1:] 会回绕导致 RuntimeError。修复在 torch/_utils.py 和 torch/distributed/_functional_collectives.py 中规范化负 gather_dim,并添加了回归测试。
Development
- First Reportviable/strict/1787817381: Fix _maybe_view_chunk_cat view shape for negative gather_dim (#194865)PyTorch Core
- Industry Responsetrunk/6d164731740e201b9c3f3c6422b14557160ab006: Fix _maybe_view_chunk_cat view shape for negative gather_dim (#194865)PyTorch Core
- Current AssessmentPyTorch 持续修复分布式训练中的边缘情况,表明框架在追求性能优化的同时,也在加强正确性保障。这有助于提升大规模训练稳定性。Agent Pulse · analysis
PyTorch 在 2026-08-27 的 viable/strict 和 trunk 发布中修复了 _maybe_view_chunk_cat 对负 gather_dim 的视图形状处理问题(PR #194865)。修复前,当 gather_dim == -1 时,视图目标形状计算 shape[gather_dim + 1:] 会回绕导致 RuntimeError。修复在 torch/_utils.py 和 torch/distributed/_functional_collectives.py 中规范化负 gather_dim,并添加了回归测试。
该修复针对分布式通信中的视图优化路径,确保负维度(如 -1)被正确规范化,避免形状计算错误。这提示在实现张量操作时,负索引处理需谨慎,尤其是在优化路径中。
PyTorch 持续修复分布式训练中的边缘情况,表明框架在追求性能优化的同时,也在加强正确性保障。这有助于提升大规模训练稳定性。
该修复降低了分布式训练中因负维度导致的运行时错误风险,提升了框架可靠性,对依赖 PyTorch 进行大规模训练的企业有积极影响。
未来可关注 PyTorch 是否会在其他张量操作中统一负维度处理,以及是否会有更多针对分布式通信的修复。