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.
6define 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.
17define 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.
28define 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.
39define 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.
51define 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.
64define 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.
78define 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.
94define 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.
108define 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.
122define 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.
136define 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.
150define 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.
164define 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.
178define 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