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
Prateek Gupta 2d75654b2c [TORCH][MLIR] Add lowering of `aten.slice_scatter` and 
`aten.select_scatter` op.

This commit adds:
1.  Lowering of `aten.slice_scatter` op into `tensor.insert_slice`
op.
2. Decomposes the `aten.select_scatter` op into `aten.slice_scater`
op.

Signed-Off-By: Prateek Gupta <gprateek93@gmail.com>
 		2022-07-11 14:07:21 +05:30
..
test torch_mlir.compile: Allow OutputType as a string. 2022-07-08 17:37:27 -07:00
torch_mlir [TORCH][MLIR] Add lowering of `aten.slice_scatter` and 2022-07-11 14:07:21 +05:30
torch_mlir_e2e_test [TORCH][MLIR] Add lowering of `aten.slice_scatter` and 2022-07-11 14:07:21 +05:30
CMakeLists.txt build: improve robustness of cmake and shell scripts (#1018) 2022-07-06 14:39:30 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00