summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/Transforms/IPO/ArgumentPromotion.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/lib/Transforms/IPO/ArgumentPromotion.cpp b/lib/Transforms/IPO/ArgumentPromotion.cpp
index 3ac1c7bb2b..4486677e63 100644
--- a/lib/Transforms/IPO/ArgumentPromotion.cpp
+++ b/lib/Transforms/IPO/ArgumentPromotion.cpp
@@ -404,6 +404,10 @@ Function *ArgPromotion::DoPromotion(Function *F,
ParamAttrsVector ParamAttrsVec;
const ParamAttrsList *PAL = F->getParamAttrs();
+ // Add any return attributes.
+ if (unsigned attrs = PAL ? PAL->getParamAttrs(0) : 0)
+ ParamAttrsVec.push_back(ParamAttrsWithIndex::get(0, attrs));
+
unsigned ArgIndex = 1;
for (Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E;
++I, ++ArgIndex) {
@@ -491,6 +495,10 @@ Function *ArgPromotion::DoPromotion(Function *F,
Instruction *Call = CS.getInstruction();
PAL = CS.getParamAttrs();
+ // Add any return attributes.
+ if (unsigned attrs = PAL ? PAL->getParamAttrs(0) : 0)
+ ParamAttrsVec.push_back(ParamAttrsWithIndex::get(0, attrs));
+
// Loop over the operands, inserting GEP and loads in the caller as
// appropriate.
CallSite::arg_iterator AI = CS.arg_begin();