John Clow
c8eae2de52
[Shape Fns] Fix optional None for the actual shape functions ( #83092 )
...
I think the person who edited `mean.dim` edited the python and the associated Torchscript version manually in two different ways. This diff fixes that all up. It also fixes the inconsistencies in the `torch.nonzero` generated file
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83092
Approved by: https://github.com/eellison
2022-08-10 18:20:18 +00:00
Kurt Mohler
2bfae07a79
Enable dim=None for torch.mean ( #81286 )
...
Part of #79525
This will require coordination with XLA before merging, just like #79881
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81286
Approved by: https://github.com/albanD
2022-07-28 22:34:56 +00:00
Kurt Mohler
23bdb570cf
Reland: Enable dim=None for torch.sum ( #79881 )
...
Part of #29137
Reland of #75845
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79881
Approved by: https://github.com/albanD , https://github.com/kulinseth
2022-07-09 00:54:42 +00:00
PyTorch MergeBot
ee6ebfc06b
Revert "Enable dim=None for torch.sum ( #75845 )"
...
This reverts commit e79a51f7dbhttps://github.com/pytorch/pytorch/pull/75845 on behalf of https://github.com/malfet due to Breaks MacOS builds, see e79a51f7db
2022-06-16 22:01:41 +00:00
Kurt Mohler
e79a51f7db
Enable dim=None for torch.sum ( #75845 )
...
Part of #29137
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75845
Approved by: https://github.com/ezyang
2022-06-16 20:17:07 +00:00
John Clow
dbee7e5499
Adding SSA support for convolution_backward
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77283
Approved by: https://github.com/Krovatkin
2022-05-20 18:39:47 +00:00
John Clow
2a99018147
Adding a way to register both upper and lower bound functions
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77388
Approved by: https://github.com/eellison
2022-05-18 17:34:07 +00:00
Sean Silva
1136965aa1
Upstream remaining shape functions from Torch-MLIR. ( #76889 )
...
Follow-on to https://github.com/pytorch/pytorch/pull/76592 adding the
rest.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76889
Approved by: https://github.com/eellison
2022-05-13 17:46:58 +00:00
John Clow
db21e22b4b
[EASY] Quick Fix for broken shape function autogen.
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76703
Approved by: https://github.com/eellison
2022-05-03 17:34:05 +00:00
Sean Silva
6b6c63ce5e
Upstream argmax shape function.
...
Keeping this first commit simple to test out the flow. Will bulk-add the
rest once this one goes through.
Shape function taken from:
5192a4e9f3/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/shape_lib_gen.py (L488)https://github.com/pytorch/pytorch/pull/76592
Approved by: https://github.com/eellison
2022-05-03 16:52:09 +00:00
Elias Ellison
f65eb09d6b
[JIT] Move Shape Function definition to python
...
Moves jit shape function registration to python. Like jit decompositions, a script must be run after adding new definitions which serializes them in a c++ file.
This was a request so that torch-mlir could define functions in python and upstream their shape functions. cc @silvasean @makslevental
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75546
Approved by: https://github.com/davidberard98
2022-04-19 20:59:44 +00:00
