[mlir][Vector] Add support for masked vector gather ops
authorDiego Caballero <diegocaballero@google.com>
Wed, 15 Feb 2023 06:00:12 +0000 (06:00 +0000)
committerDiego Caballero <diegocaballero@google.com>
Wed, 15 Feb 2023 06:10:22 +0000 (06:10 +0000)
commit1ac874c9aa1859fe67fad110c278588a5a670d78
tree75b1418ff3fa51e6ac46b0e32c16b8771083eb00
parent9452356ddcf44720387eda9b6316b7cadff1f60d
[mlir][Vector] Add support for masked vector gather ops

This patch adds support for masked vector.gather ops using the
vector.mask representation. It includes the implementation of the
MaskableOpInterface, Linalg vectorizer support and lowering to LLVM.

Reviewed By: ThomasRaoux

Differential Revision: https://reviews.llvm.org/D143939
mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
mlir/lib/Dialect/Vector/IR/VectorOps.cpp
mlir/lib/Dialect/Vector/Transforms/LowerVectorMask.cpp
mlir/test/Dialect/Vector/lower-vector-mask.mlir