[MLIR] Extend vector.scatter to support tensors
Both vector.scatter and vector.gather only accepted memref in the beginning: https://github.com/llvm/llvm-project/commit/19dbb230a245d3404a485d8684587c3d37c198d3
We extended the vector.gather when we had a need for vectorization: https://github.com/llvm/llvm-project/commit/66c2b76846246fcd1df31e1928b0df44a8fe43e1
It is not a blocker because we plan to use it for vector distribution pipeline, and it can happen on memrefs. However, I think it's worth to complete the functionality of map_scatter decomposition. Filing an issue for tracking it.
cc @banach-space who identified the same problem before: https://github.com/llvm/llvm-project/commit/1e61b374ba3ba2891dc1abda732b0b9263216785
@hanhanW
Could you please provide the current status of this issue?
If it hasn't been addressed yet, I would like to take it on.
No progress since I filed the issue. PRs are welcome. It is an upstream change and may involve others inputs.
Hi @hanhanW , if no one is working on this, could I pick up this issue?
@HalfBloodPrince010 I'm currently working on this issue.
@sakupan102 Sounds good! Sorry missed your comment was 4 days ago.
@hanhanW
I've created a pull request for this issue. Could you please review it?
https://github.com/llvm/llvm-project/pull/165548