torch-mlir/python/torch_mlir
Bruce Kim 40913a36c2
[MLIR][TORCH] Add E2E support for aten.empty_strided decomposition op (redo PR) (#2459)
Making the same PR with #2457, as I accidentally thought the review was already made and merged it (reverted).

Add decompose empty_strided op.
Referring to #1776, this decomposition op only supports default stride values, because accessing the tensor or indexing over that, the indices are determined by the strides.
In MLIR, this is not implicitly supported but assumes that the strides are default while iterating over the tensor.
2023-09-13 10:04:31 -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] Tensor[]? support operands type support using partial codegen (#2410) 2023-08-30 06:29:39 -04:00
dialects [MLIR][TORCH] Add E2E support for aten.empty_strided decomposition op (redo PR) (#2459) 2023-09-13 10:04:31 -07: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