torch-mlir/python/torch_mlir
Ramiro Leal-Cavazos ff762100b8
Add handling of namespaces to library generator (#2391)
When using custom ops, sometimes PyTorch will insert namespaces to the
abstract interpretation function name in the format:
`__torch__.{namespace_1}.{namespace_2}...{op_name}`.  The extra
namespaces are not part of the abstract interpretation function name,
so it needs to be removed before generating the library of MLIR
snippets of abstract interpretation functions. This commit adds
support for removing the namespace information.
2023-08-11 09:56:19 -07:00
..
_torch_mlir_custom_op_example [custom op] Generalize shape library logic to work with dtypes (#1594) 2022-12-13 08:25:41 -08:00
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc LTC->MLIR Debug Info support (#1922) 2023-08-02 10:29:11 -04:00
dialects Add handling of namespaces to library generator (#2391) 2023-08-11 09:56:19 -07:00
__init__.py [MLIR][TORCH] Add support for aten._unsafe_index_put.hacked_twin op 2023-08-11 08:57:01 +05:30
_dynamo_fx_importer.py Bump LLVM (#2176) 2023-06-13 16:17:23 +02:00
_version.py Fix version comparison against stable (#2209) 2023-06-07 10:19:38 +02:00
compiler_utils.py Conform to Python custom exception api 2023-07-19 21:00:55 -05:00
dynamo.py Fix version comparison against stable (#2209) 2023-06-07 10:19:38 +02:00