diff options
| -rw-r--r-- | lib/Transforms/IPO/DeadArgumentElimination.cpp | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/lib/Transforms/IPO/DeadArgumentElimination.cpp b/lib/Transforms/IPO/DeadArgumentElimination.cpp index b877012..3a4c36f 100644 --- a/lib/Transforms/IPO/DeadArgumentElimination.cpp +++ b/lib/Transforms/IPO/DeadArgumentElimination.cpp @@ -590,15 +590,19 @@ void DAE::RemoveDeadArgumentsFromFunction(Function *F) { ParamAttrsVec.push_back(ParamAttrsWithIndex::get(Args.size(), Attrs)); } - // Reconstruct the ParamAttrsList based on the vector we constructed. - PAL = ParamAttrsList::get(ParamAttrsVec); - if (ExtraArgHack) Args.push_back(UndefValue::get(Type::Int32Ty)); - // Push any varargs arguments on the list - for (; AI != CS.arg_end(); ++AI) + // Push any varargs arguments on the list. Don't forget their attributes. + for (; AI != CS.arg_end(); ++AI) { Args.push_back(*AI); + uint16_t Attrs = PAL ? PAL->getParamAttrs(index++) : 0; + if (Attrs) + ParamAttrsVec.push_back(ParamAttrsWithIndex::get(Args.size(), Attrs)); + } + + // Reconstruct the ParamAttrsList based on the vector we constructed. + PAL = ParamAttrsList::get(ParamAttrsVec); Instruction *New; if (InvokeInst *II = dyn_cast<InvokeInst>(Call)) { |