Index: llvm/trunk/include/llvm/IR/IRBuilder.h =================================================================== --- llvm/trunk/include/llvm/IR/IRBuilder.h +++ llvm/trunk/include/llvm/IR/IRBuilder.h @@ -1569,15 +1569,18 @@ } Value *CreateSelect(Value *C, Value *True, Value *False, - const Twine &Name = "", MDNode *ProfWeights = nullptr) { + const Twine &Name = "", Instruction *MDFrom = nullptr) { if (Constant *CC = dyn_cast(C)) if (Constant *TC = dyn_cast(True)) if (Constant *FC = dyn_cast(False)) return Insert(Folder.CreateSelect(CC, TC, FC), Name); SelectInst *Sel = SelectInst::Create(C, True, False); - // TODO: "unpredictable" metadata can apply to a select too. - Sel->setMetadata(LLVMContext::MD_prof, ProfWeights); + if (MDFrom) { + MDNode *Prof = MDFrom->getMetadata(LLVMContext::MD_prof); + MDNode *Unpred = MDFrom->getMetadata(LLVMContext::MD_unpredictable); + Sel = addBranchMetadata(Sel, Prof, Unpred); + } return Insert(Sel, Name); } Index: llvm/trunk/lib/Transforms/Utils/SimplifyCFG.cpp =================================================================== --- llvm/trunk/lib/Transforms/Utils/SimplifyCFG.cpp +++ llvm/trunk/lib/Transforms/Utils/SimplifyCFG.cpp @@ -1935,10 +1935,9 @@ Value *TrueVal = PN->getIncomingValue(PN->getIncomingBlock(0) == IfFalse); Value *FalseVal = PN->getIncomingValue(PN->getIncomingBlock(0) == IfTrue); - MDNode *MDN = InsertPt->getMetadata(LLVMContext::MD_prof); - Value *Select = Builder.CreateSelect(IfCond, TrueVal, FalseVal, "", MDN); - PN->replaceAllUsesWith(Select); - Select->takeName(PN); + Value *Sel = Builder.CreateSelect(IfCond, TrueVal, FalseVal, "", InsertPt); + PN->replaceAllUsesWith(Sel); + Sel->takeName(PN); PN->eraseFromParent(); } Index: llvm/trunk/test/Transforms/SimplifyCFG/PhiEliminate2.ll =================================================================== --- llvm/trunk/test/Transforms/SimplifyCFG/PhiEliminate2.ll +++ llvm/trunk/test/Transforms/SimplifyCFG/PhiEliminate2.ll @@ -5,7 +5,7 @@ define i32 @FoldTwoEntryPHINode(i1 %C, i32 %V1, i32 %V2, i16 %V3) { entry: - br i1 %C, label %then, label %else, !prof !0 + br i1 %C, label %then, label %else, !prof !0, !unpredictable !1 then: %V4 = or i32 %V2, %V1 br label %Cont @@ -21,12 +21,14 @@ ; CHECK-NEXT: entry: ; CHECK-NEXT: %V5 = sext i16 %V3 to i32 ; CHECK-NEXT: %V4 = or i32 %V2, %V1 -; CHECK-NEXT: %V6 = select i1 %C, i32 %V4, i32 %V5, !prof !0 +; CHECK-NEXT: %V6 = select i1 %C, i32 %V4, i32 %V5, !prof !0, !unpredictable !1 ; CHECK-NEXT: %0 = call i32 @FoldTwoEntryPHINode(i1 false, i32 0, i32 0, i16 0) ; CHECK-NEXT: ret i32 %V1 } !0 = !{!"branch_weights", i32 3, i32 5} +!1 = !{} ; CHECK: !0 = !{!"branch_weights", i32 3, i32 5} +; CHECK: !1 = !{}