Index: lib/IR/Instruction.cpp =================================================================== --- lib/IR/Instruction.cpp +++ lib/IR/Instruction.cpp @@ -625,20 +625,29 @@ return; auto *ProfDataName = dyn_cast(ProfileData->getOperand(0)); - if (!ProfDataName || !ProfDataName->getString().equals("branch_weights")) + if (!ProfDataName || (!ProfDataName->getString().equals("branch_weights") && + !ProfDataName->getString().equals("VP"))) return; - SmallVector Weights; - for (unsigned i = 1; i < ProfileData->getNumOperands(); i++) { - // Using APInt::div may be expensive, but most cases should fit in 64 bits. - APInt Val(128, mdconst::dyn_extract(ProfileData->getOperand(i)) - ->getValue() - .getZExtValue()); - Val *= APInt(128, S); - Weights.push_back(Val.udiv(APInt(128, T)).getLimitedValue()); - } MDBuilder MDB(getContext()); - setMetadata(LLVMContext::MD_prof, MDB.createBranchWeights(Weights)); + SmallVector Vals; + Vals.push_back(ProfileData->getOperand(0)); + + for (unsigned i = 1; i < ProfileData->getNumOperands(); i++) + if ((i % 2) == 1 && ProfDataName->getString().equals("VP")) { + Vals.push_back(ProfileData->getOperand(i)); + } else { + // Using APInt::div may be expensive, but most cases should fit 64 bits. + APInt Val(128, + mdconst::dyn_extract(ProfileData->getOperand(i)) + ->getValue() + .getZExtValue()); + Val *= APInt(128, S); + Vals.push_back(MDB.createConstant( + ConstantInt::get(Type::getInt64Ty(getContext()), + Val.udiv(APInt(128, T)).getLimitedValue()))); + } + setMetadata(LLVMContext::MD_prof, MDNode::get(getContext(), Vals)); } void Instruction::setProfWeight(uint64_t W) { Index: test/Transforms/Inline/prof-update.ll =================================================================== --- test/Transforms/Inline/prof-update.ll +++ test/Transforms/Inline/prof-update.ll @@ -3,6 +3,7 @@ declare void @ext(); declare void @ext1(); +@func = global void ()* null ; CHECK: define void @callee(i32 %n) !prof ![[ENTRY_COUNT:[0-9]*]] define void @callee(i32 %n) !prof !1 { @@ -17,12 +18,16 @@ ; ext is cloned and updated. ; CHECK: call void @ext(), !prof ![[COUNT_CALLEE:[0-9]*]] call void @ext(), !prof !2 + %f = load void ()*, void ()** @func +; CHECK: call void %f(), !prof ![[COUNT_IND_CALLEE:[0-9]*]] + call void %f(), !prof !4 ret void } ; CHECK: define void @caller() define void @caller() { ; CHECK: call void @ext(), !prof ![[COUNT_CALLER:[0-9]*]] +; CHECK: call void %f.i(), !prof ![[COUNT_IND_CALLER:[0-9]*]] call void @callee(i32 15), !prof !3 ret void } @@ -32,8 +37,11 @@ !1 = !{!"function_entry_count", i64 1000} !2 = !{!"branch_weights", i64 2000} !3 = !{!"branch_weights", i64 400} +!4 = !{!"VP", i32 0, i64 140, i64 111, i64 80, i64 222, i64 40, i64 333, i64 20} attributes #0 = { alwaysinline } ; CHECK: ![[ENTRY_COUNT]] = !{!"function_entry_count", i64 600} ; CHECK: ![[COUNT_CALLEE1]] = !{!"branch_weights", i64 2000} -; CHECK: ![[COUNT_CALLEE]] = !{!"branch_weights", i32 1200} -; CHECK: ![[COUNT_CALLER]] = !{!"branch_weights", i32 800} +; CHECK: ![[COUNT_CALLEE]] = !{!"branch_weights", i64 1200} +; CHECK: ![[COUNT_IND_CALLEE]] = !{!"VP", i32 0, i64 84, i64 111, i64 48, i64 222, i64 24, i64 333, i64 12} +; CHECK: ![[COUNT_CALLER]] = !{!"branch_weights", i64 800} +; CHECK: ![[COUNT_IND_CALLER]] = !{!"VP", i32 0, i64 56, i64 111, i64 32, i64 222, i64 16, i64 333, i64 8}