[llvm] [InstCombine] ptrmask of gep for dynamic pointer aligment (PR #80002)

[Jon Chesterfield via llvm-commits]

================
@@ -966,6 +966,41 @@ Value *InstCombinerImpl::SimplifyDemandedUseBits(Value *V, APInt DemandedMask,
                 I, 1, (DemandedMask & ~LHSKnown.Zero).zextOrTrunc(MaskWidth)))
           return I;
 
+        // Combine:
+        // (ptrmask (getelementptr i8, ptr p, imm i), imm mask)
+        //   -> (ptrmask (getelementptr i8, ptr p, imm (i & mask)), imm mask)
+        // where only the low bits known to be zero in the pointer are changed
+        Value *InnerPtr;
+        uint64_t GEPIndex;
+        uint64_t PtrMaskImmediate;
+        if (match(I, m_Intrinsic<Intrinsic::ptrmask>(
+                         m_GEP(m_Value(InnerPtr), m_ConstantInt(GEPIndex)),
+                         m_ConstantInt(PtrMaskImmediate)))) {
+          auto *GEP = cast<GetElementPtrInst>(II->getArgOperand(0));
+
+          if (GEP->getSourceElementType() == Type::getInt8Ty(I->getContext())) {
+            unsigned Log2Align = llvm::Log2(InnerPtr->getPointerAlignment(DL));
+            uint64_t PointerAlignBits = (uint64_t(1) << Log2Align) - 1;
+
+            uint64_t HighBitsGEPIndex = GEPIndex & ~PointerAlignBits;
+            uint64_t MaskedLowBitsGEPIndex =
+                GEPIndex & PointerAlignBits & PtrMaskImmediate;
+
+            uint64_t MaskedGEPIndex = HighBitsGEPIndex | MaskedLowBitsGEPIndex;
+
+            if (MaskedGEPIndex != GEPIndex) {
+              Type *I64 = Type::getInt64Ty(I->getContext());
+              auto MaskedGEP =
----------------
JonChesterfield wrote:

Nice catch on builder, yes

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