[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