From bb318c073e0442ad86aa499ed329058ebce9b4c7 Mon Sep 17 00:00:00 2001
From: Evan Cheng <evan.cheng@apple.com>
Date: Wed, 11 Jun 2008 19:07:54 +0000
Subject: Avoid duplicating loop header which leads to unnatural loops (and
 just seem like general badness to me, likely to cause code explosion). Patch
 by Florian Brandner.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@52223 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/Transforms/Scalar/TailDuplication.cpp | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'lib/Transforms')

diff --git a/lib/Transforms/Scalar/TailDuplication.cpp b/lib/Transforms/Scalar/TailDuplication.cpp
index d0998ab5a6..4e0949a2b4 100644
--- a/lib/Transforms/Scalar/TailDuplication.cpp
+++ b/lib/Transforms/Scalar/TailDuplication.cpp
@@ -33,6 +33,7 @@
 #include "llvm/Support/Debug.h"
 #include "llvm/ADT/Statistic.h"
 #include "llvm/ADT/SmallPtrSet.h"
+#include "llvm/Analysis/LoopInfo.h"
 #include <map>
 using namespace llvm;
 
@@ -50,10 +51,12 @@ namespace {
     static char ID; // Pass identification, replacement for typeid
     TailDup() : FunctionPass((intptr_t)&ID) {}
 
+    virtual void getAnalysisUsage(AnalysisUsage &AU) const;
   private:
     inline bool shouldEliminateUnconditionalBranch(TerminatorInst *, unsigned);
     inline void eliminateUnconditionalBranch(BranchInst *BI);
     SmallPtrSet<BasicBlock*, 4> CycleDetector;
+    LoopInfo *LI;  // The current loop information
   };
 }
 
@@ -69,6 +72,9 @@ FunctionPass *llvm::createTailDuplicationPass() { return new TailDup(); }
 /// a place it already pointed to earlier; see PR 2323.
 bool TailDup::runOnFunction(Function &F) {
   bool Changed = false;
+
+  LI = &getAnalysis<LoopInfo>();
+
   CycleDetector.clear();
   for (Function::iterator I = F.begin(), E = F.end(); I != E; ) {
     if (shouldEliminateUnconditionalBranch(I->getTerminator(),
@@ -83,6 +89,10 @@ bool TailDup::runOnFunction(Function &F) {
   return Changed;
 }
 
+void TailDup::getAnalysisUsage(AnalysisUsage &AU) const {
+  AU.addRequired<LoopInfo>();
+}
+
 /// shouldEliminateUnconditionalBranch - Return true if this branch looks
 /// attractive to eliminate.  We eliminate the branch if the destination basic
 /// block has <= 5 instructions in it, not counting PHI nodes.  In practice,
@@ -186,6 +196,14 @@ bool TailDup::shouldEliminateUnconditionalBranch(TerminatorInst *TI,
   if (!CycleDetector.insert(Dest))
     return false;
 
+  // Avoid non-natural loops:
+  // If a loop header is duplicated, the former natural loop will contain two
+  // paths into the loop --> the loop it not natural anymore. We want to avoid
+  // this, because other optimizaions may fail to improve the loop because of 
+  // this.
+  if (LI->isLoopHeader(Dest))
+    return false;
+
   return true;
 }
 
-- 
cgit v1.2.3-70-g09d2