torch-mlir/python/torch_mlir_e2e_test
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
..
configs Add make_fx_tosa variant to end2end tests (#2240) 2023-07-13 15:07:54 +02:00
debug [torchdynamo] Add "lockstep" numerical accuracy debugger. 2022-12-06 07:57:45 -08:00
linalg_on_tensors_backends build: update llvm tag to 41895843 2023-08-02 21:18:14 +05:30
stablehlo_backends Remove mlir-hlo (replace with stablehlo). (#2460) 2023-09-12 19:10:02 -07:00
test_suite [MLIR][TORCH] Add E2E support for aten.empty_strided decomposition op (redo PR) (#2459) 2023-09-13 10:04:31 -07:00
tosa_backends [TOSA] Add aten._index_put_impl support (#2031) 2023-07-17 09:51:24 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
__init__.py Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
annotations.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00
framework.py [Torch Dialect] Scatter reduce lowering (#1884) 2023-02-21 23:05:55 +00:00
registry.py Ensure that tests have unique names 2022-08-29 16:25:23 -07:00
reporting.py Print name of the backend when tests fail to help debugging issues in CI (#2210) 2023-06-09 10:47:07 +02:00
utils.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00