[SystemZ] Use A(G)SI when spilling the target of a constant addition
[oota-llvm.git] / test / CodeGen / SystemZ / atomicrmw-minmax-03.ll
1 ; Test 32-bit atomic minimum and maximum.
2 ;
3 ; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
4
5 ; Check signed minium.
6 define i32 @f1(i32 %dummy, i32 *%src, i32 %b) {
7 ; CHECK-LABEL: f1:
8 ; CHECK: l %r2, 0(%r3)
9 ; CHECK: [[LOOP:\.[^:]*]]:
10 ; CHECK: lr [[NEW:%r[0-9]+]], %r2
11 ; CHECK: crjle %r2, %r4, [[KEEP:\..*]]
12 ; CHECK: lr [[NEW]], %r4
13 ; CHECK: cs %r2, [[NEW]], 0(%r3)
14 ; CHECK: jl [[LOOP]]
15 ; CHECK: br %r14
16   %res = atomicrmw min i32 *%src, i32 %b seq_cst
17   ret i32 %res
18 }
19
20 ; Check signed maximum.
21 define i32 @f2(i32 %dummy, i32 *%src, i32 %b) {
22 ; CHECK-LABEL: f2:
23 ; CHECK: l %r2, 0(%r3)
24 ; CHECK: [[LOOP:\.[^:]*]]:
25 ; CHECK: lr [[NEW:%r[0-9]+]], %r2
26 ; CHECK: crjhe %r2, %r4, [[KEEP:\..*]]
27 ; CHECK: lr [[NEW]], %r4
28 ; CHECK: cs %r2, [[NEW]], 0(%r3)
29 ; CHECK: jl [[LOOP]]
30 ; CHECK: br %r14
31   %res = atomicrmw max i32 *%src, i32 %b seq_cst
32   ret i32 %res
33 }
34
35 ; Check unsigned minimum.
36 define i32 @f3(i32 %dummy, i32 *%src, i32 %b) {
37 ; CHECK-LABEL: f3:
38 ; CHECK: l %r2, 0(%r3)
39 ; CHECK: [[LOOP:\.[^:]*]]:
40 ; CHECK: lr [[NEW:%r[0-9]+]], %r2
41 ; CHECK: clrjle %r2, %r4, [[KEEP:\..*]]
42 ; CHECK: lr [[NEW]], %r4
43 ; CHECK: cs %r2, [[NEW]], 0(%r3)
44 ; CHECK: jl [[LOOP]]
45 ; CHECK: br %r14
46   %res = atomicrmw umin i32 *%src, i32 %b seq_cst
47   ret i32 %res
48 }
49
50 ; Check unsigned maximum.
51 define i32 @f4(i32 %dummy, i32 *%src, i32 %b) {
52 ; CHECK-LABEL: f4:
53 ; CHECK: l %r2, 0(%r3)
54 ; CHECK: [[LOOP:\.[^:]*]]:
55 ; CHECK: lr [[NEW:%r[0-9]+]], %r2
56 ; CHECK: clrjhe %r2, %r4, [[KEEP:\..*]]
57 ; CHECK: lr [[NEW]], %r4
58 ; CHECK: cs %r2, [[NEW]], 0(%r3)
59 ; CHECK: jl [[LOOP]]
60 ; CHECK: br %r14
61   %res = atomicrmw umax i32 *%src, i32 %b seq_cst
62   ret i32 %res
63 }
64
65 ; Check the high end of the aligned CS range.
66 define i32 @f5(i32 %dummy, i32 *%src, i32 %b) {
67 ; CHECK-LABEL: f5:
68 ; CHECK: l %r2, 4092(%r3)
69 ; CHECK: cs %r2, {{%r[0-9]+}}, 4092(%r3)
70 ; CHECK: br %r14
71   %ptr = getelementptr i32 *%src, i64 1023
72   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
73   ret i32 %res
74 }
75
76 ; Check the next word up, which requires CSY.
77 define i32 @f6(i32 %dummy, i32 *%src, i32 %b) {
78 ; CHECK-LABEL: f6:
79 ; CHECK: ly %r2, 4096(%r3)
80 ; CHECK: csy %r2, {{%r[0-9]+}}, 4096(%r3)
81 ; CHECK: br %r14
82   %ptr = getelementptr i32 *%src, i64 1024
83   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
84   ret i32 %res
85 }
86
87 ; Check the high end of the aligned CSY range.
88 define i32 @f7(i32 %dummy, i32 *%src, i32 %b) {
89 ; CHECK-LABEL: f7:
90 ; CHECK: ly %r2, 524284(%r3)
91 ; CHECK: csy %r2, {{%r[0-9]+}}, 524284(%r3)
92 ; CHECK: br %r14
93   %ptr = getelementptr i32 *%src, i64 131071
94   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
95   ret i32 %res
96 }
97
98 ; Check the next word up, which needs separate address logic.
99 define i32 @f8(i32 %dummy, i32 *%src, i32 %b) {
100 ; CHECK-LABEL: f8:
101 ; CHECK: agfi %r3, 524288
102 ; CHECK: l %r2, 0(%r3)
103 ; CHECK: cs %r2, {{%r[0-9]+}}, 0(%r3)
104 ; CHECK: br %r14
105   %ptr = getelementptr i32 *%src, i64 131072
106   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
107   ret i32 %res
108 }
109
110 ; Check the high end of the negative aligned CSY range.
111 define i32 @f9(i32 %dummy, i32 *%src, i32 %b) {
112 ; CHECK-LABEL: f9:
113 ; CHECK: ly %r2, -4(%r3)
114 ; CHECK: csy %r2, {{%r[0-9]+}}, -4(%r3)
115 ; CHECK: br %r14
116   %ptr = getelementptr i32 *%src, i64 -1
117   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
118   ret i32 %res
119 }
120
121 ; Check the low end of the CSY range.
122 define i32 @f10(i32 %dummy, i32 *%src, i32 %b) {
123 ; CHECK-LABEL: f10:
124 ; CHECK: ly %r2, -524288(%r3)
125 ; CHECK: csy %r2, {{%r[0-9]+}}, -524288(%r3)
126 ; CHECK: br %r14
127   %ptr = getelementptr i32 *%src, i64 -131072
128   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
129   ret i32 %res
130 }
131
132 ; Check the next word down, which needs separate address logic.
133 define i32 @f11(i32 %dummy, i32 *%src, i32 %b) {
134 ; CHECK-LABEL: f11:
135 ; CHECK: agfi %r3, -524292
136 ; CHECK: l %r2, 0(%r3)
137 ; CHECK: cs %r2, {{%r[0-9]+}}, 0(%r3)
138 ; CHECK: br %r14
139   %ptr = getelementptr i32 *%src, i64 -131073
140   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
141   ret i32 %res
142 }
143
144 ; Check that indexed addresses are not allowed.
145 define i32 @f12(i32 %dummy, i64 %base, i64 %index, i32 %b) {
146 ; CHECK-LABEL: f12:
147 ; CHECK: agr %r3, %r4
148 ; CHECK: l %r2, 0(%r3)
149 ; CHECK: cs %r2, {{%r[0-9]+}}, 0(%r3)
150 ; CHECK: br %r14
151   %add = add i64 %base, %index
152   %ptr = inttoptr i64 %add to i32 *
153   %res = atomicrmw min i32 *%ptr, i32 %b seq_cst
154   ret i32 %res
155 }
156
157 ; Check that constants are handled.
158 define i32 @f13(i32 %dummy, i32 *%ptr) {
159 ; CHECK-LABEL: f13:
160 ; CHECK: lhi [[LIMIT:%r[0-9]+]], 42
161 ; CHECK: l %r2, 0(%r3)
162 ; CHECK: [[LOOP:\.[^:]*]]:
163 ; CHECK: lr [[NEW:%r[0-9]+]], %r2
164 ; CHECK: crjle %r2, [[LIMIT]], [[KEEP:\..*]]
165 ; CHECK: lhi [[NEW]], 42
166 ; CHECK: cs %r2, [[NEW]], 0(%r3)
167 ; CHECK: jl [[LOOP]]
168 ; CHECK: br %r14
169   %res = atomicrmw min i32 *%ptr, i32 42 seq_cst
170   ret i32 %res
171 }