llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test/python/importer/jit_ir
History
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
..
ivalue_import Add support for ScalarType::QUInt8 2022-06-29 15:33:28 -07:00
node_import python: lower `prim::{Load,Store,Enter,Exit}` nodes to torch dialect (#983) 2022-06-30 13:17:35 -07:00
get_registered_ops.py Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
lit.local.cfg Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00