viable/strict/1790274788: Validate packed RNN batch sizes before backend dispatch (#197560)
- PyTorch 近 90 天出现 1457 次
- PyTorch 近 90 天第 1438 次发版
- 上一次：同一天稍早 · viable/strict/1790256510: [Test] Make test_foreach device-agnostic for OOT backends (#191179)
发生了什么
Issue Fixes ROCm/TheRock#8337 . Summary Validate packed RNN batch_sizes at the shared ATen entry points before selecting cuDNN/MIOpen or the native implementation. This preserves the CPU-only contract when backend capability probing falls back to the native PackedLayer and prevents it from dereferencing device memory on the host. Extend the existing regression test to cover both the normal backend dispatch and the n…
摘要按规则整理自下方来源原文