Match wrapper node for address

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@70731 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/Target/MSP430/MSP430ISelDAGToDAG.cpp b/lib/Target/MSP430/MSP430ISelDAGToDAG.cpp
index 3b28668f550658bcc497f4a52796372b0f0ab61a..e564e63598a58104bdc77a54e63dc2697b058ded 100644 (file)
--- a/lib/Target/MSP430/MSP430ISelDAGToDAG.cpp
+++ b/lib/Target/MSP430/MSP430ISelDAGToDAG.cpp
@@ -79,7 +79,8 @@ bool MSP430DAGToDAGISel::SelectAddr(SDValue Op, SDValue Addr,
     return false;
 
   // Operand is a result from ADD with constant operand which fits into i16.
-  if (Addr.getOpcode() == ISD::ADD) {
+  switch (Addr.getOpcode()) {
+  case ISD::ADD:
     if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Addr.getOperand(1))) {
       uint64_t CVal = CN->getZExtValue();
       // Offset should fit into 16 bits.
@@ -94,7 +95,21 @@ bool MSP430DAGToDAGISel::SelectAddr(SDValue Op, SDValue Addr,
         return true;
       }
     }
-  }
+    break;
+  case MSP430ISD::Wrapper:
+    SDValue N0 = Addr.getOperand(0);
+    if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(N0)) {
+      // We can match addresses of globals without any offsets
+      if (!G->getOffset()) {
+        Base = CurDAG->getTargetGlobalAddress(G->getGlobal(),
+                                              MVT::i16, 0);
+        Disp = CurDAG->getTargetConstant(0, MVT::i16);
+
+        return true;
+      }
+    }
+    break;
+  };
 
   Base = Addr;
   Disp = CurDAG->getTargetConstant(0, MVT::i16);