torch-mlir/external/llvm-external-projects/torch-mlir-dialects
Vivek Khandelwal 5dbace239b Extend tm_tensor.scatter op semantic to carry unique_indices attribute
There are cases where the op may update the same indices multiple
times. In this context, we can not parallelize updates. Instead,
we have to execute them sequentially. Adding a boolean attribute
to control the behavior.

Also adding test cases for invalid IR.
2022-02-22 22:02:10 +05:30
..
include Extend tm_tensor.scatter op semantic to carry unique_indices attribute 2022-02-22 22:02:10 +05:30
lib Extend tm_tensor.scatter op semantic to carry unique_indices attribute 2022-02-22 22:02:10 +05:30
test Extend tm_tensor.scatter op semantic to carry unique_indices attribute 2022-02-22 22:02:10 +05:30
tools Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00
CMakeLists.txt Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00