[mlir][Linalg] Use rank-reduced versions of subtensor and subtensor insert when possible.
authorMaheshRavishankar <ravishankarm@google.com>
Mon, 3 May 2021 19:50:29 +0000 (12:50 -0700)
committerMaheshRavishankar <ravishankarm@google.com>
Mon, 3 May 2021 19:51:24 +0000 (12:51 -0700)
commitfd15e2b825f26dd7eac3b4a52aab36c88e52850a
tree0226480c7c36d7e22e5ae301fd4cf82e1da5a57f
parent63f8226f259a03d7a56403be496724b77314e19a
[mlir][Linalg] Use rank-reduced versions of subtensor and subtensor insert when possible.

Convert subtensor and subtensor_insert operations to use their
rank-reduced versions to drop unit dimensions.

Differential Revision: https://reviews.llvm.org/D101495
mlir/include/mlir/Dialect/MemRef/Transforms/Passes.td
mlir/lib/Dialect/Linalg/Transforms/DropUnitDims.cpp
mlir/lib/Dialect/MemRef/Transforms/CMakeLists.txt
mlir/lib/Dialect/MemRef/Transforms/FoldSubViewOps.cpp
mlir/test/Dialect/Linalg/drop-unit-extent-dims.mlir
mlir/test/Dialect/MemRef/fold-subview-ops.mlir