projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 4f969db) | patch
Parse int64 sizes/strides (#64076)
authorBert Maher <bertrand@fb.com>
Sun, 29 Aug 2021 02:57:10 +0000 (19:57 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Sun, 29 Aug 2021 02:58:34 +0000 (19:58 -0700)
commit405c15516c3a052fbad33fa8fcb6cde84ebc97d8
treedf16e918c039bd646cfdc168c51e9d080398bde2tree | snapshot
parent4f969db325a7a70878bd3eae5bbb3fecd598d4cacommit | diff
Parse int64 sizes/strides (#64076)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64076

We were parsing sizes into int32s, so if you had a tensor with more
than 2^32 elements, you couldn't represent it.
ghstack-source-id: 136933273

Test Plan: parseIR with size of 4e9

Reviewed By: ZolotukhinM

Differential Revision: D30521116

fbshipit-source-id: 1e28e462cba52d648e0e2acb4e234d86aae25a3e
torch/csrc/jit/frontend/schema_type_parser.cpp diff | blob | history
Domain: Machine Learning / ML Framework;