[all-commits] [llvm/llvm-project] f0c93f: [mlir][vector] Merge accumulator/result transpose ...

Lei Zhang via All-commits all-commits at lists.llvm.org
Fri Oct 7 17:47:41 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f0c93fd4cac76d9ae1b2ff99c40c08abeb940f21
      https://github.com/llvm/llvm-project/commit/f0c93fd4cac76d9ae1b2ff99c40c08abeb940f21
  Author: Lei Zhang <antiagainst at google.com>
  Date:   2022-10-08 (Sat, 08 Oct 2022)

  Changed paths:
    M mlir/lib/Dialect/Vector/Transforms/VectorTransforms.cpp
    M mlir/test/Dialect/Vector/vector-reduce-to-contract.mlir

  Log Message:
  -----------
  [mlir][vector] Merge accumulator/result transpose into contract

This commit adds a pattern to merge accumulator and result
`vector.transpose` ops into `vector.contract`. This kind of
pattern can be generated for NCHW convolution vectorization,
where we use transposes to convert the 1-D NCW convolution
into NWC during vectorization. Merging the transpose would
mean we can avoid materialize vector extract/insert for
transposes and it makes further vector level transformations
easier.

Reviewed By: ThomasRaoux

Differential Revision: https://reviews.llvm.org/D135496




More information about the All-commits mailing list