torch-mlir/python/torch_mlir
Jae Hoon (Antonio) Kim 32d9b20bde
Add linspace/cumprod/roll ops (#2498)
Add linspace/cumprod/roll ops to ODS and add shape inference functions
to make it work with LTC.

Also, add some tensor utils to LTC library for searching for non-detach
copy nodes.
2023-10-03 11:01:07 -04: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 Add linspace/cumprod/roll ops (#2498) 2023-10-03 11:01:07 -04:00
dialects Add linspace/cumprod/roll ops (#2498) 2023-10-03 11:01:07 -04:00
__init__.py [MLIR][TORCH] Add e2e support for aten.pow.Scalar op 2023-08-31 21:43:24 +05:30
_dynamo_fx_importer.py [MLIR][TORCH] Add E2E support for view_as_real op (#2419) 2023-09-01 21:12:01 -07: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