llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib
History
Vivek Khandelwal 3d95c3d6c9 [MLIR][TORCH] Add value tensor variant to aten::_index_put_impl_ 
This commit adds the op `ValsemVariantAtenIndexPutImplOp` that represents
`Aten_IndexPutImpl_Op` without the underscore. This is needed to
make sure that the `ReduceOpVariants` pass turns the in-place op
into an op that takes value tensors as inputs, otherwise the
`MaximizeValueSemantics` pass will not be able to add value
semantics correctly.

This commit also adds the lowering of `ValsemVariantAtenIndexPutImplOp` op.

This commit also updates the `torch.bincount` op test cases.
 		2022-03-16 22:02:02 +05:30
..
CAPI [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
Conversion [MLIR][TORCH] Add value tensor variant to aten::_index_put_impl_ 2022-03-16 22:02:02 +05:30
Dialect [MLIR][TORCH] Add value tensor variant to aten::_index_put_impl_ 2022-03-16 22:02:02 +05:30
RefBackend LLVM bump 2022-02-16 15:28:13 -05:00
CMakeLists.txt Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00
InitAll.cpp Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00