[llvm] 595ffc5 - [InstCombine] add tests for add with sub-from-constant operand; NFC

[Sanjay Patel via llvm-commits]

Author: Sanjay Patel
Date: 2023-02-20T13:45:06-05:00
New Revision: 595ffc56be70d42abeaeac3ee1c47c6eb6c894b3

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

LOG: [InstCombine] add tests for add with sub-from-constant operand; NFC

Added: 
    

Modified: 
    llvm/test/Transforms/InstCombine/addsub-constant-folding.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/addsub-constant-folding.ll b/llvm/test/Transforms/InstCombine/addsub-constant-folding.ll
index 90383eab4247..8dd6b6bcdcd8 100644
--- a/llvm/test/Transforms/InstCombine/addsub-constant-folding.ll
+++ b/llvm/test/Transforms/InstCombine/addsub-constant-folding.ll
@@ -587,3 +587,51 @@ define i32 @addsub_combine_constants_use3(i32 %x, i32 %y) {
   %a2 = add i32 %a1, %s
   ret i32 %a2
 }
+
+define i5 @sub_from_constant(i5 %x, i5 %y) {
+; CHECK-LABEL: @sub_from_constant(
+; CHECK-NEXT:    [[SUB:%.*]] = sub i5 10, [[X:%.*]]
+; CHECK-NEXT:    [[R:%.*]] = add i5 [[SUB]], [[Y:%.*]]
+; CHECK-NEXT:    ret i5 [[R]]
+;
+  %sub = sub i5 10, %x
+  %r = add i5 %sub, %y
+  ret i5 %r
+}
+
+define i5 @sub_from_constant_commute(i5 %x, i5 %p) {
+; CHECK-LABEL: @sub_from_constant_commute(
+; CHECK-NEXT:    [[Y:%.*]] = mul i5 [[P:%.*]], [[P]]
+; CHECK-NEXT:    [[SUB:%.*]] = sub nsw i5 10, [[X:%.*]]
+; CHECK-NEXT:    [[R:%.*]] = add nsw i5 [[Y]], [[SUB]]
+; CHECK-NEXT:    ret i5 [[R]]
+;
+  %y = mul i5 %p, %p  ; thwart complexity-based canonicalization
+  %sub = sub nsw i5 10, %x
+  %r = add nsw i5 %y, %sub
+  ret i5 %r
+}
+
+define <2 x i8> @sub_from_constant_vec(<2 x i8> %x, <2 x i8> %y) {
+; CHECK-LABEL: @sub_from_constant_vec(
+; CHECK-NEXT:    [[SUB:%.*]] = sub nuw <2 x i8> <i8 2, i8 -42>, [[X:%.*]]
+; CHECK-NEXT:    [[R:%.*]] = add nuw <2 x i8> [[SUB]], [[Y:%.*]]
+; CHECK-NEXT:    ret <2 x i8> [[R]]
+;
+  %sub = sub nuw <2 x i8> <i8 2, i8 -42>, %x
+  %r = add nuw <2 x i8> %sub, %y
+  ret <2 x i8> %r
+}
+
+define i8 @sub_from_constant_extra_use(i8 %x, i8 %y) {
+; CHECK-LABEL: @sub_from_constant_extra_use(
+; CHECK-NEXT:    [[SUB:%.*]] = sub i8 1, [[X:%.*]]
+; CHECK-NEXT:    call void @use(i8 [[SUB]])
+; CHECK-NEXT:    [[R:%.*]] = add i8 [[SUB]], [[Y:%.*]]
+; CHECK-NEXT:    ret i8 [[R]]
+;
+  %sub = sub i8 1, %x
+  call void @use(i8 %sub)
+  %r = add i8 %sub, %y
+  ret i8 %r
+}