iree icon indicating copy to clipboard operation
iree copied to clipboard

[CodeGen] Fix gather fusion on vector distribute path

Open pashu123 opened this issue 6 months ago • 1 comments

Don't attach lowering config to the gather operation. Let it fuse with the consumer operation. This eliminates the creation of a temporary buffer. The change restricts element-wise fusion in the case of softmax with a gather-like operation. Fixes https://github.com/iree-org/iree/issues/21107

pashu123 avatar Jun 17 '25 15:06 pashu123

@pashu123 if this PR fixes the issue, you can add Fixes https://github.com/iree-org/iree/issues/21107 in the PR description. The issue will be closed when you land the PR.

hanhanW avatar Jun 18 '25 21:06 hanhanW

@Groverkss can you review this please.

MaheshRavishankar avatar Jun 26 '25 19:06 MaheshRavishankar