ee6ebfc06b
Revert "Enable dim=None for torch.sum ( #75845 )"
...
This reverts commit e79a51f7db181be2e6e196d6d9d90403022bc465.
Reverted https://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
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
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
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
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
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
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
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
