test/Transforms/SimplifyCFG/preserve-branchweights-switch-create.ll

   1 ; RUN: opt -simplifycfg -S -o - < %s | FileCheck %s
   2
   3 declare void @func2(i32)
   4 declare void @func4(i32)
   5 declare void @func6(i32)
   6 declare void @func8(i32)
   7
   8 ;; test1 - create a switch with case 2 and case 4 from two branches: N == 2
   9 ;; and N == 4.
  10 define void @test1(i32 %N) nounwind uwtable {
  11 entry:
  12   %cmp = icmp eq i32 %N, 2
  13   br i1 %cmp, label %if.then, label %if.else, !prof !0
  14 ; CHECK: test1
  15 ; CHECK: switch i32 %N
  16 ; CHECK: ], !prof !0
  17
  18 if.then:
  19   call void @func2(i32 %N) nounwind
  20   br label %if.end9
  21
  22 if.else:
  23   %cmp2 = icmp eq i32 %N, 4
  24   br i1 %cmp2, label %if.then7, label %if.else8, !prof !1
  25
  26 if.then7:
  27   call void @func4(i32 %N) nounwind
  28   br label %if.end
  29
  30 if.else8:
  31   call void @func8(i32 %N) nounwind
  32   br label %if.end
  33
  34 if.end:
  35   br label %if.end9
  36
  37 if.end9:
  38   ret void
  39 }
  40
  41 ;; test2 - Merge two switches where PredDefault == BB.
  42 define void @test2(i32 %M, i32 %N) nounwind uwtable {
  43 entry:
  44   %cmp = icmp sgt i32 %M, 2
  45   br i1 %cmp, label %sw1, label %sw2
  46
  47 sw1:
  48   switch i32 %N, label %sw2 [
  49     i32 2, label %sw.bb
  50     i32 3, label %sw.bb1
  51   ], !prof !2
  52 ; CHECK: test2
  53 ; CHECK: switch i32 %N, label %sw.epilog
  54 ; CHECK: i32 2, label %sw.bb
  55 ; CHECK: i32 3, label %sw.bb1
  56 ; CHECK: i32 4, label %sw.bb5
  57 ; CHECK: ], !prof !1
  58
  59 sw.bb:
  60   call void @func2(i32 %N) nounwind
  61   br label %sw.epilog
  62
  63 sw.bb1:
  64   call void @func4(i32 %N) nounwind
  65   br label %sw.epilog
  66
  67 sw2:
  68 ;; Here "case 2" is invalidated if control is transferred through default case
  69 ;; of the first switch.
  70   switch i32 %N, label %sw.epilog [
  71     i32 2, label %sw.bb4
  72     i32 4, label %sw.bb5
  73   ], !prof !3
  74
  75 sw.bb4:
  76   call void @func6(i32 %N) nounwind
  77   br label %sw.epilog
  78
  79 sw.bb5:
  80   call void @func8(i32 %N) nounwind
  81   br label %sw.epilog
  82
  83 sw.epilog:
  84   ret void
  85 }
  86
  87 !0 = metadata !{metadata !"branch_weights", i32 64, i32 4}
  88 !1 = metadata !{metadata !"branch_weights", i32 4, i32 64}
  89 ; CHECK: !0 = metadata !{metadata !"branch_weights", i32 256, i32 4352, i32 16}
  90 !2 = metadata !{metadata !"branch_weights", i32 4, i32 4, i32 8}
  91 !3 = metadata !{metadata !"branch_weights", i32 8, i32 8, i32 4}
  92 ; CHECK: !1 = metadata !{metadata !"branch_weights", i32 32, i32 48, i32 96, i32 16}