[llvm] e6c5d13 - [InstCombine] add tests for SimplifySelectsFeedingBinaryOp(); NFC

[Sanjay Patel via llvm-commits]

Author: Sanjay Patel
Date: 2022-09-07T08:23:35-04:00
New Revision: e6c5d13ac762e7535191e6c3e16b00fc93433c73

URL: https://github.com/llvm/llvm-project/commit/e6c5d13ac762e7535191e6c3e16b00fc93433c73
DIFF: https://github.com/llvm/llvm-project/commit/e6c5d13ac762e7535191e6c3e16b00fc93433c73.diff

LOG: [InstCombine] add tests for SimplifySelectsFeedingBinaryOp(); NFC

Added: 
    

Modified: 
    llvm/test/Transforms/InstCombine/add.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/add.ll b/llvm/test/Transforms/InstCombine/add.ll
index b6ffa0015a786..02c46eced9eb1 100644
--- a/llvm/test/Transforms/InstCombine/add.ll
+++ b/llvm/test/Transforms/InstCombine/add.ll
@@ -2060,6 +2060,70 @@ define i8 @select_negate_or_nonzero_use(i1 %b, i8 %x, i8 %y) {
   ret i8 %add
 }
 
+define i5 @select_negate_not(i1 %b, i5 %x, i5 %y) {
+; CHECK-LABEL: @select_negate_not(
+; CHECK-NEXT:    [[NEGX:%.*]] = sub i5 0, [[X:%.*]]
+; CHECK-NEXT:    [[NOTY:%.*]] = xor i5 [[Y:%.*]], -1
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[B:%.*]], i5 [[NOTY]], i5 [[NEGX]]
+; CHECK-NEXT:    [[ADD:%.*]] = add i5 [[SEL]], [[Y]]
+; CHECK-NEXT:    ret i5 [[ADD]]
+;
+  %negx = sub i5 0, %x
+  %noty = xor i5 %y, -1
+  %sel = select i1 %b, i5 %noty, i5 %negx
+  %add = add i5 %sel, %y
+  ret i5 %add
+}
+
+define i5 @select_negate_not_commute(i1 %b, i5 %x, i5 %p) {
+; CHECK-LABEL: @select_negate_not_commute(
+; CHECK-NEXT:    [[Y:%.*]] = mul i5 [[P:%.*]], [[P]]
+; CHECK-NEXT:    [[NEGX:%.*]] = sub i5 0, [[X:%.*]]
+; CHECK-NEXT:    [[NOTY:%.*]] = xor i5 [[Y]], -1
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[B:%.*]], i5 [[NOTY]], i5 [[NEGX]]
+; CHECK-NEXT:    [[ADD:%.*]] = add i5 [[Y]], [[SEL]]
+; CHECK-NEXT:    ret i5 [[ADD]]
+;
+  %y = mul i5 %p, %p ; thwart complexity-based canonicalization
+  %negx = sub i5 0, %x
+  %noty = xor i5 %y, -1
+  %sel = select i1 %b, i5 %noty, i5 %negx
+  %add = add i5 %y, %sel
+  ret i5 %add
+}
+
+define i5 @select_negate_not_swap(i1 %b, i5 %x, i5 %y) {
+; CHECK-LABEL: @select_negate_not_swap(
+; CHECK-NEXT:    [[NEGX:%.*]] = sub i5 0, [[X:%.*]]
+; CHECK-NEXT:    [[NOTY:%.*]] = xor i5 [[Y:%.*]], -1
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[B:%.*]], i5 [[NEGX]], i5 [[NOTY]]
+; CHECK-NEXT:    [[ADD:%.*]] = add i5 [[SEL]], [[Y]]
+; CHECK-NEXT:    ret i5 [[ADD]]
+;
+  %negx = sub i5 0, %x
+  %noty = xor i5 %y, -1
+  %sel = select i1 %b, i5 %negx, i5 %noty
+  %add = add i5 %sel, %y
+  ret i5 %add
+}
+
+define i5 @select_negate_not_swap_commute(i1 %b, i5 %x, i5 %p) {
+; CHECK-LABEL: @select_negate_not_swap_commute(
+; CHECK-NEXT:    [[Y:%.*]] = mul i5 [[P:%.*]], [[P]]
+; CHECK-NEXT:    [[NEGX:%.*]] = sub i5 0, [[X:%.*]]
+; CHECK-NEXT:    [[NOTY:%.*]] = xor i5 [[Y]], -1
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[B:%.*]], i5 [[NEGX]], i5 [[NOTY]]
+; CHECK-NEXT:    [[ADD:%.*]] = add i5 [[Y]], [[SEL]]
+; CHECK-NEXT:    ret i5 [[ADD]]
+;
+  %y = mul i5 %p, %p ; thwart complexity-based canonicalization
+  %negx = sub i5 0, %x
+  %noty = xor i5 %y, -1
+  %sel = select i1 %b, i5 %negx, i5 %noty
+  %add = add i5 %y, %sel
+  ret i5 %add
+}
+
 define i32 @add_select_sub_both_arms_simplify(i1 %b, i32 %a) {
 ; CHECK-LABEL: @add_select_sub_both_arms_simplify(
 ; CHECK-NEXT:    [[ADD:%.*]] = select i1 [[B:%.*]], i32 [[A:%.*]], i32 99