pytorch
13a3e5c7 - Catch overflows in calculating storage byte size

Commit
2 years ago
Catch overflows in calculating storage byte size Fixes #73184 In the issue the output tensor's shape is `[2, 4, 536870912, 536870912]` which results in a `numel()` slightly below the point of overflow. When the storage is created it does `numel() * 8` which overflows and a much smaller storage is allocated than required. Pull Request resolved: https://github.com/pytorch/pytorch/pull/73719 Approved by: https://github.com/ezyang, https://github.com/malfet
Author
Committer
Parents
Loading