mirror of https://github.com/llvm/torch-mlir
f49ebf1690
This replaces the ad-hoc use of `i64` throughout the Torch layer, and helps to keep it crystal clear the distinction between `!torch.int` (which is modeling the Python `int` type) and the various types that serve as dtypes of tensors, which are a totally different type universe. Changes: - `!torch.int` type and C bindings. - Change `torch.constant.int` parser to not need the `: i64` at the end. - `m_TorchConstantInt` matcher to aid with matching constants. - BackendTypeConversion changes for `!torch.int` -> `i64` type conversion. - Refactor finalizing patterns in FinalizingBackendTypeConversionPass (they were getting very repetitive). - Mechanical rewriting of `!torch.int` to `i64` in all the tests, and `AnyTorchIntType` to `Torch_IntType` in the `.td` files. |
||
|---|---|---|
| .. | ||
| acap_dispatch.cpp | ||
| acap_dispatch.h | ||
| class_annotator.cpp | ||
| class_annotator.h | ||
| debug.cpp | ||
| debug.h | ||
| func_builder.cpp | ||
| func_builder.h | ||
| function_importer.cpp | ||
| function_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 | ||