[mlir][Vector] Enable masking for ops with index semantics
authorDiego Caballero <diegocaballero@google.com>
Mon, 3 Apr 2023 21:36:13 +0000 (21:36 +0000)
committerDiego Caballero <diegocaballero@google.com>
Mon, 3 Apr 2023 21:58:27 +0000 (21:58 +0000)
commit5217782014f4cb29fb71737e7baa6750a66024e5
treed3b39b185ceddae60c4d56dcfd48b3f5026cd2f3
parent04798db4ea42e201adc4d91a9452235a15fd59e8
[mlir][Vector] Enable masking for ops with index semantics

Masking was already supported for linalg.index and n-D extract but
disabled while waiting for some n-D extract vectorization patches to
land. This patch is just enabling masking for them and adding a couple
of tests.

Reviewed By: ThomasRaoux

Differential Revision: https://reviews.llvm.org/D147359
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
mlir/test/Dialect/Linalg/vectorization.mlir