diff options
| author | Matthias Springer <springerm@google.com> | 2022-12-14 14:02:13 +0100 |
|---|---|---|
| committer | Matthias Springer <springerm@google.com> | 2022-12-14 14:10:04 +0100 |
| commit | e07149c91f4bb43c1413bc1fbe19dc6eff2fcde6 (patch) | |
| tree | f10ddb556a37ccfa2c5052b41da70a7a9ba0dd82 /llvm/lib/FileCheck/FileCheck.cpp | |
| parent | befd167050ce46517a799cf0f997e2f75c20eb3b (diff) | |
| download | llvm-e07149c91f4bb43c1413bc1fbe19dc6eff2fcde6.zip llvm-e07149c91f4bb43c1413bc1fbe19dc6eff2fcde6.tar.gz llvm-e07149c91f4bb43c1413bc1fbe19dc6eff2fcde6.tar.bz2 | |
[mlir][linalg] Add option to generate rank-reducing slices in DropUnitDims
This change extends the `ReplaceUnitExtents` pattern so that users can choose between of two strategies for generating rank reductions:
* CollapseShapeOp / ExpandShapeOp (was already implemented but code was cleaned up; default strategy)
* rank-reducing ExtractSliceOp / InsertSliceOp
Also add helper functions to the memref dialect that we already have on the tensor dialect: `getMixedSizes`, `createCanonicalRankReducingSubViewOp`, `rankReduceIfNeeded`.
We are using ReassociationIndices instead of ReassoicationExprs in many other places and this makes the code easier to read. Also adding a new test case (that also passed before).
Differential Revision: https://reviews.llvm.org/D139947
Diffstat (limited to 'llvm/lib/FileCheck/FileCheck.cpp')
0 files changed, 0 insertions, 0 deletions