When loop extraction succeeds, make sure to map the function pointers over

to avoid dangling references.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@12470 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 11 insertions, 0 deletions

diff --git a/tools/bugpoint/Miscompilation.cpp b/tools/bugpoint/Miscompilation.cpp
index 21f4bcc..6524f7a 100644
--- a/tools/bugpoint/Miscompilation.cpp
+++ b/tools/bugpoint/Miscompilation.cpp
@@ -268,6 +268,17 @@ static bool ExtractLoops(BugDriver &BD,
       exit(1);
     }
     delete ToOptimizeLoopExtracted;
+
+    // All of the Function*'s in the MiscompiledFunctions list are in the old
+    // module.  Make sure to update them to point to the corresponding functions
+    // in the new module.
+    for (unsigned i = 0, e = MiscompiledFunctions.size(); i != e; ++i) {
+      Function *OldF = MiscompiledFunctions[i];
+      Function *NewF = 
+        ToNotOptimize->getFunction(OldF->getName(), OldF->getFunctionType());
+      MiscompiledFunctions[i] = NewF;
+    }
+
     BD.setNewProgram(ToNotOptimize);
     MadeChange = true;
   }