torch-mlir/python
Ashay Rane f947443f98
python: lower `prim::{Load,Store,Enter,Exit}` nodes to torch dialect (#983)
TorchScript nodes like `prim::Load` and `prim::Store` aren't supported
in torch-mlir because they can't be lowered to backends, but such nodes
can occur in the TorchScript IR.

This patch adds a rudimentary translation from such nodes to
corresponding ops in the Torch dialect.  Since we expected such nodes to
go away during lowering because of the SymbolDCE pass, this patch does
not add code to lower these ops beyond the Torch dialect.
2022-06-30 13:17:35 -07:00
..
test Update test to pass with newer versions of tanh (#990) 2022-06-28 20:28:13 -07:00
torch_mlir python: lower `prim::{Load,Store,Enter,Exit}` nodes to torch dialect (#983) 2022-06-30 13:17:35 -07:00
torch_mlir_e2e_test [tosa] aten.embedding op support (#991) 2022-06-30 13:13:52 -07:00
CMakeLists.txt Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00