[Mlir-commits] [mlir] [mlir][vector] Canonicalize gathers/scatters with trivial offsets (PR #117939)

Ivan Butygin llvmlistbot at llvm.org
Tue Dec 31 09:07:45 PST 2024


Hardcode84 wrote:

> @Hardcode84 Would you have some linalg examples that vectorize into these contiguous gathers? That would be helpful, but no worries if Vector is your actual starting point here.

In my specific case (https://github.com/iree-org/iree-turbine/blob/main/iree/turbine/kernel/wave/codegen.py#L808), I generate gathers directly, without going through linalg. Also as side note, indices are constructed from user-provided sympy exprs, so we don't know beforehand if it's a step or not.

https://github.com/llvm/llvm-project/pull/117939


More information about the Mlir-commits mailing list