torch-mlir/python/torch_mlir_e2e_test
Bruce Kim a520d39f84
[MLIR][TORCH] Add device "cpu" support for aten.to.dtype_layout op (#2481)
This PR adds device="cpu" support for `aten.to_dtypeLayout` op and
corresponding e2e test suit.
(refer:  PR https://github.com/llvm/torch-mlir/pull/812/)
2023-09-25 10:00:19 -04: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 device "cpu" support for aten.to.dtype_layout op (#2481) 2023-09-25 10:00:19 -04: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