[mlir][vector] Allow transposing multi_reduction when the parallel dim is in the...
authorBenjamin Kramer <benny.kra@googlemail.com>
Tue, 24 Jan 2023 17:01:26 +0000 (18:01 +0100)
committerBenjamin Kramer <benny.kra@googlemail.com>
Thu, 26 Jan 2023 17:06:42 +0000 (18:06 +0100)
commit77f4c91c5e052f43245555fffbb0649191e3e32f
treeedd28107931a24f54ef80c1dce3888da4bfc2bdf
parent76790cf1db53c35384a6c8c7400c03e5f028f9bc
[mlir][vector] Allow transposing multi_reduction when the parallel dim is in the middle

The check for the outer lowering wasn't quite right.

Differential Revision: https://reviews.llvm.org/D142483
mlir/lib/Dialect/Vector/Transforms/VectorMultiDimReductionTransforms.cpp
mlir/test/Dialect/Vector/vector-multi-reduction-lowering.mlir
mlir/test/Dialect/Vector/vector-multi-reduction-outer-lowering.mlir