// Because we added the pointer to the callee as first argument, all
// argument attribute indices have to be incremented by one.
- SmallVector<AttributeSetNode *, 8> AttributesVec;
- const AttributeList &InvokeAL = CI->getAttributes();
-
- // Add any return attributes.
- AttributesVec.push_back(InvokeAL.getRetAttributes());
- // No attributes for the callee pointer.
- AttributesVec.push_back(nullptr);
- // Copy the argument attributes from the original
- for (unsigned i = 1, e = CI->getNumArgOperands(); i <= e; ++i) {
- AttributesVec.push_back(InvokeAL.getParamAttributes(i));
+ SmallVector<AttributeList, 8> AttributesVec;
+ const AttributeList &InvokePAL = CI->getAttributes();
+ CallSite::arg_iterator AI = CI->arg_begin();
+ unsigned i = 1; // Argument attribute index starts from 1
+ for (unsigned e = CI->getNumArgOperands(); i <= e; ++AI, ++i) {
+ if (InvokePAL.hasAttributes(i)) {
+ AttrBuilder B(InvokePAL, i);
+ AttributesVec.push_back(AttributeList::get(C, i + 1, B));
+ }
}
-
+ // Add any return attributes.
+ if (InvokePAL.hasAttributes(AttributeList::ReturnIndex))
+ AttributesVec.push_back(
+ AttributeList::get(C, InvokePAL.getRetAttributes()));
// Add any function attributes.
- AttributesVec.push_back(InvokeAL.getFnAttributes());
+ if (InvokePAL.hasAttributes(AttributeList::FunctionIndex))
+ AttributesVec.push_back(AttributeList::get(C, InvokePAL.getFnAttributes()));
// Reconstruct the AttributesList based on the vector we constructed.
- AttributeList NewCallAL = AttributeList::get(C, AttributesVec);
- NewCall->setAttributes(NewCallAL);
+ AttributeList NewCallPAL = AttributeList::get(C, AttributesVec);
+ NewCall->setAttributes(NewCallPAL);
CI->replaceAllUsesWith(NewCall);