torch-mlir

Author	SHA1	Message	Date
Ramiro Leal-Cavazos	3aef3a9e30	Remove duplicate example + fix README typo	2021-09-30 13:23:25 -07:00
Sean Silva	bc62a7fbf3	Update to new name of torchscript-module-to-linalg-on-tensors-backend-pipeline	2021-09-30 19:46:05 +00:00
Sean Silva	8b2c099914	Update llvm-project to 204d301bb1921431a853c0bfba32007c018df1d5 This brings in the fix for the obscure RefBackend bug we were hitting.	2021-09-28 17:38:10 -07:00
Ramiro Leal-Cavazos	b59f2cb673	Implement the lazytensor package (#331 ) Implement the `lazytensor` python package for converting lazy computations captured by the Lazy Tensor Core into MLIR. This PR also fixes a few things with `torchfx` and its example	2021-09-28 17:25:06 -07:00
Sean Silva	64ce5d54d3	Update examples. TorchFX example has been simplified, since it seems to be hitting that weird RefBackend bug. Will dig into that.	2021-09-27 17:20:50 -07:00
Ramiro Leal-Cavazos	2b18aad807	Removed import typo in torchfx example	2021-09-22 16:49:46 -07:00
Sean Silva	8779d920b2	Remove "refjit" terminology. We now use RefBackend/refbackend consistently.	2021-09-22 15:41:23 -07:00
Ramiro Leal-Cavazos	1f00f95d2e	WIP implementation of torchfx (#304 ) Implements a python package for taking a `torch.fx.GraphModule` and turning it into MLIR in the `torch` dialect that can then be further compiled by `npcomp`. This is a WIP, so the coverage of PyTorch operations is very small.	2021-09-22 10:27:55 -07:00