diff options
Diffstat (limited to 'test/Transforms/SimplifyCFG/preserve-branchweights.ll')
| -rw-r--r-- | test/Transforms/SimplifyCFG/preserve-branchweights.ll | 230 |
1 files changed, 229 insertions, 1 deletions
diff --git a/test/Transforms/SimplifyCFG/preserve-branchweights.ll b/test/Transforms/SimplifyCFG/preserve-branchweights.ll index c791785..beef527 100644 --- a/test/Transforms/SimplifyCFG/preserve-branchweights.ll +++ b/test/Transforms/SimplifyCFG/preserve-branchweights.ll @@ -79,10 +79,238 @@ Z: ret void } +;; test5 - The case where it jumps to the default target will be removed. +define void @test5(i32 %M, i32 %N) nounwind uwtable { +entry: + switch i32 %N, label %sw2 [ + i32 1, label %sw2 + i32 2, label %sw.bb + i32 3, label %sw.bb1 + ], !prof !3 +; CHECK: test5 +; CHECK: switch i32 %N, label %sw2 [ +; CHECK: i32 3, label %sw.bb1 +; CHECK: i32 2, label %sw.bb +; CHECK: ], !prof !2 + +sw.bb: + call void @helper(i32 0) + br label %sw.epilog + +sw.bb1: + call void @helper(i32 1) + br label %sw.epilog + +sw2: + call void @helper(i32 2) + br label %sw.epilog + +sw.epilog: + ret void +} + +;; test6 - Some cases of the second switch are pruned during optimization. +;; Then the second switch will be converted to a branch, finally, the first +;; switch and the branch will be merged into a single switch. +define void @test6(i32 %M, i32 %N) nounwind uwtable { +entry: + switch i32 %N, label %sw2 [ + i32 1, label %sw2 + i32 2, label %sw.bb + i32 3, label %sw.bb1 + ], !prof !4 +; CHECK: test6 +; CHECK: switch i32 %N, label %sw.epilog +; CHECK: i32 3, label %sw.bb1 +; CHECK: i32 2, label %sw.bb +; CHECK: i32 4, label %sw.bb5 +; CHECK: ], !prof !3 + +sw.bb: + call void @helper(i32 0) + br label %sw.epilog + +sw.bb1: + call void @helper(i32 1) + br label %sw.epilog + +sw2: +;; Here "case 2" is invalidated since the default case of the first switch +;; does not include "case 2". + switch i32 %N, label %sw.epilog [ + i32 2, label %sw.bb4 + i32 4, label %sw.bb5 + ], !prof !5 + +sw.bb4: + call void @helper(i32 2) + br label %sw.epilog + +sw.bb5: + call void @helper(i32 3) + br label %sw.epilog + +sw.epilog: + ret void +} + +;; This test is based on test1 but swapped the targets of the second branch. +define void @test1_swap(i1 %a, i1 %b) { +; CHECK: @test1_swap +entry: + br i1 %a, label %Y, label %X, !prof !0 +; CHECK: br i1 %or.cond, label %Y, label %Z, !prof !4 + +X: + %c = or i1 %b, false + br i1 %c, label %Y, label %Z, !prof !1 + +Y: + call void @helper(i32 0) + ret void + +Z: + call void @helper(i32 1) + ret void +} + +define void @test7(i1 %a, i1 %b) { +; CHECK: @test7 +entry: + %c = or i1 %b, false + br i1 %a, label %Y, label %X, !prof !0 +; CHECK: br i1 %brmerge, label %Y, label %Z, !prof !5 + +X: + br i1 %c, label %Y, label %Z, !prof !6 + +Y: + call void @helper(i32 0) + ret void + +Z: + call void @helper(i32 1) + ret void +} + +; Test basic folding to a conditional branch. +define void @test8(i64 %x, i64 %y) nounwind { +; CHECK: @test8 +entry: + %lt = icmp slt i64 %x, %y +; CHECK: br i1 %lt, label %a, label %b, !prof !6 + %qux = select i1 %lt, i32 0, i32 2 + switch i32 %qux, label %bees [ + i32 0, label %a + i32 1, label %b + i32 2, label %b + ], !prof !7 +a: + call void @helper(i32 0) nounwind + ret void +b: + call void @helper(i32 1) nounwind + ret void +bees: + call void @helper(i32 2) nounwind + ret void +} + +; Test edge splitting when the default target has icmp and unconditinal +; branch +define i1 @test9(i32 %x, i32 %y) nounwind { +; CHECK: @test9 +entry: + switch i32 %x, label %bees [ + i32 0, label %a + i32 1, label %end + i32 2, label %end + ], !prof !7 +; CHECK: switch i32 %x, label %bees [ +; CHECK: i32 0, label %a +; CHECK: i32 1, label %end +; CHECK: i32 2, label %end +; CHECK: i32 92, label %end +; CHECK: ], !prof !7 + +a: + call void @helper(i32 0) nounwind + %reta = icmp slt i32 %x, %y + ret i1 %reta + +bees: + %tmp = icmp eq i32 %x, 92 + br label %end + +end: +; CHECK: end: +; CHECK: %ret = phi i1 [ true, %entry ], [ false, %bees ], [ true, %entry ], [ true, %entry ] + %ret = phi i1 [ true, %entry ], [%tmp, %bees], [true, %entry] + call void @helper(i32 2) nounwind + ret i1 %ret +} + +define void @test10(i32 %x) nounwind readnone ssp noredzone { +entry: + switch i32 %x, label %lor.rhs [ + i32 2, label %lor.end + i32 1, label %lor.end + i32 3, label %lor.end + ], !prof !7 + +lor.rhs: + call void @helper(i32 1) nounwind + ret void + +lor.end: + call void @helper(i32 0) nounwind + ret void + +; CHECK: test10 +; CHECK: %x.off = add i32 %x, -1 +; CHECK: %switch = icmp ult i32 %x.off, 3 +; CHECK: br i1 %switch, label %lor.end, label %lor.rhs, !prof !8 +} + +; Remove dead cases from the switch. +define void @test11(i32 %x) nounwind { + %i = shl i32 %x, 1 + switch i32 %i, label %a [ + i32 21, label %b + i32 24, label %c + ], !prof !8 +; CHECK: %cond = icmp eq i32 %i, 24 +; CHECK: br i1 %cond, label %c, label %a, !prof !9 + +a: + call void @helper(i32 0) nounwind + ret void +b: + call void @helper(i32 1) nounwind + ret void +c: + call void @helper(i32 2) nounwind + ret void +} + !0 = metadata !{metadata !"branch_weights", i32 3, i32 5} !1 = metadata !{metadata !"branch_weights", i32 1, i32 1} !2 = metadata !{metadata !"branch_weights", i32 1, i32 2} +!3 = metadata !{metadata !"branch_weights", i32 4, i32 3, i32 2, i32 1} +!4 = metadata !{metadata !"branch_weights", i32 4, i32 3, i32 2, i32 1} +!5 = metadata !{metadata !"branch_weights", i32 7, i32 6, i32 5} +!6 = metadata !{metadata !"branch_weights", i32 1, i32 3} +!7 = metadata !{metadata !"branch_weights", i32 33, i32 9, i32 8, i32 7} +!8 = metadata !{metadata !"branch_weights", i32 33, i32 9, i32 8} ; CHECK: !0 = metadata !{metadata !"branch_weights", i32 5, i32 11} ; CHECK: !1 = metadata !{metadata !"branch_weights", i32 1, i32 5} -; CHECK-NOT: !2 +; CHECK: !2 = metadata !{metadata !"branch_weights", i32 7, i32 1, i32 2} +; CHECK: !3 = metadata !{metadata !"branch_weights", i32 49, i32 12, i32 24, i32 35} +; CHECK: !4 = metadata !{metadata !"branch_weights", i32 11, i32 5} +; CHECK: !5 = metadata !{metadata !"branch_weights", i32 17, i32 15} +; CHECK: !6 = metadata !{metadata !"branch_weights", i32 9, i32 7} +; CHECK: !7 = metadata !{metadata !"branch_weights", i32 17, i32 9, i32 8, i32 7, i32 17} +; CHECK: !8 = metadata !{metadata !"branch_weights", i32 24, i32 33} +; CHECK: !9 = metadata !{metadata !"branch_weights", i32 8, i32 33} +; CHECK-NOT: !9 |
