torch-mlir

Commit Graph

Author	SHA1	Message	Date
Phoenix Meadowlark	699bf5df45	Add cos_e2e.py, test_utils and support for tensor inputs (#134 )	2020-11-24 19:02:50 -08:00
Sean Silva	ec1336a8a3	Make pytorch/backend/refjit.py a bit tidier - Print out initial PyTorch IR. - Rename ambiguous "frontend IR" to "TCF IR". - Add newlines to prints - Rename FRONTEND_PASSES to TORCH_TO_TCF_PASSES	2020-11-20 17:21:24 -08:00
Stella Laurenzo	b4c7ae1e0c	Repurpose numpy-compiler compiler/runtime flow for PyTorch. * A bit gross because I took the chance to upgrade all of the backend bits to the new MLIR Python bindings and we still co-mingle the old and new for now. * Since the Python created PassManagers are configured for explicit nesting, I had to upgrade some of the pass pipelines to be explicit. * The demo in mul_maximum_e2e.py now compiles, runs through PyTorch and through the JIT, prints and asserts the same results. * I am not claiming that this is the prettiest API in this patch: consider that this is just directly using low-level APIs and there should be an intervening high level API.	2020-11-11 10:38:13 -08:00

Author

SHA1

Message

Date

Phoenix Meadowlark

699bf5df45

Add cos_e2e.py, test_utils and support for tensor inputs (#134 )

2020-11-24 19:02:50 -08:00

Sean Silva

ec1336a8a3

Make pytorch/backend/refjit.py a bit tidier

- Print out initial PyTorch IR.
- Rename ambiguous "frontend IR" to "TCF IR".
- Add newlines to prints
- Rename FRONTEND_PASSES to TORCH_TO_TCF_PASSES

2020-11-20 17:21:24 -08:00

Stella Laurenzo

b4c7ae1e0c

Repurpose numpy-compiler compiler/runtime flow for PyTorch.

* A bit gross because I took the chance to upgrade all of the backend bits to the new MLIR Python bindings and we still co-mingle the old and new for now.
* Since the Python created PassManagers are configured for explicit nesting, I had to upgrade some of the pass pipelines to be explicit.
* The demo in mul_maximum_e2e.py now compiles, runs through PyTorch and through the JIT, prints and asserts the same results.
* I am not claiming that this is the prettiest API in this patch: consider that this is just directly using low-level APIs and there should be an intervening high level API.

2020-11-11 10:38:13 -08:00

3 Commits (b94a859e03b4f42be85e4a8901e384023facb7cf)