[polly] r333089 - [Acc] Re-land r326643 to finally fix PR33208.

[Philip Pfaffe via llvm-commits]

Author: pfaffe
Date: Wed May 23 07:52:35 2018
New Revision: 333089

URL: http://llvm.org/viewvc/llvm-project?rev=333089&view=rev
Log:
[Acc] Re-land r326643 to finally fix PR33208.

Other than before, don't clear out LI entirely but only those relevant
loops.

Modified:
    polly/trunk/lib/CodeGen/PPCGCodeGeneration.cpp

Modified: polly/trunk/lib/CodeGen/PPCGCodeGeneration.cpp
URL: http://llvm.org/viewvc/llvm-project/polly/trunk/lib/CodeGen/PPCGCodeGeneration.cpp?rev=333089&r1=333088&r2=333089&view=diff
==============================================================================
--- polly/trunk/lib/CodeGen/PPCGCodeGeneration.cpp (original)
+++ polly/trunk/lib/CodeGen/PPCGCodeGeneration.cpp Wed May 23 07:52:35 2018
@@ -1563,12 +1563,14 @@ void GPUNodeBuilder::clearScalarEvolutio
 }
 
 void GPUNodeBuilder::clearLoops(Function *F) {
+  SmallSet<Loop *, 1> WorkList;
   for (BasicBlock &BB : *F) {
     Loop *L = LI.getLoopFor(&BB);
     if (L)
-      SE.forgetLoop(L);
-    LI.removeBlock(&BB);
+      WorkList.insert(L);
   }
+  for (auto *L : WorkList)
+    LI.erase(L);
 }
 
 std::tuple<Value *, Value *> GPUNodeBuilder::getGridSizes(ppcg_kernel *Kernel) {