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
