torch-mlir/python
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
..
test torch_mlir.compile: Fix API footgun 2022-06-05 18:10:07 -07:00
torch_mlir [MLIR][TORCH] Add decomposition of aten.floor_divide op 2022-06-14 08:56:25 +05:30
torch_mlir_e2e_test [LINALG] Add 0-rank case for `aten.permute` (#940) 2022-06-14 12:50:46 -07:00
CMakeLists.txt Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00