torch-mlir

Commit Graph

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

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

1 Commits (a25163fbfa1ef2769b688a522b47d8185e5d8beb)