torch-mlir/lib/Dialect/Torch
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
..
IR Integrate llvm-project and mlir-hlo. (#2454) 2023-09-12 15:09:57 -07:00
Transforms [MLIR][TORCH] Add E2E support for aten.empty_strided decomposition op (redo PR) (#2459) 2023-09-13 10:04:31 -07:00
Utils [MLIR][TORCH] Add E2E support for view_as_real op (#2419) 2023-09-01 21:12:01 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00