mirror of https://github.com/llvm/torch-mlir
a46b5c6af2
The `copy_` op being replaced by `RecomposeSliceCopy_` operates on a subset of the tensor being mutated, while the `index_put` op being used to replace the `copy_` op operates on the entire tensor being mutated. This means that the result type of the `index_put` should be the type of the input to `index_put` and we need to make sure that `copy_` does not have users before replacing to avoid type conflicts. This commit also fixes the result type used for the `AtenArangeStartStepOp`, and an off-by-1 error when creating the indices vector. Lastly, this commit also clamps the `end` value from the slice to the size of the dimension. |
||
---|---|---|
.. | ||
main.py | ||
xfail_sets.py |