llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test/Dialect/Torch
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
..
GlobalizeObjectGraph Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
adjust-calling-conventions.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
canonicalize.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
decompose-complex-ops.mlir Improve "pseudo" op terminology. 2022-03-15 17:57:52 -07:00
drop-shape-calculations.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
inline-global-slots.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
invalid.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
maximize-value-semantics.mlir Add maximize-value-semantics support for multiple non-value tensor inputs (#659) 2022-03-15 18:13:45 -07:00
ops.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
prepare-for-globalize-object-graph.mlir Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
promote-types.mlir Introduce new shape library design. 2022-03-15 12:41:58 -07:00
reduce-op-variants-error.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
reduce-op-variants.mlir [MLIR][TORCH] Add value tensor variant to aten::_index_put_impl_ 2022-03-16 22:02:02 +05:30
refine-public-return.mlir Introduce new shape library design. 2022-03-15 12:41:58 -07:00
refine-types-branch.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
refine-types-ops.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
refine-types.mlir Introduce new shape library design. 2022-03-15 12:41:58 -07:00
reify-shape-calculations.mlir Improve "pseudo" op terminology. 2022-03-15 17:57:52 -07:00
simplify-shape-calculations.mlir Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00