diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyLowerEmscriptenEHSjLj.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyLowerEmscriptenEHSjLj.cpp --- a/llvm/lib/Target/WebAssembly/WebAssemblyLowerEmscriptenEHSjLj.cpp +++ b/llvm/lib/Target/WebAssembly/WebAssemblyLowerEmscriptenEHSjLj.cpp @@ -200,6 +200,7 @@ #include "llvm/Support/CommandLine.h" #include "llvm/Transforms/Utils/BasicBlockUtils.h" #include "llvm/Transforms/Utils/SSAUpdater.h" +#include "llvm/Transforms/Utils/SSAUpdaterBulk.h" using namespace llvm; @@ -641,25 +642,23 @@ void WebAssemblyLowerEmscriptenEHSjLj::rebuildSSA(Function &F) { DominatorTree &DT = getAnalysis(F).getDomTree(); DT.recalculate(F); // CFG has been changed - SSAUpdater SSA; + SSAUpdaterBulk SSA; for (BasicBlock &BB : F) { for (Instruction &I : BB) { - SSA.Initialize(I.getType(), I.getName()); - SSA.AddAvailableValue(&BB, &I); - for (auto UI = I.use_begin(), UE = I.use_end(); UI != UE;) { - Use &U = *UI; - ++UI; + unsigned VarID = SSA.AddVariable(I.getName(), I.getType()); + SSA.AddAvailableValue(VarID, &BB, &I); + for (auto &U : I.uses()) { auto *User = cast(U.getUser()); if (auto *UserPN = dyn_cast(User)) if (UserPN->getIncomingBlock(U) == &BB) continue; - if (DT.dominates(&I, User)) continue; - SSA.RewriteUseAfterInsertions(U); + SSA.AddUse(VarID, &U); } } } + SSA.RewriteAllUses(&DT); } // Replace uses of longjmp with emscripten_longjmp. emscripten_longjmp takes @@ -1301,24 +1300,14 @@ for (Instruction *I : SetjmpTableSizeInsts) SetjmpTableSizeSSA.AddAvailableValue(I->getParent(), I); - for (auto UI = SetjmpTable->use_begin(), UE = SetjmpTable->use_end(); - UI != UE;) { - // Grab the use before incrementing the iterator. - Use &U = *UI; - // Increment the iterator before removing the use from the list. - ++UI; + for (auto &U : make_early_inc_range(SetjmpTable->uses())) if (auto *I = dyn_cast(U.getUser())) if (I->getParent() != Entry) SetjmpTableSSA.RewriteUse(U); - } - for (auto UI = SetjmpTableSize->use_begin(), UE = SetjmpTableSize->use_end(); - UI != UE;) { - Use &U = *UI; - ++UI; + for (auto &U : make_early_inc_range(SetjmpTableSize->uses())) if (auto *I = dyn_cast(U.getUser())) if (I->getParent() != Entry) SetjmpTableSizeSSA.RewriteUse(U); - } // Finally, our modifications to the cfg can break dominance of SSA variables. // For example, in this code, diff --git a/llvm/test/CodeGen/WebAssembly/lower-em-sjlj.ll b/llvm/test/CodeGen/WebAssembly/lower-em-sjlj.ll --- a/llvm/test/CodeGen/WebAssembly/lower-em-sjlj.ll +++ b/llvm/test/CodeGen/WebAssembly/lower-em-sjlj.ll @@ -142,7 +142,7 @@ ; CHECK-NEXT: %[[SETJMP_TABLE_SIZE1:.*]] = call i32 @getTempRet0() ; CHECK: if.then.split: -; CHECK: %[[VAR1:.*]] = phi i32 [ %[[VAR0]], %if.then ], [ %[[VAR2:.*]], %if.end3 ] +; CHECK: %[[VAR1:.*]] = phi i32 [ %[[VAR2:.*]], %if.end3 ], [ %[[VAR0]], %if.then ] ; CHECK: %[[SETJMP_TABLE_SIZE2:.*]] = phi i32 [ %[[SETJMP_TABLE_SIZE1]], %if.then ], [ %[[SETJMP_TABLE_SIZE3:.*]], %if.end3 ] ; CHECK: %[[SETJMP_TABLE2:.*]] = phi i32* [ %[[SETJMP_TABLE1]], %if.then ], [ %[[SETJMP_TABLE3:.*]], %if.end3 ] ; CHECK: store i32 %[[VAR1]], i32* @global_var, align 4 @@ -222,7 +222,7 @@ br label %for.cond ; CHECK: for.inc.split: - ; CHECK: %var[[VARNO]] = phi i32 [ %var, %for.inc ] + ; CHECK: %var[[VARNO]] = phi i32 [ undef, %if.end ], [ %var, %for.inc ] } ; Tests cases where longjmp function pointer is used in other ways than direct