1 ; Test 32-bit rotates left. 2 ; 3 ; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s 4 5 ; Check the low end of the RLL range. 6 define i32 @f1(i32 %a) { 7 ; CHECK-LABEL: f1: 8 ; CHECK: rll %r2, %r2, 1 9 ; CHECK: br %r14 10 %parta = shl i32 %a, 1 11 %partb = lshr i32 %a, 31 12 %or = or i32 %parta, %partb 13 ret i32 %or 14 } 15 16 ; Check the high end of the defined RLL range. 17 define i32 @f2(i32 %a) { 18 ; CHECK-LABEL: f2: 19 ; CHECK: rll %r2, %r2, 31 20 ; CHECK: br %r14 21 %parta = shl i32 %a, 31 22 %partb = lshr i32 %a, 1 23 %or = or i32 %parta, %partb 24 ret i32 %or 25 } 26 27 ; We don't generate shifts by out-of-range values. 28 define i32 @f3(i32 %a) { 29 ; CHECK-LABEL: f3: 30 ; CHECK-NOT: rll 31 ; CHECK: br %r14 32 %parta = shl i32 %a, 32 33 %partb = lshr i32 %a, 0 34 %or = or i32 %parta, %partb 35 ret i32 %or 36 } 37 38 ; Check variable shifts. 39 define i32 @f4(i32 %a, i32 %amt) { 40 ; CHECK-LABEL: f4: 41 ; CHECK: rll %r2, %r2, 0(%r3) 42 ; CHECK: br %r14 43 %amtb = sub i32 32, %amt 44 %parta = shl i32 %a, %amt 45 %partb = lshr i32 %a, %amtb 46 %or = or i32 %parta, %partb 47 ret i32 %or 48 } 49 50 ; Check shift amounts that have a constant term. 51 define i32 @f5(i32 %a, i32 %amt) { 52 ; CHECK-LABEL: f5: 53 ; CHECK: rll %r2, %r2, 10(%r3) 54 ; CHECK: br %r14 55 %add = add i32 %amt, 10 56 %sub = sub i32 32, %add 57 %parta = shl i32 %a, %add 58 %partb = lshr i32 %a, %sub 59 %or = or i32 %parta, %partb 60 ret i32 %or 61 } 62 63 ; ...and again with a truncated 64-bit shift amount. 64 define i32 @f6(i32 %a, i64 %amt) { 65 ; CHECK-LABEL: f6: 66 ; CHECK: rll %r2, %r2, 10(%r3) 67 ; CHECK: br %r14 68 %add = add i64 %amt, 10 69 %addtrunc = trunc i64 %add to i32 70 %sub = sub i32 32, %addtrunc 71 %parta = shl i32 %a, %addtrunc 72 %partb = lshr i32 %a, %sub 73 %or = or i32 %parta, %partb 74 ret i32 %or 75 } 76 77 ; ...and again with a different truncation representation. 78 define i32 @f7(i32 %a, i64 %amt) { 79 ; CHECK-LABEL: f7: 80 ; CHECK: rll %r2, %r2, 10(%r3) 81 ; CHECK: br %r14 82 %add = add i64 %amt, 10 83 %sub = sub i64 32, %add 84 %addtrunc = trunc i64 %add to i32 85 %subtrunc = trunc i64 %sub to i32 86 %parta = shl i32 %a, %addtrunc 87 %partb = lshr i32 %a, %subtrunc 88 %or = or i32 %parta, %partb 89 ret i32 %or 90 } 91 92 ; Check shift amounts that have the largest in-range constant term. We could 93 ; mask the amount instead. 94 define i32 @f8(i32 %a, i32 %amt) { 95 ; CHECK-LABEL: f8: 96 ; CHECK: rll %r2, %r2, 524287(%r3) 97 ; CHECK: br %r14 98 %add = add i32 %amt, 524287 99 %sub = sub i32 32, %add 100 %parta = shl i32 %a, %add 101 %partb = lshr i32 %a, %sub 102 %or = or i32 %parta, %partb 103 ret i32 %or 104 } 105 106 ; Check the next value up, which without masking must use a separate 107 ; addition. 108 define i32 @f9(i32 %a, i32 %amt) { 109 ; CHECK-LABEL: f9: 110 ; CHECK: afi %r3, 524288 111 ; CHECK: rll %r2, %r2, 0(%r3) 112 ; CHECK: br %r14 113 %add = add i32 %amt, 524288 114 %sub = sub i32 32, %add 115 %parta = shl i32 %a, %add 116 %partb = lshr i32 %a, %sub 117 %or = or i32 %parta, %partb 118 ret i32 %or 119 } 120 121 ; Check cases where 1 is subtracted from the shift amount. 122 define i32 @f10(i32 %a, i32 %amt) { 123 ; CHECK-LABEL: f10: 124 ; CHECK: rll %r2, %r2, -1(%r3) 125 ; CHECK: br %r14 126 %suba = sub i32 %amt, 1 127 %subb = sub i32 32, %suba 128 %parta = shl i32 %a, %suba 129 %partb = lshr i32 %a, %subb 130 %or = or i32 %parta, %partb 131 ret i32 %or 132 } 133 134 ; Check the lowest value that can be subtracted from the shift amount. 135 ; Again, we could mask the shift amount instead. 136 define i32 @f11(i32 %a, i32 %amt) { 137 ; CHECK-LABEL: f11: 138 ; CHECK: rll %r2, %r2, -524288(%r3) 139 ; CHECK: br %r14 140 %suba = sub i32 %amt, 524288 141 %subb = sub i32 32, %suba 142 %parta = shl i32 %a, %suba 143 %partb = lshr i32 %a, %subb 144 %or = or i32 %parta, %partb 145 ret i32 %or 146 } 147 148 ; Check the next value down, which without masking must use a separate 149 ; addition. 150 define i32 @f12(i32 %a, i32 %amt) { 151 ; CHECK-LABEL: f12: 152 ; CHECK: afi %r3, -524289 153 ; CHECK: rll %r2, %r2, 0(%r3) 154 ; CHECK: br %r14 155 %suba = sub i32 %amt, 524289 156 %subb = sub i32 32, %suba 157 %parta = shl i32 %a, %suba 158 %partb = lshr i32 %a, %subb 159 %or = or i32 %parta, %partb 160 ret i32 %or 161 } 162 163 ; Check that we don't try to generate "indexed" shifts. 164 define i32 @f13(i32 %a, i32 %b, i32 %c) { 165 ; CHECK-LABEL: f13: 166 ; CHECK: ar {{%r3, %r4|%r4, %r3}} 167 ; CHECK: rll %r2, %r2, 0({{%r[34]}}) 168 ; CHECK: br %r14 169 %add = add i32 %b, %c 170 %sub = sub i32 32, %add 171 %parta = shl i32 %a, %add 172 %partb = lshr i32 %a, %sub 173 %or = or i32 %parta, %partb 174 ret i32 %or 175 } 176 177 ; Check that the shift amount uses an address register. It cannot be in %r0. 178 define i32 @f14(i32 %a, i32 *%ptr) { 179 ; CHECK-LABEL: f14: 180 ; CHECK: l %r1, 0(%r3) 181 ; CHECK: rll %r2, %r2, 0(%r1) 182 ; CHECK: br %r14 183 %amt = load i32 *%ptr 184 %amtb = sub i32 32, %amt 185 %parta = shl i32 %a, %amt 186 %partb = lshr i32 %a, %amtb 187 %or = or i32 %parta, %partb 188 ret i32 %or 189 } 190
