commit	0c007eecabfc213b77d66577e0a830c6d20ccc07	[log] [tgz]
author	Charitha Saumya <136391709+charithaintc@users.noreply.github.com>	Thu Apr 18 11:13:49 2024 -0700
committer	Copybara-Service <copybara-worker@google.com>	Thu Apr 18 11:26:38 2024 -0700
tree	0dd92a57c28649fd2e6122d5f483b30d407f48e3
parent	f5dac211f7cc5728e4db7f4bad45454a7615b604 [diff]

[mlir][vector] Add support for linearizing Extract, ExtractStridedSlice, Shuffle VectorOps in VectorLinearize (#88204)

This PR adds support for converting `vector.extract_strided_slice` and
`vector.extract` operations to equivalent `vector.shuffle` operations
that operates on linearized (1-D) vectors. `vector.shuffle` operations
operating on n-D (n > 1) are also converted to equivalent shuffle
operations working on linearized vectors.

GitOrigin-RevId: c577f91d266b74d1b5df475fa2dce7c47fc6c57e

include/mlir/Dialect/Vector/Transforms/VectorRewritePatterns.h[diff]
lib/Dialect/Vector/Transforms/VectorLinearize.cpp[diff]
test/Dialect/Vector/linearize.mlir[diff]
test/lib/Dialect/Vector/TestVectorTransforms.cpp[diff]

4 files changed

tree: 0dd92a57c28649fd2e6122d5f483b30d407f48e3

benchmark/
cmake/
docs/
examples/
include/
lib/
python/
test/
tools/
unittests/
utils/
.clang-format
.clang-tidy
CMakeLists.txt
LICENSE.TXT
README.md

README.md

Multi-Level Intermediate Representation

See https://mlir.llvm.org/ for more information.