[llvm] 621b10c - [InstSimplify] Add tests for a peculiar abs of abs form (NFC)

[Nikita Popov via llvm-commits]

Author: Nikita Popov
Date: 2020-09-05T21:42:22+02:00
New Revision: 621b10ca187bdd6de691338e48b288ea1c6a5822

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

LOG: [InstSimplify] Add tests for a peculiar abs of abs form (NFC)

This pattern shows up when canonicalizing to spf abs form to
intrinsic abs form.

Added: 
    

Modified: 
    llvm/test/Transforms/InstSimplify/abs_intrinsic.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstSimplify/abs_intrinsic.ll b/llvm/test/Transforms/InstSimplify/abs_intrinsic.ll
index aa299c1084b7..e9305a927c42 100644
--- a/llvm/test/Transforms/InstSimplify/abs_intrinsic.ll
+++ b/llvm/test/Transforms/InstSimplify/abs_intrinsic.ll
@@ -201,3 +201,63 @@ define i1 @abs_ule_int_min(i8 %x) {
   %c = icmp ule i8 %abs, 128
   ret i1 %c
 }
+
+define i32 @select_abs_of_abs_eq(i32 %x) {
+; CHECK-LABEL: @select_abs_of_abs_eq(
+; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
+; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[X]], 0
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], i32 [[NEG]], i32 [[ABS]]
+; CHECK-NEXT:    ret i32 [[SEL]]
+;
+  %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
+  %neg = sub i32 0, %abs
+  %cmp = icmp eq i32 %x, 0
+  %sel = select i1 %cmp, i32 %neg, i32 %abs
+  ret i32 %sel
+}
+
+define i32 @select_abs_of_abs_ne(i32 %x) {
+; CHECK-LABEL: @select_abs_of_abs_ne(
+; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
+; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i32 [[X]], 0
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], i32 [[ABS]], i32 [[NEG]]
+; CHECK-NEXT:    ret i32 [[SEL]]
+;
+  %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
+  %neg = sub i32 0, %abs
+  %cmp = icmp ne i32 %x, 0
+  %sel = select i1 %cmp, i32 %abs, i32 %neg
+  ret i32 %sel
+}
+
+define i32 @select_nabs_of_abs_eq(i32 %x) {
+; CHECK-LABEL: @select_nabs_of_abs_eq(
+; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
+; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[X]], 0
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], i32 [[ABS]], i32 [[NEG]]
+; CHECK-NEXT:    ret i32 [[SEL]]
+;
+  %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
+  %neg = sub i32 0, %abs
+  %cmp = icmp eq i32 %x, 0
+  %sel = select i1 %cmp, i32 %abs, i32 %neg
+  ret i32 %sel
+}
+
+define i32 @select_nabs_of_abs_ne(i32 %x) {
+; CHECK-LABEL: @select_nabs_of_abs_ne(
+; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
+; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i32 [[X]], 0
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], i32 [[NEG]], i32 [[ABS]]
+; CHECK-NEXT:    ret i32 [[SEL]]
+;
+  %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
+  %neg = sub i32 0, %abs
+  %cmp = icmp ne i32 %x, 0
+  %sel = select i1 %cmp, i32 %neg, i32 %abs
+  ret i32 %sel
+}