llvm/test/Transforms/CorrelatedValuePropagation/profmd.ll

   1 ; RUN: opt < %s -passes=correlated-propagation -S | FileCheck %s
   2
   3 ; Removed several cases from switch.
   4 define i32 @switch1(i32 %s) {
   5 ; CHECK-LABEL: @switch1(
   6 ; CHECK-NEXT:  entry:
   7 ; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[S:%.*]], 0
   8 ; CHECK-NEXT:    br i1 [[CMP]], label [[NEGATIVE:%.*]], label [[OUT:%.*]]
   9 ;
  10 entry:
  11   %cmp = icmp slt i32 %s, 0
  12   br i1 %cmp, label %negative, label %out
  13
  14 negative:
  15 ; CHECK:       negative:
  16 ; CHECK-NEXT:    switch i32 [[S]], label [[OUT]] [
  17 ; CHECK-NEXT:    i32 -2, label [[NEXT:%.*]]
  18 ; CHECK-NEXT:    i32 -1, label [[NEXT]]
  19   switch i32 %s, label %out [
  20   i32 0, label %out
  21   i32 1, label %out
  22   i32 -1, label %next
  23   i32 -2, label %next
  24   i32 2, label %out
  25   i32 3, label %out
  26 ; CHECK-NEXT: !prof ![[MD0:[0-9]+]]
  27   ], !prof !{!"branch_weights", i32 99, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6}
  28
  29 out:
  30   %p = phi i32 [ 1, %entry ], [ -1, %negative ], [ -1, %negative ], [ -1, %negative ], [ -1, %negative ], [ -1, %negative ]
  31   ret i32 %p
  32
  33 next:
  34   %q = phi i32 [ 0, %negative ], [ 0, %negative ]
  35   ret i32 %q
  36 }
  37
  38 ; Removed all cases from switch.
  39 define i32 @switch2(i32 %s) {
  40 ; CHECK-LABEL: @switch2(
  41 ; CHECK-NEXT:  entry:
  42 ; CHECK-NEXT:    [[CMP:%.*]] = icmp sgt i32 [[S:%.*]], 0
  43 ; CHECK-NEXT:    br i1 [[CMP]], label [[POSITIVE:%.*]], label [[OUT:%.*]]
  44 ;
  45 entry:
  46   %cmp = icmp sgt i32 %s, 0
  47   br i1 %cmp, label %positive, label %out
  48
  49 positive:
  50   switch i32 %s, label %out [
  51   i32 0, label %out
  52   i32 -1, label %next
  53   i32 -2, label %next
  54   ], !prof !{!"branch_weights", i32 99, i32 1, i32 2, i32 3}
  55
  56 out:
  57   %p = phi i32 [ -1, %entry ], [ 1, %positive ], [ 1, %positive ]
  58   ret i32 %p
  59
  60 next:
  61   %q = phi i32 [ 0, %positive ], [ 0, %positive ]
  62   ret i32 %q
  63 }
  64
  65 ; Change switch into conditional branch.
  66 define i32 @switch3(i32 %s) {
  67 ; CHECK-LABEL: @switch3(
  68 ;
  69 entry:
  70   %cmp = icmp sgt i32 %s, 0
  71   br i1 %cmp, label %positive, label %out
  72
  73 positive:
  74 ; CHECK:      positive:
  75 ; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 %s, 1
  76 ; CHECK-NEXT:    br i1 [[CMP]], label [[NEXT:%.*]], label [[OUT:%.*]], !prof ![[MD1:[0-9]+]]
  77   switch i32 %s, label %out [
  78   i32 1, label %next
  79   i32 -1, label %next
  80   i32 -2, label %next
  81   ], !prof !{!"branch_weights", i32 99, i32 1, i32 2, i32 3}
  82
  83 out:
  84   %p = phi i32 [ -1, %entry ], [ 1, %positive ]
  85   ret i32 %p
  86
  87 next:
  88   %q = phi i32 [ 0, %positive ], [ 0, %positive ], [ 0, %positive ]
  89   ret i32 %q
  90 }
  91
  92 ; Removed all cases from switch.
  93 define i32 @switch4(i32 %s) {
  94 ; CHECK-LABEL: @switch4(
  95 ; CHECK-NEXT:  entry:
  96 ; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[S:%.*]], 0
  97 ; CHECK-NEXT:    br i1 [[CMP]], label [[NEGATIVE:%.*]], label [[OUT:%.*]]
  98 ;
  99 entry:
 100   %cmp = icmp slt i32 %s, 0
 101   br i1 %cmp, label %negative, label %out
 102
 103 negative:
 104 ; CHECK:       negative:
 105 ; CHECK-NEXT:    br label %out
 106   switch i32 %s, label %out [
 107   i32 0, label %out
 108   i32 1, label %out
 109   i32 2, label %out
 110   i32 3, label %out
 111   ], !prof !{!"branch_weights", i32 99, i32 1, i32 2, i32 3, i32 4}
 112
 113 out:
 114   %p = phi i32 [ 1, %entry ], [ -1, %negative ], [ -1, %negative ], [ -1, %negative ], [ -1, %negative ], [ -1, %negative ]
 115   ret i32 %p
 116 }
 117
 118 ; CHECK: ![[MD0]] = !{!"branch_weights", i32 99, i32 4, i32 3}
 119 ; CHECK: ![[MD1]] = !{!"branch_weights", i32 1, i32 99}