mirror of https://github.com/llvm/torch-mlir
51d4d55f8a
This commit adds support for multi-dimensional tensors as input to the `_index_put_impl_` op. The support was to some degree already there, since `ScatterOp` already supports multi-dimensional tensors. This commit also adds a bit more error checking to `index_put` and refactors the code for creating `ScatterOp`s to mimic the way one would make a `Linalg::GenericOp`. |
||
|---|---|---|
| .. | ||
| IR | ||
| Transforms | ||
| Utils | ||
| CMakeLists.txt | ||