[all-commits] [llvm/llvm-project] a490d3: [mlir][Vector] Add ExtractOp folding when fed by a...

[Nicolas Vasilache via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: a490d387e6e6085b35a6850581b62db3d2d47009
      https://github.com/llvm/llvm-project/commit/a490d387e6e6085b35a6850581b62db3d2d47009
  Author: Nicolas Vasilache <ntv at google.com>
  Date:   2020-07-10 (Fri, 10 Jul 2020)

  Changed paths:
    M mlir/include/mlir/IR/AffineMap.h
    M mlir/lib/Dialect/Vector/VectorOps.cpp
    M mlir/lib/IR/AffineMap.cpp
    M mlir/test/Dialect/Vector/canonicalize.mlir

  Log Message:
  -----------
  [mlir][Vector] Add ExtractOp folding when fed by a TransposeOp

TransposeOp are often followed by ExtractOp.
In certain cases however, it is unnecessary (and even detrimental) to lower a TransposeOp to either a flat transpose (llvm.matrix intrinsics) or to unrolled scalar insert / extract chains.

Providing foldings of ExtractOp mitigates some of the unnecessary complexity.

Differential revision: https://reviews.llvm.org/D83487