diff options
| author | Nicolas Vasilache <ntv@google.com> | 2019-01-28 18:28:43 -0800 |
|---|---|---|
| committer | jpienaar <jpienaar@google.com> | 2019-03-29 15:42:38 -0700 |
| commit | ae772b79659afff6695170b7d404113c32e35a0d (patch) | |
| tree | 4e2a7254f58f016816fc0bd3f1812e39cc872e17 /mlir/lib/Transforms/LoopUnroll.cpp | |
| parent | ad637f3ccee4be5aa9f738d6665d0b1a326613a7 (diff) | |
| download | bcm5719-llvm-ae772b79659afff6695170b7d404113c32e35a0d.tar.gz bcm5719-llvm-ae772b79659afff6695170b7d404113c32e35a0d.zip | |
Automated rollback of changelist 231318632.
PiperOrigin-RevId: 231327161
Diffstat (limited to 'mlir/lib/Transforms/LoopUnroll.cpp')
| -rw-r--r-- | mlir/lib/Transforms/LoopUnroll.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/mlir/lib/Transforms/LoopUnroll.cpp b/mlir/lib/Transforms/LoopUnroll.cpp index 6d63e4afd2d..39ef758833b 100644 --- a/mlir/lib/Transforms/LoopUnroll.cpp +++ b/mlir/lib/Transforms/LoopUnroll.cpp @@ -119,6 +119,15 @@ PassResult LoopUnroll::runOnFunction(Function *f) { return true; } + bool walkIfInstPostOrder(IfInst *ifInst) { + bool hasInnerLoops = + walkPostOrder(ifInst->getThen()->begin(), ifInst->getThen()->end()); + if (ifInst->hasElse()) + hasInnerLoops |= + walkPostOrder(ifInst->getElse()->begin(), ifInst->getElse()->end()); + return hasInnerLoops; + } + bool walkOpInstPostOrder(OperationInst *opInst) { for (auto &blockList : opInst->getBlockLists()) for (auto &block : blockList) |
