viable/strict/1790375212: Fix int32 overflow in cdist CUDA backward kernel (#198452)
PyTorch 发布 viable/strict/1790375212 版本，修复 cdist CUDA 反向核的 int32 溢出问题（#198452）。该问题为 #128791 的剩余部分：前向崩溃已由 #188006 修复，反向核仍在相同规模下失败。在 2026-09-23 nightly（H200）上，torch.cdist(x1, x2, p=1) 反向在 (b, r, m) = (2, 8192, 32) 即 r1*r2*m = 2^31 时抛出 illegal memory access，而 (2, 8191, 32) 正常；另一组 (32, 8192, 1) 即 dist.numel() = 2^31 时同样失败，(31, 8192, 1) 正常。根因是 cdist_backward_kernel_cuda_impl 用 int 计算索引，而前向核已对相同量使用 int64_t；启动器向上取整网格后，当 dist.numel() 为 2^31 时多余线程的 y >= 2^31，回绕为负并通过 y >= count 检查。
Development
- First Reportviable/strict/1790375212: Fix int32 overflow in cdist CUDA backward kernel (#198452)PyTorch Core
- Current Assessment该修复本身是开源框架的常规缺陷收敛，不构成产业级事件；但它说明大规模张量（元素数逼近 2^31）在 GPU 反向路径上的边界处理仍是框架成熟度的实际短板，影响依赖 cdist 做大规模距离计算的训练与检索负载。Agent Pulse · analysis
PyTorch 修复了 cdist CUDA 反向核中的 int32 溢出（#198452），这是 #128791 的收尾部分：前向崩溃已由 #188006 修复，反向核在相同规模下仍会崩溃。证据显示，在 2026-09-23 nightly 的 H200 上，torch.cdist(x1, x2, p=1) 反向在 (b, r, m) = (2, 8192, 32)（r1*r2*m = 2^31）时抛出 illegal memory access，而 (2, 8191, 32) 正常；另一组 (32, 8192, 1)（dist.numel() = 2^31）失败，(31, 8192, 1) 正常。根因是反向核用 int 计算索引，前向核已改用 int64_t；网格向上取整导致多余线程 y 回绕为负并通过边界检查。修复包含新测试，并在 H200 服务器上运行验证。
这是典型的索引宽度不一致缺陷：同一组量在前向核已用 int64_t，反向核仍用 int，网格向上取整产生的越界线程在 int 回绕后绕过 y >= count 检查，触发非法访存。可验证的下一信号是反向核索引是否统一为 int64_t，以及新增测试是否覆盖 2^31 边界两侧的 (b, r, m) 组合。
该修复本身是开源框架的常规缺陷收敛，不构成产业级事件；但它说明大规模张量（元素数逼近 2^31）在 GPU 反向路径上的边界处理仍是框架成熟度的实际短板，影响依赖 cdist 做大规模距离计算的训练与检索负载。
对使用 torch.cdist 处理大规模成对距离的用户，该修复降低训练或推理中因非法访存而中断的风险；评估价值在于确认所用 PyTorch 版本是否包含 #198452，而非改变成本结构。
可观察的下一信号是后续 nightly 中 cdist 反向在 2^31 边界用例是否稳定通过，以及是否出现同类 int32 索引缺陷的批量清理提交。