llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/examples/torchfx
History
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
..
examples WIP implementation of torchfx (#304) 2021-09-22 10:27:55 -07:00
annotator.py WIP implementation of torchfx (#304) 2021-09-22 10:27:55 -07:00
builder.py WIP implementation of torchfx (#304) 2021-09-22 10:27:55 -07:00
loc.py WIP implementation of torchfx (#304) 2021-09-22 10:27:55 -07:00
torch_mlir_types.py WIP implementation of torchfx (#304) 2021-09-22 10:27:55 -07:00