[llvm] [Attributor] Change allocation size and load/store offsets using AAPointerInfo for Alloca instructions and keep track of instructions causing an Access (PR #72029)

[Shilei Tian via llvm-commits]

================
@@ -13435,6 +13627,64 @@ struct AAAllocationInfoImpl : public AAAllocationInfo {
             A, this, IRP, DepClassTy::OPTIONAL, IsKnownNoCapture))
       return indicatePessimisticFixpoint();
 
+    const AAPrivatizablePtr *AAPrivatizablePtrI =
+        A.getOrCreateAAFor<AAPrivatizablePtr>(getIRPosition(), *this,
+                                              DepClassTy::OPTIONAL);
+
+    // If this allocation is privatizable we don't want to modify its allocation
+    // size.
+    // TODO: update AAPointerInfo to update bins once AAPrivitizable makes a
+    // change.
+    if (AAPrivatizablePtrI && (AAPrivatizablePtrI->isAssumedPrivatizablePtr() ||
+                               AAPrivatizablePtrI->isKnownPrivatizablePtr()))
+      return indicateOptimisticFixpoint();
+
+    // For all call sites, check if the called function can privatize the
+    // pointer.
+    for (Use &U : I->uses()) {
+      auto *CB = dyn_cast<CallBase>(U.getUser());
+      if (!CB)
+        continue;
+
+      unsigned ArgIdx = 0;
+      for (Use *It = CB->arg_begin(); It != CB->arg_end(); It++) {
+        Value *ArgVal = *It;
+
+        // Remove any pointer casts.
+        Value *Stripped = ArgVal->stripPointerCasts();
+        if (Stripped != I)
+          continue;
+
+        Function *Callee = CB->getCalledFunction();
+        if (!Callee)
+          continue;
+
+        if (ArgIdx >= Callee->arg_size())
+          continue;
+
+        Argument &FunctionDefArg = *Callee->getArg(ArgIdx);
+
+        IRPosition FunctionDefArgPos = IRPosition::argument(FunctionDefArg);
+
+        const AAPrivatizablePtr *AAPrivateArgPos =
+            A.getOrCreateAAFor<AAPrivatizablePtr>(FunctionDefArgPos, *this,
+                                                  DepClassTy::OPTIONAL);
+
+        if (!AAPrivateArgPos)
+          continue;
+
+        // If this allocation is privitizable we don't want to modify its
+        // allocation size.
+        // TODO: update AAPointerInfo to update bins once AAPrivitizable makes a
+        // change.
+        if (AAPrivateArgPos->isAssumedPrivatizablePtr() ||
----------------
shiltian wrote:

Here the same applies

https://github.com/llvm/llvm-project/pull/72029