[llvm] [InferAlignment] Increase alignment in masked load / store instrinsics if known (PR #156057)

[Nikita Popov via llvm-commits]

================
@@ -35,8 +37,39 @@ static bool tryToImproveAlign(
       return true;
     }
   }
-  // TODO: Also handle memory intrinsics.
-  return false;
+
+  IntrinsicInst *II = dyn_cast<IntrinsicInst>(I);
+  if (!II)
+    return false;
+
+  // TODO: Handle more memory intrinsics.
+  switch (II->getIntrinsicID()) {
+  case Intrinsic::masked_load:
+  case Intrinsic::masked_store: {
+    Value *PtrOp = II->getIntrinsicID() == Intrinsic::masked_load
+                       ? II->getArgOperand(0)
+                       : II->getArgOperand(1);
+    Value *AlignOp = II->getIntrinsicID() == Intrinsic::masked_load
+                         ? II->getArgOperand(1)
+                         : II->getArgOperand(2);
+
+    Align OldAlign = cast<ConstantInt>(AlignOp)->getAlignValue();
+    Align PrefAlign = getKnownAlignment(PtrOp, DL, II);
+    Align NewAlign = Fn(PtrOp, OldAlign, PrefAlign);
+    if (NewAlign <= OldAlign)
+      return false;
+
+    Value *V = llvm::ConstantInt::get(llvm::Type::getInt32Ty(II->getContext()),
----------------
nikic wrote:

```suggestion
    Value *V = ConstantInt::get(llvm::Type::getInt32Ty(II->getContext()),
```

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