torch-mlir/python
Ramiro Leal-Cavazos f204210266
[LINALG] Fix handling of size-1 dims in `aten.view` again. (#992)
A previous fix to the handling of size-1 dims in
`aten.view` (https://github.com/llvm/torch-mlir/pull/962) resulted in
the wrong grouping of dimensions when size-1 dims where between two
dims of size greater than 1. This commit fixes that.
2022-06-30 16:39:25 -07:00
..
test Update test to pass with newer versions of tanh (#990) 2022-06-28 20:28:13 -07:00
torch_mlir python: lower `prim::{Load,Store,Enter,Exit}` nodes to torch dialect (#983) 2022-06-30 13:17:35 -07:00
torch_mlir_e2e_test [LINALG] Fix handling of size-1 dims in `aten.view` again. (#992) 2022-06-30 16:39:25 -07:00
CMakeLists.txt Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00