[llvm-commits] [llvm] r50022 - /llvm/trunk/lib/Transforms/Scalar/JumpThreading.cpp

[Chris Lattner]

Author: lattner
Date: Sun Apr 20 21:57:57 2008
New Revision: 50022

URL: http://llvm.org/viewvc/llvm-project?rev=50022&view=rev
Log:
Use the new SplitBlockPredecessors to implement a todo.

Modified:
    llvm/trunk/lib/Transforms/Scalar/JumpThreading.cpp

Modified: llvm/trunk/lib/Transforms/Scalar/JumpThreading.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Transforms/Scalar/JumpThreading.cpp?rev=50022&r1=50021&r2=50022&view=diff

==============================================================================
--- llvm/trunk/lib/Transforms/Scalar/JumpThreading.cpp (original)
+++ llvm/trunk/lib/Transforms/Scalar/JumpThreading.cpp Sun Apr 20 21:57:57 2008
@@ -17,6 +17,7 @@
 #include "llvm/Pass.h"
 #include "llvm/ADT/DenseMap.h"
 #include "llvm/ADT/Statistic.h"
+#include "llvm/Transforms/Utils/BasicBlockUtils.h"
 #include "llvm/Transforms/Utils/Local.h"
 #include "llvm/Support/CommandLine.h"
 #include "llvm/Support/Compiler.h"
@@ -194,10 +195,21 @@
     SuccBB = SI->getSuccessor(SI->findCaseValue(PredCst));
   }
   
-  // TODO: If there are multiple preds with the same incoming value for the PHI,
-  // factor them together so we get one thread block for the whole group.  This
-  // is important for things like "phi i1 [true, true, false, true, x]" where
-  // we only need to clone the block for the true blocks once.
+  // If there are multiple preds with the same incoming value for the PHI,
+  // factor them together so we get one block to thread for the whole group.
+  // This is important for things like "phi i1 [true, true, false, true, x]"
+  // where we only need to clone the block for the true blocks once.
+  SmallVector<BasicBlock*, 16> CommonPreds;
+  for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i)
+    if (PN->getIncomingValue(i) == PredCst)
+      CommonPreds.push_back(PN->getIncomingBlock(i));
+  if (CommonPreds.size() != 1) {
+    DOUT << "  Factoring out " << CommonPreds.size()
+         << " common predecessors.\n";
+    PredBB = SplitBlockPredecessors(BB, &CommonPreds[0], CommonPreds.size(),
+                                    ".thr_comm", this);
+  }
+  
   
   DOUT << "  Threading edge from '" << PredBB->getNameStart() << "' to '"
        << SuccBB->getNameStart() << "' with cost: " << JumpThreadCost