4 Commits

Author	SHA1	Message	Date
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) Pull Request resolved: 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