Commit Graph

1 Commits

Author SHA1 Message Date
Peter Bell
13a3e5c70c 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
2022-03-31 16:16:03 +00:00