llvm-project/llvm/test/Transforms/SimplifyCFG/preserve-branchweights-swit...

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

141 lines
3.0 KiB
LLVM
Raw Normal View History

; RUN: opt -simplifycfg -S -o - < %s | FileCheck %s
declare void @func2(i32)
declare void @func4(i32)
declare void @func6(i32)
declare void @func8(i32)
;; test1 - create a switch with case 2 and case 4 from two branches: N == 2
;; and N == 4.
define void @test1(i32 %N) nounwind uwtable {
entry:
%cmp = icmp eq i32 %N, 2
br i1 %cmp, label %if.then, label %if.else, !prof !0
; CHECK: test1
; CHECK: switch i32 %N
; CHECK: ], !prof !0
if.then:
call void @func2(i32 %N) nounwind
br label %if.end9
if.else:
%cmp2 = icmp eq i32 %N, 4
br i1 %cmp2, label %if.then7, label %if.else8, !prof !1
if.then7:
call void @func4(i32 %N) nounwind
br label %if.end
if.else8:
call void @func8(i32 %N) nounwind
br label %if.end
if.end:
br label %if.end9
if.end9:
ret void
}
;; test2 - Merge two switches where PredDefault == BB.
define void @test2(i32 %M, i32 %N) nounwind uwtable {
entry:
%cmp = icmp sgt i32 %M, 2
br i1 %cmp, label %sw1, label %sw2
sw1:
switch i32 %N, label %sw2 [
i32 2, label %sw.bb
i32 3, label %sw.bb1
], !prof !2
; CHECK: test2
; CHECK: switch i32 %N, label %sw.epilog
; CHECK: i32 2, label %sw.bb
; CHECK: i32 3, label %sw.bb1
; CHECK: i32 4, label %sw.bb5
; CHECK: ], !prof !1
sw.bb:
call void @func2(i32 %N) nounwind
br label %sw.epilog
sw.bb1:
call void @func4(i32 %N) nounwind
br label %sw.epilog
sw2:
;; Here "case 2" is invalidated if control is transferred through default case
;; of the first switch.
switch i32 %N, label %sw.epilog [
i32 2, label %sw.bb4
i32 4, label %sw.bb5
], !prof !3
sw.bb4:
call void @func6(i32 %N) nounwind
br label %sw.epilog
sw.bb5:
call void @func8(i32 %N) nounwind
br label %sw.epilog
sw.epilog:
ret void
}
;; test3 - Merge two switches where PredDefault != BB.
define void @test3(i32 %M, i32 %N) nounwind uwtable {
entry:
%cmp = icmp sgt i32 %M, 2
br i1 %cmp, label %sw1, label %sw2
sw1:
switch i32 %N, label %sw.bb [
i32 2, label %sw2
i32 3, label %sw2
i32 1, label %sw.bb1
], !prof !4
; CHECK: test3
; CHECK: switch i32 %N, label %sw.bb
; CHECK: i32 1, label %sw.bb1
; CHECK: i32 3, label %sw.bb4
; CHECK: i32 2, label %sw.epilog
; CHECK: ], !prof !3
sw.bb:
call void @func2(i32 %N) nounwind
br label %sw.epilog
sw.bb1:
call void @func4(i32 %N) nounwind
br label %sw.epilog
sw2:
switch i32 %N, label %sw.epilog [
i32 3, label %sw.bb4
i32 4, label %sw.bb5
], !prof !5
sw.bb4:
call void @func6(i32 %N) nounwind
br label %sw.epilog
sw.bb5:
call void @func8(i32 %N) nounwind
br label %sw.epilog
sw.epilog:
ret void
}
!0 = !{!"branch_weights", i64 64, i64 4}
!1 = !{!"branch_weights", i64 4, i64 64}
; CHECK: !0 = !{!"branch_weights", i64 256, i64 4352, i64 16}
!2 = !{!"branch_weights", i64 4, i64 4, i64 8}
!3 = !{!"branch_weights", i64 8, i64 8, i64 4}
; CHECK: !1 = !{!"branch_weights", i64 32, i64 48, i64 96, i64 16}
!4 = !{!"branch_weights", i64 7, i64 6, i64 4, i64 3}
!5 = !{!"branch_weights", i64 17, i64 13, i64 9}
; CHECK: !3 = !{!"branch_weights", i64 7, i64 3, i64 4, i64 6}