mirror of https://github.com/llvm/torch-mlir
f947443f98
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. |
||
|---|---|---|
| .. | ||
| _torch_mlir_custom_op_example | ||
| dialects | ||
| eager_mode | ||
| __init__.py | ||
| compiler_utils.py | ||