[mlir][Linalg] Disable fusion of tensor_reshape op by expansion when unit-dims are...
authorMaheshRavishankar <ravishankarm@google.com>
Fri, 22 Jan 2021 20:53:27 +0000 (12:53 -0800)
committerMaheshRavishankar <ravishankarm@google.com>
Fri, 22 Jan 2021 20:55:25 +0000 (12:55 -0800)
commit430d43e010bdd07d73c4d0d6536206d22d35a2cb
tree9248a2682c429b3b4d22a270a9897e1a9d61f2f5
parent20f2e32d2c545e6e23dc5c69c42caac7a4bca0fc
[mlir][Linalg] Disable fusion of tensor_reshape op by expansion when unit-dims are involved

Fusion of generic/indexed_generic operations with tensor_reshape by
expansion when the latter just adds/removes unit-dimensions is
disabled since it just adds unit-trip count loops.

Differential Revision: https://reviews.llvm.org/D94626
mlir/include/mlir/Dialect/Linalg/Passes.h
mlir/lib/Dialect/Linalg/Transforms/DropUnitDims.cpp
mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
mlir/test/Dialect/Linalg/drop-unit-extent-dims.mlir
mlir/test/Dialect/Linalg/reshape_fusion.mlir