Edward Z. Yang
f7ee061638
Wconstab/reland pysymint ( #79795 )
...
rebased https://github.com/pytorch/pytorch/pull/79617/ to see if issues are reproducible.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79795
Approved by: https://github.com/malfet
2022-06-20 22:55:06 +00:00
PyTorch MergeBot
44436947bc
Revert "Reland PySymInt ( #79617 )"
...
This reverts commit 8ef6356f26https://github.com/pytorch/pytorch/pull/79617 on behalf of https://github.com/zengk95 due to this is breaking periodic jobs (and maybe pull) on trunk
2022-06-16 19:40:27 +00:00
Nikolay Korovaiko
8ef6356f26
Reland PySymInt ( #79617 )
...
Fixes #ISSUE_NUMBER
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79617
Approved by: https://github.com/Chillee
2022-06-16 04:18:06 +00:00
PyTorch MergeBot
b8db0a0475
Revert "Python Bindings for SymInts ( #78135 )"
...
This reverts commit d332724071https://github.com/pytorch/pytorch/pull/78135 on behalf of https://github.com/ezyang due to broke torchvision tests
2022-06-15 13:52:14 +00:00
Nikolay Korovaiko
d332724071
Python Bindings for SymInts ( #78135 )
...
This PR adds support for `SymInt`s in python. Namely,
* `THPVariable_size` now returns `sym_sizes()`
* python arg parser is modified to parse PyObjects into ints and `SymbolicIntNode`s
* pybind11 bindings for `SymbolicIntNode` are added, so size expressions can be traced
* a large number of tests added to demonstrate how to implement python symints.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78135
Approved by: https://github.com/ezyang
2022-06-14 02:17:59 +00:00
Michael Suo
68e22aa9fc
[symint] add support for negative integers
...
The bit packing scheme is described in the comments.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77913
Approved by: https://github.com/Krovatkin
2022-05-20 03:46:29 +00:00
Nikolay Korovaiko
99339fddd9
move SymInt and SymIntArrayRef to c10/core ( #77009 )
...
Fixes #ISSUE_NUMBER
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77009
Approved by: https://github.com/ezyang , https://github.com/malfet
2022-05-11 16:21:31 +00:00
