[mlir] Handle an edge case when folding reshapes with multiple trailing 1 dimensions
authorBenjamin Kramer <benny.kra@googlemail.com>
Mon, 29 Nov 2021 15:52:37 +0000 (16:52 +0100)
committerBenjamin Kramer <benny.kra@googlemail.com>
Mon, 29 Nov 2021 17:31:43 +0000 (18:31 +0100)
commit8d474f1d157530577f06ce3ef9187e1aaf31a59e
treec26108788e938660b479bc24ad37b82cdb4abe97
parentf240e528cea25fd2a9ae01b1e1fe77f507ed7a2c
[mlir] Handle an edge case when folding reshapes with multiple trailing 1 dimensions

We would exit early and miss this case.

Differential Revision: https://reviews.llvm.org/D114711
mlir/lib/Dialect/Utils/ReshapeOpsUtils.cpp
mlir/test/Dialect/Linalg/canonicalize.mlir