llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python
History
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
..
test torch_mlir.compile: Allow OutputType as a string. 2022-07-08 17:37:27 -07:00
torch_mlir [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00
torch_mlir_e2e_test [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00
CMakeLists.txt Add option to expose custom PyTorch repo/branch (#1103) 2022-07-24 20:08:48 -07:00
TorchMLIRModule.cpp Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00