test/CodeGen/ARM/arm-and-tst-peephole.ll

   1 ; RUN: llc -mtriple=arm-eabi -arm-atomic-cfg-tidy=0 %s -o - | FileCheck -check-prefix=ARM %s
   2 ; RUN: llc -mtriple=thumb-eabi -arm-atomic-cfg-tidy=0 %s -o - | FileCheck -check-prefix=THUMB %s
   3 ; RUN: llc -mtriple=thumb-eabi -arm-atomic-cfg-tidy=0 -mcpu=arm1156t2-s -mattr=+thumb2 %s -o - | FileCheck -check-prefix=T2 %s
   4 ; RUN: llc -mtriple=thumbv8-eabi -arm-atomic-cfg-tidy=0 %s -o - | FileCheck -check-prefix=V8 %s
   5
   6 ; FIXME: The -mtriple=thumb test doesn't change if -disable-peephole is specified.
   7
   8 %struct.Foo = type { i8* }
   9
  10 ; ARM-LABEL:   foo:
  11 ; THUMB-LABEL: foo:
  12 ; T2-LABEL:    foo:
  13 define %struct.Foo* @foo(%struct.Foo* %this, i32 %acc) nounwind readonly align 2 {
  14 entry:
  15   %scevgep = getelementptr %struct.Foo, %struct.Foo* %this, i32 1
  16   br label %tailrecurse
  17
  18 tailrecurse:                                      ; preds = %sw.bb, %entry
  19   %lsr.iv2 = phi %struct.Foo* [ %scevgep3, %sw.bb ], [ %scevgep, %entry ]
  20   %lsr.iv = phi i32 [ %lsr.iv.next, %sw.bb ], [ 1, %entry ]
  21   %acc.tr = phi i32 [ %or, %sw.bb ], [ %acc, %entry ]
  22   %lsr.iv24 = bitcast %struct.Foo* %lsr.iv2 to i8**
  23   %scevgep5 = getelementptr i8*, i8** %lsr.iv24, i32 -1
  24   %tmp2 = load i8*, i8** %scevgep5
  25   %0 = ptrtoint i8* %tmp2 to i32
  26
  27 ; ARM:      ands {{r[0-9]+}}, {{r[0-9]+}}, #3
  28 ; ARM-NEXT: beq
  29
  30 ; THUMB:      movs r[[R0:[0-9]+]], #3
  31 ; THUMB-NEXT: ands r[[R0]], r
  32 ; THUMB-NEXT: cmp r[[R0]], #0
  33 ; THUMB-NEXT: beq
  34
  35 ; T2:      ands {{r[0-9]+}}, {{r[0-9]+}}, #3
  36 ; T2-NEXT: beq
  37
  38   %and = and i32 %0, 3
  39   %tst = icmp eq i32 %and, 0
  40   br i1 %tst, label %sw.bb, label %tailrecurse.switch
  41
  42 tailrecurse.switch:                               ; preds = %tailrecurse
  43 ; V8-LABEL: %tailrecurse.switch
  44 ; V8: cmp
  45 ; V8-NEXT: beq
  46 ; V8-NEXT: %tailrecurse.switch
  47 ; V8: cmp
  48 ; V8-NEXT: beq
  49 ; V8-NEXT: %tailrecurse.switch
  50 ; V8: cmp
  51 ; V8-NEXT: beq
  52 ; V8-NEXT: %sw.epilog
  53 ; V8-NEXT: bx lr
  54   switch i32 %and, label %sw.epilog [
  55     i32 1, label %sw.bb
  56     i32 3, label %sw.bb6
  57     i32 2, label %sw.bb8
  58   ], !prof !1
  59
  60 sw.bb:                                            ; preds = %tailrecurse.switch, %tailrecurse
  61   %shl = shl i32 %acc.tr, 1
  62   %or = or i32 %and, %shl
  63   %lsr.iv.next = add i32 %lsr.iv, 1
  64   %scevgep3 = getelementptr %struct.Foo, %struct.Foo* %lsr.iv2, i32 1
  65   br label %tailrecurse
  66
  67 sw.bb6:                                           ; preds = %tailrecurse.switch
  68   ret %struct.Foo* %lsr.iv2
  69
  70 sw.bb8:                                           ; preds = %tailrecurse.switch
  71   %tmp1 = add i32 %acc.tr, %lsr.iv
  72   %add.ptr11 = getelementptr inbounds %struct.Foo, %struct.Foo* %this, i32 %tmp1
  73   ret %struct.Foo* %add.ptr11
  74
  75 sw.epilog:                                        ; preds = %tailrecurse.switch
  76   ret %struct.Foo* undef
  77 }
  78
  79 ; Another test that exercises the AND/TST peephole optimization and also
  80 ; generates a predicated ANDS instruction. Check that the predicate is printed
  81 ; after the "S" modifier on the instruction.
  82
  83 %struct.S = type { i8* (i8*)*, [1 x i8] }
  84
  85 ; ARM-LABEL: bar:
  86 ; THUMB-LABEL: bar:
  87 ; T2-LABEL: bar:
  88 ; V8-LABEL: bar:
  89 define internal zeroext i8 @bar(%struct.S* %x, %struct.S* nocapture %y) nounwind readonly {
  90 entry:
  91   %0 = getelementptr inbounds %struct.S, %struct.S* %x, i32 0, i32 1, i32 0
  92   %1 = load i8, i8* %0, align 1
  93   %2 = zext i8 %1 to i32
  94 ; ARM: ands
  95 ; THUMB: ands
  96 ; T2: ands
  97 ; V8: ands
  98 ; V8-NEXT: beq
  99   %3 = and i32 %2, 112
 100   %4 = icmp eq i32 %3, 0
 101   br i1 %4, label %return, label %bb
 102
 103 bb:                                               ; preds = %entry
 104 ; V8-NEXT: %bb
 105   %5 = getelementptr inbounds %struct.S, %struct.S* %y, i32 0, i32 1, i32 0
 106   %6 = load i8, i8* %5, align 1
 107   %7 = zext i8 %6 to i32
 108 ; ARM: andsne
 109 ; THUMB: ands
 110 ; T2: andsne
 111 ; V8: ands
 112 ; V8-NEXT: beq
 113   %8 = and i32 %7, 112
 114   %9 = icmp eq i32 %8, 0
 115   br i1 %9, label %return, label %bb2
 116
 117 bb2:                                              ; preds = %bb
 118 ; V8-NEXT: %bb2
 119 ; V8-NEXT: cmp
 120 ; V8-NEXT: it   ne
 121 ; V8-NEXT: cmpne
 122 ; V8-NEXT: bne
 123   %10 = icmp eq i32 %3, 16
 124   %11 = icmp eq i32 %8, 16
 125   %or.cond = or i1 %10, %11
 126   br i1 %or.cond, label %bb4, label %return
 127
 128 bb4:                                              ; preds = %bb2
 129   %12 = ptrtoint %struct.S* %x to i32
 130   %phitmp = trunc i32 %12 to i8
 131   ret i8 %phitmp
 132
 133 return:                                           ; preds = %bb2, %bb, %entry
 134   ret i8 1
 135 }
 136
 137
 138 ; We were looking through multiple COPY instructions to find an AND we might
 139 ; fold into a TST, but in doing so we changed the register being tested allowing
 140 ; folding of unrelated tests (in this case, a TST against r1 was eliminated in
 141 ; favour of an AND of r0).
 142
 143 define i32 @test_tst_assessment(i32 %a, i32 %b) {
 144 ; ARM-LABEL: test_tst_assessment:
 145 ; ARM:       @ %bb.0:
 146 ; ARM-NEXT:    and r0, r0, #1
 147 ; ARM-NEXT:    tst r1, #1
 148 ; ARM-NEXT:    subne r0, r0, #1
 149 ; ARM-NEXT:    mov pc, lr
 150 ;
 151 ; THUMB-LABEL: test_tst_assessment:
 152 ; THUMB:       @ %bb.0:
 153 ; THUMB-NEXT:    movs r2, r0
 154 ; THUMB-NEXT:    movs r0, #1
 155 ; THUMB-NEXT:    ands r0, r2
 156 ; THUMB-NEXT:    lsls r1, r1, #31
 157 ; THUMB-NEXT:    beq .LBB2_2
 158 ; THUMB-NEXT:  @ %bb.1:
 159 ; THUMB-NEXT:    subs r0, r0, #1
 160 ; THUMB-NEXT:  .LBB2_2:
 161 ; THUMB-NEXT:    bx lr
 162 ;
 163 ; T2-LABEL: test_tst_assessment:
 164 ; T2:       @ %bb.0:
 165 ; T2-NEXT:    and r0, r0, #1
 166 ; T2-NEXT:    lsls r1, r1, #31
 167 ; T2-NEXT:    it ne
 168 ; T2-NEXT:    subne r0, #1
 169 ; T2-NEXT:    bx lr
 170 ;
 171 ; V8-LABEL: test_tst_assessment:
 172 ; V8:       @ %bb.0:
 173 ; V8-NEXT:    and r0, r0, #1
 174 ; V8-NEXT:    lsls r1, r1, #31
 175 ; V8-NEXT:    it ne
 176 ; V8-NEXT:    subne r0, #1
 177 ; V8-NEXT:    bx lr
 178   %and1 = and i32 %a, 1
 179   %sub = sub i32 %and1, 1
 180   %and2 = and i32 %b, 1
 181   %cmp = icmp eq i32 %and2, 0
 182   %sel = select i1 %cmp, i32 %and1, i32 %sub
 183   ret i32 %sel
 184 }
 185
 186 !1 = !{!"branch_weights", i32 1, i32 1, i32 3, i32 2 }