Replace custom code with canFoldInAddressingMode

Author: ritter-x2a

llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp

@@ -2779,23 +2779,9 @@ SDValue DAGCombiner::visitPTRADD(SDNode *N) {
   // This should be done late, after legalization, so that it doesn't block
   // other ptradd combines that could enable more offset folding.
   if (LegalOperations && DAG.haveNoCommonBitsSet(N0, N1)) {
-    bool TransformCanBreakAddrMode = false;
-    if (auto *C = dyn_cast<ConstantSDNode>(N1)) {
-      TargetLoweringBase::AddrMode AM;
-      AM.HasBaseReg = true;
-      AM.BaseOffs = C->getSExtValue();
-      TransformCanBreakAddrMode = any_of(N->users(), [&](SDNode *User) {
-        if (auto *LoadStore = dyn_cast<MemSDNode>(User);
-            LoadStore && LoadStore->getBasePtr().getNode() == N) {
-          unsigned AS = LoadStore->getAddressSpace();
-          EVT AccessVT = LoadStore->getMemoryVT();
-          Type *AccessTy = AccessVT.getTypeForEVT(*DAG.getContext());
-          return TLI.isLegalAddressingMode(DAG.getDataLayout(), AM, AccessTy,
-                                           AS);
-        }
-        return false;
-      });
-    }
+    bool TransformCanBreakAddrMode = any_of(N->users(), [&](SDNode *User) {
+      return canFoldInAddressingMode(N, User, DAG, TLI);
+    });
 
     if (!TransformCanBreakAddrMode)
       return DAG.getNode(ISD::OR, DL, PtrVT, N0, N1, SDNodeFlags::Disjoint);