[SelectionDAG] Combine range metadata when loads are CSEd.

When CSEing a load with an existing load with different range
metadata, clear the range metadata on the existing
load.

This is conservative, alternatively we could calculate new range
metadata using getMostGenericRange. Without a test case I wasn't
sure it was worth it.

getMostGenericRange takes a non-const MDNode*, but all of SelectionDAG
uses const MDNode*. A const_cast will need to be used somewhere or
we need to make the code base consistent about whether MDNode pointers
should be const or not.

I'm sure this isn't the only place that needs to be updated to handle
the CSE.

Fixes #145363.

Author: topperc

llvm/include/llvm/CodeGen/SelectionDAGNodes.h

@@ -1497,6 +1497,14 @@ class MemSDNode : public SDNode {
     MMO->refineAlignment(NewMMO);
   }
 
+  void refineRanges(const MachineMemOperand *NewMMO) {
+    // If this node has range metadata that is different than NewMMO, clear the
+    // range metadata.
+    // FIXME: Union the ranges instead?
+    if (getRanges() && getRanges() != NewMMO->getRanges())
+      MMO->clearRanges();
+  }
+
   const SDValue &getChain() const { return getOperand(0); }
 
   const SDValue &getBasePtr() const {

llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp

@@ -1275,6 +1275,8 @@ SelectionDAG::AddModifiedNodeToCSEMaps(SDNode *N) {
       // to replace the dead one with the existing one.  This can cause
       // recursive merging of other unrelated nodes down the line.
       Existing->intersectFlagsWith(N->getFlags());
+      if (auto *MemNode = dyn_cast<MemSDNode>(Existing))
+        MemNode->refineRanges(cast<MemSDNode>(N)->getMemOperand());
       ReplaceAllUsesWith(N, Existing);
 
       // N is now dead. Inform the listeners and delete it.
@@ -9109,8 +9111,9 @@ SDValue SelectionDAG::getAtomic(unsigned Opcode, const SDLoc &dl, EVT MemVT,
   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
   ID.AddInteger(MMO->getFlags());
   void* IP = nullptr;
-  if (SDNode *E = FindNodeOrInsertPos(ID, dl, IP)) {
-    cast<AtomicSDNode>(E)->refineAlignment(MMO);
+  if (auto *E = cast_or_null<AtomicSDNode>(FindNodeOrInsertPos(ID, dl, IP))) {
+    E->refineAlignment(MMO);
+    E->refineRanges(MMO);
     return SDValue(E, 0);
   }
 
@@ -9401,8 +9404,9 @@ SDValue SelectionDAG::getLoad(ISD::MemIndexedMode AM, ISD::LoadExtType ExtType,
   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
   ID.AddInteger(MMO->getFlags());
   void *IP = nullptr;
-  if (SDNode *E = FindNodeOrInsertPos(ID, dl, IP)) {
-    cast<LoadSDNode>(E)->refineAlignment(MMO);
+  if (auto *E = cast_or_null<LoadSDNode>(FindNodeOrInsertPos(ID, dl, IP))) {
+    E->refineAlignment(MMO);
+    E->refineRanges(MMO);
     return SDValue(E, 0);
   }
   auto *N = newSDNode<LoadSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs, AM,
@@ -9622,8 +9626,9 @@ SDValue SelectionDAG::getLoadVP(ISD::MemIndexedMode AM,
   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
   ID.AddInteger(MMO->getFlags());
   void *IP = nullptr;
-  if (SDNode *E = FindNodeOrInsertPos(ID, dl, IP)) {
-    cast<VPLoadSDNode>(E)->refineAlignment(MMO);
+  if (auto *E = cast_or_null<VPLoadSDNode>(FindNodeOrInsertPos(ID, dl, IP))) {
+    E->refineAlignment(MMO);
+    E->refineRanges(MMO);
     return SDValue(E, 0);
   }
   auto *N = newSDNode<VPLoadSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs, AM,

llvm/test/CodeGen/RISCV/pr145363.ll

@@ -9,7 +9,7 @@ define i32 @f(ptr %p) {
 ; CHECK-NEXT:    lw a0, 0(a0)
 ; CHECK-NEXT:    lui a1, 294471
 ; CHECK-NEXT:    addi a1, a1, 1064
-; CHECK-NEXT:    or a0, a0, a1
+; CHECK-NEXT:    addw a0, a0, a1
 ; CHECK-NEXT:    ret
   %load = load i32, ptr %p, align 4, !range !0
   %load2 = load i32, ptr %p, align 4
@@ -25,7 +25,7 @@ define i32 @test(ptr %p, i32 %x, ptr %q) {
 ; CHECK-NEXT:    lw a1, 0(a0)
 ; CHECK-NEXT:    lui a0, 294471
 ; CHECK-NEXT:    addi a0, a0, 1064
-; CHECK-NEXT:    or a0, a1, a0
+; CHECK-NEXT:    addw a0, a1, a0
 ; CHECK-NEXT:    sw a1, 0(a2)
 ; CHECK-NEXT:    ret
   %load = load i32, ptr %p, align 4, !range !0