torch-mlir/python/torch_mlir_e2e_test
Quinn Dawkins 647e75e029
Allow expanding and collapsing in aten::view (#1082)
- Supports cases where the view op expands and collapses dims
simulataneously. This does not handle the case where it is neither
expanding nor collapsing (e.g. [2, 3] -> [3, 2])
 - Additionally fixes a previous bug with adding 1-sized dims on both
sides of a tensor with aten.view
2022-07-20 17:35:51 -04:00
..
eager_backends add ci tests (#754) 2022-05-25 14:59:59 -05:00
linalg_on_tensors_backends bump llvm tag to 3580daa (#1078) 2022-07-18 16:49:03 -07:00
test_suite Allow expanding and collapsing in aten::view (#1082) 2022-07-20 17:35:51 -04:00
torchscript Fix multi-threaded tests on macOS (#1005) 2022-07-05 00:05:36 -07:00
tosa_backends bump llvm tag to 3580daa (#1078) 2022-07-18 16:49:03 -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
utils.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00