[mlir][vector] Add patterns to ppropagate vector distribution
authorThomas Raoux <thomasraoux@google.com>
Mon, 6 Jun 2022 23:37:00 +0000 (23:37 +0000)
committerThomas Raoux <thomasraoux@google.com>
Mon, 13 Jun 2022 16:38:50 +0000 (16:38 +0000)
commit1c84800c42d2183a29392c175c8d5f20a4be65d2
tree3617ce68f1503bcebdadaf2728a1786c51f2d1a0
parentc030f467036403bee2d640e910154281377c6f39
[mlir][vector] Add patterns to ppropagate vector distribution

Add patterns to propagate vector distribution and remove dead
arguments. This handles propagation for several vector operations.

Differential Revision: https://reviews.llvm.org/D127167
mlir/include/mlir/Dialect/Vector/Transforms/VectorDistribution.h
mlir/lib/Dialect/Vector/Transforms/VectorDistribute.cpp
mlir/test/Dialect/Vector/vector-warp-distribute.mlir
mlir/test/Integration/Dialect/Vector/GPU/CUDA/test-warp-distribute.mlir
mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp