diff options
| author | MLIR Team <no-reply@google.com> | 2019-04-04 15:19:17 -0700 |
|---|---|---|
| committer | Mehdi Amini <joker.eph@gmail.com> | 2019-04-05 07:42:01 -0700 |
| commit | 0cd589c337eae12454452356f96752b229b373f9 (patch) | |
| tree | 2de78d12c69c245ba6b223be24ffbb1917a58a9d /mlir/lib/Transforms/LoopTiling.cpp | |
| parent | a8f4b9eeeb91044afc00229cea34b91dffee9f28 (diff) | |
| download | bcm5719-llvm-0cd589c337eae12454452356f96752b229b373f9.tar.gz bcm5719-llvm-0cd589c337eae12454452356f96752b229b373f9.zip | |
Create a LoopUtil function to return perfectly nested loop set
--
PiperOrigin-RevId: 242019230
Diffstat (limited to 'mlir/lib/Transforms/LoopTiling.cpp')
| -rw-r--r-- | mlir/lib/Transforms/LoopTiling.cpp | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/mlir/lib/Transforms/LoopTiling.cpp b/mlir/lib/Transforms/LoopTiling.cpp index 956d50ec26f..c215fa34172 100644 --- a/mlir/lib/Transforms/LoopTiling.cpp +++ b/mlir/lib/Transforms/LoopTiling.cpp @@ -270,11 +270,7 @@ static void getTileableBands(Function &f, // (inclusive). auto getMaximalPerfectLoopNest = [&](AffineForOp root) { SmallVector<AffineForOp, 6> band; - AffineForOp currInst = root; - do { - band.push_back(currInst); - } while (currInst.getBody()->getOperations().size() == 2 && - (currInst = currInst.getBody()->front().dyn_cast<AffineForOp>())); + getPerfectlyNestedLoops(band, root); bands->push_back(band); }; |
