[InstCombine] Fold usub_sat((sub nuw C1, A), C2) to usub_sat(C1 - C2, A) or 0

Author: elhewaty

llvm/lib/Transforms/InstCombine/InstCombineCalls.cpp

@@ -2139,6 +2139,23 @@ Instruction *InstCombinerImpl::visitCallInst(CallInst &CI) {
       }
     }
 
+    // usub_sat((sub nuw C1, A), C2) -> usub_sat(C1 - C2, A) if C2 u< C1
+    // usub_sat((sub nuw C1, A), C2) -> 0 otherwise
+    const APInt *C1, *C2;
+    Value *A;
+    if (IID == Intrinsic::usub_sat &&
+        match(Arg0, m_OneUse(m_NUWSub(m_APInt(C1), m_Value(A)))) &&
+        match(Arg1, m_APInt(C2))) {
+
+      if (C2->ult(*C1)) {
+        auto *New = Builder.CreateBinaryIntrinsic(
+            Intrinsic::usub_sat, ConstantInt::get(Ty, *C1 - *C2), A);
+        return replaceInstUsesWith(*SI, New);
+      } else {
+        return replaceInstUsesWith(*SI, ConstantInt::get(Ty, 0));
+      }
+    }
+
     // ssub.sat(X, C) -> sadd.sat(X, -C) if C != MIN
     Constant *C;
     if (IID == Intrinsic::ssub_sat && match(Arg1, m_Constant(C)) &&

llvm/test/Transforms/InstCombine/unsigned_saturated_sub.ll

@@ -12,8 +12,7 @@ declare void @usei1(i1)
 
 define i32 @usub_sat_C1_C2(i32 %a){
 ; CHECK-LABEL: @usub_sat_C1_C2(
-; CHECK-NEXT:    [[ADD:%.*]] = sub nuw i32 64, [[A:%.*]]
-; CHECK-NEXT:    [[COND:%.*]] = call i32 @llvm.usub.sat.i32(i32 [[ADD]], i32 14)
+; CHECK-NEXT:    [[COND:%.*]] = call i32 @llvm.usub.sat.i32(i32 50, i32 [[A:%.*]])
 ; CHECK-NEXT:    ret i32 [[COND]]
 ;
   %add = sub nuw i32 64, %a
@@ -23,9 +22,7 @@ define i32 @usub_sat_C1_C2(i32 %a){
 
 define i32 @usub_sat_C1_C2_produce_0(i32 %a){
 ; CHECK-LABEL: @usub_sat_C1_C2_produce_0(
-; CHECK-NEXT:    [[ADD:%.*]] = sub nuw i32 14, [[A:%.*]]
-; CHECK-NEXT:    [[COND:%.*]] = call i32 @llvm.usub.sat.i32(i32 [[ADD]], i32 14)
-; CHECK-NEXT:    ret i32 [[COND]]
+; CHECK-NEXT:    ret i32 0
 ;
   %add = sub nuw i32 14, %a
   %cond = call i32 @llvm.usub.sat.i32(i32 %add, i32 14)
@@ -34,9 +31,7 @@ define i32 @usub_sat_C1_C2_produce_0(i32 %a){
 
 define i32 @usub_sat_C1_C2_produce_0_too(i32 %a){
 ; CHECK-LABEL: @usub_sat_C1_C2_produce_0_too(
-; CHECK-NEXT:    [[ADD:%.*]] = sub nuw i32 12, [[A:%.*]]
-; CHECK-NEXT:    [[COND:%.*]] = call i32 @llvm.usub.sat.i32(i32 [[ADD]], i32 14)
-; CHECK-NEXT:    ret i32 [[COND]]
+; CHECK-NEXT:    ret i32 0
 ;
   %add = sub nuw i32 12, %a
   %cond = call i32 @llvm.usub.sat.i32(i32 %add, i32 14)
@@ -46,8 +41,7 @@ define i32 @usub_sat_C1_C2_produce_0_too(i32 %a){
 ; vector tests
 define <2 x i16> @usub_sat_C1_C2_splat(<2 x i16> %a) {
 ; CHECK-LABEL: @usub_sat_C1_C2_splat(
-; CHECK-NEXT:    [[ADD:%.*]] = sub nuw <2 x i16> <i16 64, i16 64>, [[A:%.*]]
-; CHECK-NEXT:    [[COND:%.*]] = call <2 x i16> @llvm.usub.sat.v2i16(<2 x i16> [[ADD]], <2 x i16> <i16 14, i16 14>)
+; CHECK-NEXT:    [[COND:%.*]] = call <2 x i16> @llvm.usub.sat.v2i16(<2 x i16> <i16 50, i16 50>, <2 x i16> [[A:%.*]])
 ; CHECK-NEXT:    ret <2 x i16> [[COND]]
 ;
   %add = sub nuw <2 x i16> <i16 64, i16 64>, %a
@@ -57,9 +51,7 @@ define <2 x i16> @usub_sat_C1_C2_splat(<2 x i16> %a) {
 
 define <2 x i16> @usub_sat_C1_C2_splat_produce_0(<2 x i16> %a){
 ; CHECK-LABEL: @usub_sat_C1_C2_splat_produce_0(
-; CHECK-NEXT:    [[ADD:%.*]] = sub nuw <2 x i16> <i16 14, i16 14>, [[A:%.*]]
-; CHECK-NEXT:    [[COND:%.*]] = call <2 x i16> @llvm.usub.sat.v2i16(<2 x i16> [[ADD]], <2 x i16> <i16 14, i16 14>)
-; CHECK-NEXT:    ret <2 x i16> [[COND]]
+; CHECK-NEXT:    ret <2 x i16> zeroinitializer
 ;
   %add = sub nuw <2 x i16> <i16 14, i16 14>, %a
   %cond = call <2 x i16> @llvm.usub.sat.v2i16(<2 x i16> %add, <2 x i16> <i16 14, i16 14>)
@@ -68,9 +60,7 @@ define <2 x i16> @usub_sat_C1_C2_splat_produce_0(<2 x i16> %a){
 
 define <2 x i16> @usub_sat_C1_C2_splat_produce_0_too(<2 x i16> %a){
 ; CHECK-LABEL: @usub_sat_C1_C2_splat_produce_0_too(
-; CHECK-NEXT:    [[ADD:%.*]] = sub nuw <2 x i16> <i16 12, i16 12>, [[A:%.*]]
-; CHECK-NEXT:    [[COND:%.*]] = call <2 x i16> @llvm.usub.sat.v2i16(<2 x i16> [[ADD]], <2 x i16> <i16 14, i16 14>)
-; CHECK-NEXT:    ret <2 x i16> [[COND]]
+; CHECK-NEXT:    ret <2 x i16> zeroinitializer
 ;
   %add = sub nuw <2 x i16> <i16 12, i16 12>, %a
   %cond = call <2 x i16> @llvm.usub.sat.v2i16(<2 x i16> %add, <2 x i16> <i16 14, i16 14>)