torch-mlir/lib/Conversion/TorchToLinalg
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
..
CMakeLists.txt Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
CustomOpExample.cpp Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
DataMovement.cpp [LINALG] Add 0-rank case for `aten.permute` (#940) 2022-06-14 12:50:46 -07:00
IndirectDataMovement.cpp [LINALG] Bug fix i64 vs i32 type comparison. 2022-04-22 08:09:58 +05:30
Linear.cpp Handle `nn.Linear(..., bias=False)` case for TorchToLinalg (#919) 2022-06-08 21:13:43 -05:00
Pooling.cpp [MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op 2022-05-27 07:56:37 +05:30
PopulatePatterns.h Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
Random.cpp Improve "pseudo" op terminology. 2022-03-15 17:57:52 -07:00
Reduction.cpp torch,linalg: add support for translating aten.linalg.vector_norm (#839) 2022-05-19 15:48:15 -07:00
TensorConstructors.cpp [LINALG] Add support for preserve memory format in aten_empty_like op. 2022-05-10 09:37:55 +05:30
TensorScalarInterop.cpp [MLIR][TORCH] Add E2E support for ScalarImplicit, Int.Scalar op 2022-05-10 22:40:49 +05:30
TorchToLinalg.cpp Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
Uncategorized.cpp [MLIR][TORCH] Add E2E support for aten.div.Tensor_mode op 2022-06-07 22:26:44 +05:30
Utils.cpp torch,linalg: add support for translating aten.linalg.vector_norm (#839) 2022-05-19 15:48:15 -07:00
Utils.h torch,linalg: add support for translating aten.linalg.vector_norm (#839) 2022-05-19 15:48:15 -07:00