[mlir][vector] Add unrolling pattern for multidim_reduce op
authorThomas Raoux <thomasraoux@google.com>
Wed, 9 Mar 2022 00:08:11 +0000 (00:08 +0000)
committerThomas Raoux <thomasraoux@google.com>
Mon, 14 Mar 2022 15:22:24 +0000 (15:22 +0000)
commitf69175b1e6aba63ad349644256c58c0e3b3316f1
treed35527ee51a07d5f3949afa4a13b82f4615c5980
parentf00cd27646c08ee2ab68a561c3db34dd8f259951
[mlir][vector] Add unrolling pattern for multidim_reduce op

Implement the vectorLoopUnroll interface for MultiDimReduceOp and add a
pattern to do the unrolling following the same interface other vector
unroll patterns.

Differential Revision: https://reviews.llvm.org/D121263
mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
mlir/lib/Dialect/Vector/IR/VectorOps.cpp
mlir/lib/Dialect/Vector/Transforms/VectorUnrollDistribute.cpp
mlir/test/Dialect/Vector/vector-unroll-options.mlir
mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp