test/CodeGen/AArch64/tbz-tbnz.ll

   1 ; RUN: llc -O1 -march=aarch64 < %s | FileCheck %s
   2
   3 declare void @t()
   4
   5 define void @test1(i32 %a) {
   6 ; CHECK-LABEL: @test1
   7 entry:
   8   %sub = add nsw i32 %a, -12
   9   %cmp = icmp slt i32 %sub, 0
  10   br i1 %cmp, label %if.then, label %if.end
  11
  12 ; CHECK: sub [[CMP:w[0-9]+]], w0, #12
  13 ; CHECK: tbz [[CMP]], #31
  14
  15 if.then:
  16   call void @t()
  17   br label %if.end
  18
  19 if.end:
  20   ret void
  21 }
  22
  23 define void @test2(i64 %a) {
  24 ; CHECK-LABEL: @test2
  25 entry:
  26   %sub = add nsw i64 %a, -12
  27   %cmp = icmp slt i64 %sub, 0
  28   br i1 %cmp, label %if.then, label %if.end
  29
  30 ; CHECK: sub [[CMP:x[0-9]+]], x0, #12
  31 ; CHECK: tbz [[CMP]], #63
  32
  33 if.then:
  34   call void @t()
  35   br label %if.end
  36
  37 if.end:
  38   ret void
  39 }
  40
  41 define void @test3(i32 %a) {
  42 ; CHECK-LABEL: @test3
  43 entry:
  44   %sub = add nsw i32 %a, -12
  45   %cmp = icmp sgt i32 %sub, -1
  46   br i1 %cmp, label %if.then, label %if.end
  47
  48 ; CHECK: sub [[CMP:w[0-9]+]], w0, #12
  49 ; CHECK: tbnz [[CMP]], #31
  50
  51 if.then:
  52   call void @t()
  53   br label %if.end
  54
  55 if.end:
  56   ret void
  57 }
  58
  59 define void @test4(i64 %a) {
  60 ; CHECK-LABEL: @test4
  61 entry:
  62   %sub = add nsw i64 %a, -12
  63   %cmp = icmp sgt i64 %sub, -1
  64   br i1 %cmp, label %if.then, label %if.end
  65
  66 ; CHECK: sub [[CMP:x[0-9]+]], x0, #12
  67 ; CHECK: tbnz [[CMP]], #63
  68
  69 if.then:
  70   call void @t()
  71   br label %if.end
  72
  73 if.end:
  74   ret void
  75 }
  76
  77 define void @test5(i32 %a) {
  78 ; CHECK-LABEL: @test5
  79 entry:
  80   %sub = add nsw i32 %a, -12
  81   %cmp = icmp sge i32 %sub, 0
  82   br i1 %cmp, label %if.then, label %if.end
  83
  84 ; CHECK: sub [[CMP:w[0-9]+]], w0, #12
  85 ; CHECK: tbnz [[CMP]], #31
  86
  87 if.then:
  88   call void @t()
  89   br label %if.end
  90
  91 if.end:
  92   ret void
  93 }
  94
  95 define void @test6(i64 %a) {
  96 ; CHECK-LABEL: @test6
  97 entry:
  98   %sub = add nsw i64 %a, -12
  99   %cmp = icmp sge i64 %sub, 0
 100   br i1 %cmp, label %if.then, label %if.end
 101
 102 ; CHECK: sub [[CMP:x[0-9]+]], x0, #12
 103 ; CHECK: tbnz [[CMP]], #63
 104
 105 if.then:
 106   call void @t()
 107   br label %if.end
 108
 109 if.end:
 110   ret void
 111 }
 112
 113 define void @test7(i32 %a) {
 114 ; CHECK-LABEL: @test7
 115 entry:
 116   %sub = sub nsw i32 %a, 12
 117   %cmp = icmp slt i32 %sub, 0
 118   br i1 %cmp, label %if.then, label %if.end
 119
 120 ; CHECK: sub [[CMP:w[0-9]+]], w0, #12
 121 ; CHECK: tbz [[CMP]], #31
 122
 123 if.then:
 124   call void @t()
 125   br label %if.end
 126
 127 if.end:
 128   ret void
 129 }
 130
 131 define void @test8(i64 %val1, i64 %val2, i64 %val3) {
 132 ; CHECK-LABEL: @test8
 133   %and1 = and i64 %val1, %val2
 134   %tst1 = icmp slt i64 %and1, 0
 135   br i1 %tst1, label %if.then1, label %if.end
 136
 137 ; CHECK: tst x0, x1
 138 ; CHECK-NEXT: b.ge
 139
 140 if.then1:
 141   %and2 = and i64 %val2, %val3
 142   %tst2 = icmp sge i64 %and2, 0
 143   br i1 %tst2, label %if.then2, label %if.end
 144
 145 ; CHECK: and [[CMP:x[0-9]+]], x1, x2
 146 ; CHECK-NOT: cmp
 147 ; CHECK: tbnz [[CMP]], #63
 148
 149 if.then2:
 150   %shifted_op1 = shl i64 %val2, 63
 151   %shifted_and1 = and i64 %val1, %shifted_op1
 152   %tst3 = icmp slt i64 %shifted_and1, 0
 153   br i1 %tst3, label %if.then3, label %if.end
 154
 155 ; CHECK: tst x0, x1, lsl #63
 156 ; CHECK: b.ge
 157
 158 if.then3:
 159   %shifted_op2 = shl i64 %val2, 62
 160   %shifted_and2 = and i64 %val1, %shifted_op2
 161   %tst4 = icmp sge i64 %shifted_and2, 0
 162   br i1 %tst4, label %if.then4, label %if.end
 163
 164 ; CHECK: tst x0, x1, lsl #62
 165 ; CHECK: b.lt
 166
 167 if.then4:
 168   call void @t()
 169   br label %if.end
 170
 171 if.end:
 172   ret void
 173 }
 174
 175 define void @test9(i64 %val1) {
 176 ; CHECK-LABEL: @test9
 177   %tst = icmp slt i64 %val1, 0
 178   br i1 %tst, label %if.then, label %if.end
 179
 180 ; CHECK-NOT: cmp
 181 ; CHECK: tbz x0, #63
 182
 183 if.then:
 184   call void @t()
 185   br label %if.end
 186
 187 if.end:
 188   ret void
 189 }
 190
 191 define void @test10(i64 %val1) {
 192 ; CHECK-LABEL: @test10
 193   %tst = icmp slt i64 %val1, 0
 194   br i1 %tst, label %if.then, label %if.end
 195
 196 ; CHECK-NOT: cmp
 197 ; CHECK: tbz x0, #63
 198
 199 if.then:
 200   call void @t()
 201   br label %if.end
 202
 203 if.end:
 204   ret void
 205 }
 206
 207 define void @test11(i64 %val1, i64* %ptr) {
 208 ; CHECK-LABEL: @test11
 209
 210 ; CHECK: ldr [[CMP:x[0-9]+]], [x1]
 211 ; CHECK-NOT: cmp
 212 ; CHECK: tbz [[CMP]], #63
 213
 214   %val = load i64* %ptr
 215   %tst = icmp slt i64 %val, 0
 216   br i1 %tst, label %if.then, label %if.end
 217
 218 if.then:
 219   call void @t()
 220   br label %if.end
 221
 222 if.end:
 223   ret void
 224 }
 225
 226 define void @test12(i64 %val1) {
 227 ; CHECK-LABEL: @test12
 228   %tst = icmp slt i64 %val1, 0
 229   br i1 %tst, label %if.then, label %if.end
 230
 231 ; CHECK-NOT: cmp
 232 ; CHECK: tbz x0, #63
 233
 234 if.then:
 235   call void @t()
 236   br label %if.end
 237
 238 if.end:
 239   ret void
 240 }
 241
 242 define void @test13(i64 %val1, i64 %val2) {
 243 ; CHECK-LABEL: @test13
 244   %or = or i64 %val1, %val2
 245   %tst = icmp slt i64 %or, 0
 246   br i1 %tst, label %if.then, label %if.end
 247
 248 ; CHECK: orr [[CMP:x[0-9]+]], x0, x1
 249 ; CHECK-NOT: cmp
 250 ; CHECK: tbz [[CMP]], #63
 251
 252 if.then:
 253   call void @t()
 254   br label %if.end
 255
 256 if.end:
 257   ret void
 258 }