torch-mlir

Commit Graph

Author	SHA1	Message	Date
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
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

Author

SHA1

Message

Date

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

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

2 Commits (5917f1dc47f3fb44e0192c5460a0a937f2915f91)