mirror of https://github.com/llvm/torch-mlir
158c5c484d
This required restructuring of how we model TorchScript on import. The
main difference is that now we split out a `torch.class_type` that holds
methods and declarations of the types of each slot. This is more
consistent with TorchScript (our previous representation was
"denormalized").
Recommended reading order:
1. check out the description of `torch.class_type` in `TorchOps.td` and
look at `test/Dialect/Torch/ops.mlir` and
`frontends/pytorch/test/module_import/` to familiarize with the new
representation.
- Just look at the new IR. The diff between the old names and new
names is confusing.
2. check out `test/Dialect/Torch/globalize-object-graph*.mlir`
and read along with the pass description in
`include/npcomp/Dialect/Torch/Transforms/Passes.td`
3. Read the code in `GlobalizeObjectGraph.cpp` and miscellaneous changes
in `ivalue_importer.cpp`, `TorchOps.cpp`, etc.
|
||
|---|---|---|
| .. | ||
| acap_dispatch.cpp | ||
| acap_dispatch.h | ||
| debug.cpp | ||
| debug.h | ||
| func_builder.cpp | ||
| func_builder.h | ||
| graph_importer.cpp | ||
| graph_importer.h | ||
| ivalue_importer.cpp | ||
| ivalue_importer.h | ||
| mlir_utils.h | ||
| module_builder.cpp | ||
| module_builder.h | ||
| node_importer.cpp | ||
| node_importer.h | ||
| op_builder.cpp | ||
| op_builder.h | ||
| python_bindings.cpp | ||
| torch_to_mlir_utils.cpp | ||
| torch_to_mlir_utils.h | ||