[mlir][Vector] Fix `vector.shuffle` folder for poison indices (#124863)
commitc3c326213e80abd6db9da83dbf0ab8452780705c
authorDiego Caballero <dieg0ca6aller0@gmail.com>
Fri, 31 Jan 2025 23:47:47 +0000 (31 15:47 -0800)
committerGitHub <noreply@github.com>
Fri, 31 Jan 2025 23:47:47 +0000 (31 15:47 -0800)
treed66a7bd835068547eb85091d50d9d859fc530781
parentdb7e2e57822142c3fcf82781f8a375d3ea5e20ae
[mlir][Vector] Fix `vector.shuffle` folder for poison indices (#124863)

This PR fixes the folder of a `vector.shuffle` with constant input
vectors in the presence of a poison index. Partially poison vectors are
currently not supported in UB so the folder select v1[0] for elements
indexed by poison.
mlir/lib/Dialect/Vector/IR/VectorOps.cpp
mlir/test/Dialect/Vector/canonicalize.mlir