Browse Source

[Attributor][FIX] Lookup of (call site) argument attributes

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@365977 91177308-0d34-0410-b5e6-96231b3b80d8
Johannes Doerfert 6 years ago
parent
commit
5968ac87ea
1 changed files with 5 additions and 2 deletions
  1. 5 2
      include/llvm/Transforms/IPO/Attributor.h

+ 5 - 2
include/llvm/Transforms/IPO/Attributor.h

@@ -179,9 +179,12 @@ struct Attributor {
     assert(AAType::ID != Attribute::None &&
            "Cannot lookup generic abstract attributes!");
 
-    // Determine the argument number automatically for llvm::Arguments.
+    // Determine the argument number automatically for llvm::Arguments if none
+    // is set. Do not override a given one as it could be a use of the argument
+    // in a call site.
     if (auto *Arg = dyn_cast<Argument>(&V))
-      ArgNo = Arg->getArgNo();
+      if (ArgNo == -1)
+        ArgNo = Arg->getArgNo();
 
     // If a function was given together with an argument number, perform the
     // lookup for the actual argument instead. Don't do it for variadic