[llvm] e218da7 - [JumpThreading] Add a test for simplification of cast of any op; NFC

[Juneyoung Lee via llvm-commits]

Author: Juneyoung Lee
Date: 2020-08-04T19:08:58+09:00
New Revision: e218da7ff39df3aeb7c923fbd75aaa8ace581e37

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

LOG: [JumpThreading] Add a test for simplification of cast of any op; NFC

Added: 
    llvm/test/Transforms/JumpThreading/threadable-edge-cast.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/JumpThreading/threadable-edge-cast.ll b/llvm/test/Transforms/JumpThreading/threadable-edge-cast.ll
new file mode 100644
index 0000000000000..71c34dea3a069
--- /dev/null
+++ b/llvm/test/Transforms/JumpThreading/threadable-edge-cast.ll
@@ -0,0 +1,101 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -jump-threading -S < %s | FileCheck %s
+
+declare i32 @f1()
+declare i32 @f2()
+declare void @f3()
+
+; trunc(and) should be simplified
+define i32 @test(i1 %cond0) {
+; CHECK-LABEL: @test(
+; CHECK-NEXT:    br i1 [[COND0:%.*]], label [[T1:%.*]], label [[F1:%.*]]
+; CHECK:       T1:
+; CHECK-NEXT:    [[V1:%.*]] = call i32 @f1()
+; CHECK-NEXT:    br label [[MERGE:%.*]]
+; CHECK:       F1:
+; CHECK-NEXT:    [[V2:%.*]] = call i32 @f2()
+; CHECK-NEXT:    br label [[MERGE]]
+; CHECK:       Merge:
+; CHECK-NEXT:    [[A:%.*]] = phi i32 [ 10, [[T1]] ], [ 0, [[F1]] ]
+; CHECK-NEXT:    [[B:%.*]] = phi i32 [ [[V1]], [[T1]] ], [ [[V2]], [[F1]] ]
+; CHECK-NEXT:    [[AND:%.*]] = and i32 [[A]], 1
+; CHECK-NEXT:    [[COND:%.*]] = trunc i32 [[AND]] to i1
+; CHECK-NEXT:    br i1 [[COND]], label [[T2:%.*]], label [[F2:%.*]]
+; CHECK:       T2:
+; CHECK-NEXT:    call void @f3()
+; CHECK-NEXT:    ret i32 [[B]]
+; CHECK:       F2:
+; CHECK-NEXT:    ret i32 [[B]]
+;
+  br i1 %cond0, label %T1, label %F1
+T1:
+  %v1 = call i32 @f1()
+  br label %Merge
+
+F1:
+  %v2 = call i32 @f2()
+  br label %Merge
+
+Merge:
+  %A = phi i32 [10, %T1], [0, %F1]
+  %B = phi i32 [%v1, %T1], [%v2, %F1]
+  %And = and i32 %A, 1
+  %cond = trunc i32 %And to i1
+  br i1 %cond, label %T2, label %F2
+
+T2:
+  call void @f3()
+  ret i32 %B
+
+F2:
+  ret i32 %B
+}
+
+; trunc(select) should be simplified
+define i32 @test2(i1 %cond0) {
+; CHECK-LABEL: @test2(
+; CHECK-NEXT:    br i1 [[COND0:%.*]], label [[T1:%.*]], label [[F1:%.*]]
+; CHECK:       T1:
+; CHECK-NEXT:    [[V1:%.*]] = call i32 @f1()
+; CHECK-NEXT:    br label [[MERGE:%.*]]
+; CHECK:       F1:
+; CHECK-NEXT:    [[V2:%.*]] = call i32 @f2()
+; CHECK-NEXT:    br label [[MERGE]]
+; CHECK:       Merge:
+; CHECK-NEXT:    [[A:%.*]] = phi i1 [ true, [[T1]] ], [ false, [[F1]] ]
+; CHECK-NEXT:    [[B:%.*]] = phi i32 [ [[V1]], [[T1]] ], [ [[V2]], [[F1]] ]
+; CHECK-NEXT:    [[A2:%.*]] = xor i1 [[A]], true
+; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[A2]], i32 10, i32 0
+; CHECK-NEXT:    [[COND:%.*]] = trunc i32 [[SEL]] to i1
+; CHECK-NEXT:    br i1 [[COND]], label [[T2:%.*]], label [[F2:%.*]]
+; CHECK:       T2:
+; CHECK-NEXT:    call void @f3()
+; CHECK-NEXT:    ret i32 [[B]]
+; CHECK:       F2:
+; CHECK-NEXT:    ret i32 [[B]]
+;
+  br i1 %cond0, label %T1, label %F1
+T1:
+  %v1 = call i32 @f1()
+  br label %Merge
+
+F1:
+  %v2 = call i32 @f2()
+  br label %Merge
+
+Merge:
+  %A = phi i1 [1, %T1], [0, %F1]
+  %B = phi i32 [%v1, %T1], [%v2, %F1]
+  %A2 = xor i1 %A, 1
+  %sel = select i1 %A2, i32 10, i32 0
+  %cond = trunc i32 %sel to i1
+  br i1 %cond, label %T2, label %F2
+
+T2:
+  call void @f3()
+  ret i32 %B
+
+F2:
+  ret i32 %B
+}
+