torch-mlir/python/torch_mlir_e2e_test
Ramiro Leal-Cavazos 93f6d8e776
[LINALG] Add 0-rank case for `aten.permute` (#940)
The function `AffineMap::inferFromExprList` does not work if the first
vector of expressions is empty, because it uses these expressions to
obtain the context. This prevented `aten.permute` from working for
inputs of 0-rank. This commit adds support for 0-rank inputs.
2022-06-14 12:50:46 -07:00
..
eager_backends add ci tests (#754) 2022-05-25 14:59:59 -05:00
linalg_on_tensors_backends llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
test_suite [LINALG] Add 0-rank case for `aten.permute` (#940) 2022-06-14 12:50:46 -07:00
torchscript add ci tests (#754) 2022-05-25 14:59:59 -05:00
tosa_backends llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -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