viable/strict/1790375212: Fix int32 overflow in cdist CUDA backward kernel (#198452)
- PyTorch 近 90 天出现 1504 次
- PyTorch 近 90 天第 1485 次发版
- 上一次:同一天稍早 · viable/strict/1790371571
发生了什么
Fixes #128791 . The forward half of that issue was already fixed by #188006 ; this fixes the same crash in the backward kernel. I reviewed the diff and the new test, and the tests below were run on our H200 server. The description below was drafted with Claude Code and reviewed by me: The forward crash reported in #128791 was fixed by #188006 , which caps the cdist/pdist forward grid and grid-strides over the output…
摘要按规则整理自下方来源原文