[llvm] 4fb8e0b - [JumpThreading] Add a test for handling zero !prof branch_weights

[Yevgeny Rouban via llvm-commits]

Author: Yevgeny Rouban
Date: 2022-01-26T13:42:21+07:00
New Revision: 4fb8e0b8621a641b38cf2218cc20301da346cedf

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

LOG: [JumpThreading] Add a test for handling zero !prof branch_weights

The test was a part of the revision D81499 and should have been
added with commit 707836ed4ed.

Reviewed By: yamauchi, wenlei
Differential Revision: https://reviews.llvm.org/D81499

Added: 
    llvm/test/Transforms/JumpThreading/threading_prof3.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/JumpThreading/threading_prof3.ll b/llvm/test/Transforms/JumpThreading/threading_prof3.ll
new file mode 100644
index 0000000000000..b235c9b72b938
--- /dev/null
+++ b/llvm/test/Transforms/JumpThreading/threading_prof3.ll
@@ -0,0 +1,30 @@
+; RUN: opt -jump-threading -S < %s | FileCheck %s
+; RUN: opt -passes=jump-threading -S < %s | FileCheck %s
+
+; Check that all zero branch weights do not cause a crash.
+define void @zero_branch_weights(i32 %tmp, i32 %tmp3) {
+bb:
+  %tmp1 = icmp eq i32 %tmp, 1
+  br i1 %tmp1, label %bb5, label %bb2
+; CHECK-NOT: br i1 %tmp1,{{.*}} !prof
+
+bb2:
+  %tmp4 = icmp ne i32 %tmp3, 1
+  br label %bb5
+; CHECK: br i1 %tmp4, {{.*}} !prof ![[PROF:[0-9]+]]
+
+bb5:
+  %tmp6 = phi i1 [ false, %bb ], [ %tmp4, %bb2 ]
+  br i1 %tmp6, label %bb8, label %bb7, !prof !{!"branch_weights", i32 0, i32 0}
+
+bb7:
+  br label %bb9
+
+bb8:
+  br label %bb9
+
+bb9:
+  ret void
+}
+
+;CHECK: ![[PROF]] = !{!"branch_weights", i32 0, i32 0}