torch-mlir/python/torch_mlir
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
..
_torch_mlir_custom_op_example Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
dialects [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py [MHLO] Init MHLO integration. (#1083) 2022-07-20 16:18:16 -07:00
compiler_utils.py Add named exception TorchMlirCompilerError. (#1064) 2022-07-15 16:32:36 -05:00