Add support for llvm.uadd.with.overflow.

llvm-svn: 59926

1 files changed, 5 insertions, 5 deletions

diff --git a/llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp b/llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp
index c1157106148..d24c8462221 100644
--- a/llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp
@@ -4168,13 +4168,16 @@ SDValue SelectionDAGLegalize::LegalizeOp(SDValue Op) {
     break;
   }
 
-  case ISD::SADDO: {
+  case ISD::SADDO:
+  case ISD::UADDO: {
     SDValue LHS = LegalizeOp(Node->getOperand(0));
     SDValue RHS = LegalizeOp(Node->getOperand(1));
 
     SDValue Sum = DAG.getNode(ISD::ADD, LHS.getValueType(), LHS, RHS);
     MVT OType = SDValue(Node, 1).getValueType();
-    SDValue Cmp = DAG.getSetCC(OType, Sum, LHS, ISD::SETLT);
+    SDValue Cmp = DAG.getSetCC(OType, Sum, LHS,
+                               (Node->getOpcode() == ISD::SADDO) ?
+                               ISD::SETLT : ISD::SETULT);
 
     MVT ValueVTs[] = { LHS.getValueType(), OType };
     SDValue Ops[] = { Sum, Cmp };
@@ -4185,9 +4188,6 @@ SDValue SelectionDAGLegalize::LegalizeOp(SDValue Op) {
     DAG.ReplaceAllUsesOfValueWith(SDValue(Node, 1), SDValue(RNode, 1));
     break;
   }
-  case ISD::UADDO: {
-    break;
-  }
   }
   
   assert(Result.getValueType() == Op.getValueType() &&