torch-mlir/python/torch_mlir_e2e_test
Quinn Dawkins 11a8901078
[MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097)
- Includes a canonicalizer for `aten.add.t`needed for successfully lowering the shape function
 - Only offers support for statically sized index tensors when there is more than one
 - Dynamic shape support remains for single indexing tensors
2022-07-28 19:00:02 -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 [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -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