aboutsummaryrefslogtreecommitdiff
path: root/lib/Transforms/Utils/CloneTrace.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'lib/Transforms/Utils/CloneTrace.cpp')
-rw-r--r--lib/Transforms/Utils/CloneTrace.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/lib/Transforms/Utils/CloneTrace.cpp b/lib/Transforms/Utils/CloneTrace.cpp
index 07111393e2..1cedfd872c 100644
--- a/lib/Transforms/Utils/CloneTrace.cpp
+++ b/lib/Transforms/Utils/CloneTrace.cpp
@@ -68,6 +68,11 @@ llvm::CloneTrace(const std::vector<BasicBlock*> &origTrace) {
//Second loop to do the remapping
for (std::vector<BasicBlock *>::const_iterator BB = clonedTrace.begin(),
BE = clonedTrace.end(); BB != BE; ++BB) {
+
+ //Remap the unwind_to label
+ if (BasicBlock *UnwindDest = (*BB)->getUnwindDest())
+ (*BB)->setUnwindDest(cast<BasicBlock>(ValueMap[UnwindDest]));
+
for (BasicBlock::iterator I = (*BB)->begin(); I != (*BB)->end(); ++I) {
//Loop over all the operands of the instruction
for (unsigned op=0, E = I->getNumOperands(); op != E; ++op) {