Skip to content

Commit 4cce1e7

Browse files
davemgreendavemgreen
committed
[AArch64] Add some i128 add/sub/mul with overflow tests. NFC
1 parent 7b96272 commit 4cce1e7

File tree

1 file changed

+343
-0
lines changed

1 file changed

+343
-0
lines changed

‎llvm/test/CodeGen/AArch64/i128_with_overflow.ll

Lines changed: 343 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,343 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
2+
; RUN: llc -mtriple=aarch64 -o - %s | FileCheck %s --check-prefixes=CHECK,CHECK-SD
3+
; RUN: llc -mtriple=aarch64 -global-isel -global-isel-abort=2 -o - %s 2>&1 | FileCheck %s --check-prefixes=CHECK,CHECK-GI
4+
5+
; CHECK-GI: warning: Instruction selection used fallback path for test_umul_i128
6+
; CHECK-GI-NEXT: warning: Instruction selection used fallback path for test_smul_i128
7+
8+
define i128 @test_uadd_i128(i128 noundef %x, i128 noundef %y) {
9+
; CHECK-SD-LABEL: test_uadd_i128:
10+
; CHECK-SD: // %bb.0: // %entry
11+
; CHECK-SD-NEXT: adds x0, x0, x2
12+
; CHECK-SD-NEXT: adcs x1, x1, x3
13+
; CHECK-SD-NEXT: cset w8, hs
14+
; CHECK-SD-NEXT: cmp w8, #1
15+
; CHECK-SD-NEXT: b.ne .LBB0_2
16+
; CHECK-SD-NEXT: // %bb.1: // %if.then
17+
; CHECK-SD-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
18+
; CHECK-SD-NEXT: .cfi_def_cfa_offset 16
19+
; CHECK-SD-NEXT: .cfi_offset w30, -16
20+
; CHECK-SD-NEXT: bl error
21+
; CHECK-SD-NEXT: // kill: def $w0 killed $w0 def $x0
22+
; CHECK-SD-NEXT: sxtw x0, w0
23+
; CHECK-SD-NEXT: asr x1, x0, #63
24+
; CHECK-SD-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
25+
; CHECK-SD-NEXT: .LBB0_2: // %cleanup
26+
; CHECK-SD-NEXT: ret
27+
;
28+
; CHECK-GI-LABEL: test_uadd_i128:
29+
; CHECK-GI: // %bb.0: // %entry
30+
; CHECK-GI-NEXT: adds x0, x0, x2
31+
; CHECK-GI-NEXT: adcs x1, x1, x3
32+
; CHECK-GI-NEXT: b.lo .LBB0_2
33+
; CHECK-GI-NEXT: // %bb.1: // %if.then
34+
; CHECK-GI-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
35+
; CHECK-GI-NEXT: .cfi_def_cfa_offset 16
36+
; CHECK-GI-NEXT: .cfi_offset w30, -16
37+
; CHECK-GI-NEXT: bl error
38+
; CHECK-GI-NEXT: // kill: def $w0 killed $w0 def $x0
39+
; CHECK-GI-NEXT: asr w1, w0, #31
40+
; CHECK-GI-NEXT: bfi x0, x1, #32, #32
41+
; CHECK-GI-NEXT: bfi x1, x1, #32, #32
42+
; CHECK-GI-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
43+
; CHECK-GI-NEXT: .LBB0_2: // %cleanup
44+
; CHECK-GI-NEXT: ret
45+
entry:
46+
%0 = tail call { i128, i1 } @llvm.uadd.with.overflow.i128(i128 %x, i128 %y)
47+
%1 = extractvalue { i128, i1 } %0, 1
48+
br i1 %1, label %if.then, label %if.end
49+
50+
if.then:
51+
%call = tail call i32 @error()
52+
%conv1 = sext i32 %call to i128
53+
br label %cleanup
54+
55+
if.end:
56+
%2 = extractvalue { i128, i1 } %0, 0
57+
br label %cleanup
58+
59+
cleanup:
60+
%retval.0 = phi i128 [ %conv1, %if.then ], [ %2, %if.end ]
61+
ret i128 %retval.0
62+
}
63+
64+
define i128 @test_sadd_i128(i128 noundef %x, i128 noundef %y) {
65+
; CHECK-SD-LABEL: test_sadd_i128:
66+
; CHECK-SD: // %bb.0: // %entry
67+
; CHECK-SD-NEXT: adds x0, x0, x2
68+
; CHECK-SD-NEXT: adcs x1, x1, x3
69+
; CHECK-SD-NEXT: cset w8, vs
70+
; CHECK-SD-NEXT: cmp w8, #1
71+
; CHECK-SD-NEXT: b.ne .LBB1_2
72+
; CHECK-SD-NEXT: // %bb.1: // %if.then
73+
; CHECK-SD-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
74+
; CHECK-SD-NEXT: .cfi_def_cfa_offset 16
75+
; CHECK-SD-NEXT: .cfi_offset w30, -16
76+
; CHECK-SD-NEXT: bl error
77+
; CHECK-SD-NEXT: // kill: def $w0 killed $w0 def $x0
78+
; CHECK-SD-NEXT: sxtw x0, w0
79+
; CHECK-SD-NEXT: asr x1, x0, #63
80+
; CHECK-SD-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
81+
; CHECK-SD-NEXT: .LBB1_2: // %cleanup
82+
; CHECK-SD-NEXT: ret
83+
;
84+
; CHECK-GI-LABEL: test_sadd_i128:
85+
; CHECK-GI: // %bb.0: // %entry
86+
; CHECK-GI-NEXT: adds x0, x0, x2
87+
; CHECK-GI-NEXT: adcs x1, x1, x3
88+
; CHECK-GI-NEXT: b.vc .LBB1_2
89+
; CHECK-GI-NEXT: // %bb.1: // %if.then
90+
; CHECK-GI-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
91+
; CHECK-GI-NEXT: .cfi_def_cfa_offset 16
92+
; CHECK-GI-NEXT: .cfi_offset w30, -16
93+
; CHECK-GI-NEXT: bl error
94+
; CHECK-GI-NEXT: // kill: def $w0 killed $w0 def $x0
95+
; CHECK-GI-NEXT: asr w1, w0, #31
96+
; CHECK-GI-NEXT: bfi x0, x1, #32, #32
97+
; CHECK-GI-NEXT: bfi x1, x1, #32, #32
98+
; CHECK-GI-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
99+
; CHECK-GI-NEXT: .LBB1_2: // %cleanup
100+
; CHECK-GI-NEXT: ret
101+
entry:
102+
%0 = tail call { i128, i1 } @llvm.sadd.with.overflow.i128(i128 %x, i128 %y)
103+
%1 = extractvalue { i128, i1 } %0, 1
104+
br i1 %1, label %if.then, label %if.end
105+
106+
if.then:
107+
%call = tail call i32 @error()
108+
%conv1 = sext i32 %call to i128
109+
br label %cleanup
110+
111+
if.end:
112+
%2 = extractvalue { i128, i1 } %0, 0
113+
br label %cleanup
114+
115+
cleanup:
116+
%retval.0 = phi i128 [ %conv1, %if.then ], [ %2, %if.end ]
117+
ret i128 %retval.0
118+
}
119+
120+
define i128 @test_usub_i128(i128 noundef %x, i128 noundef %y) {
121+
; CHECK-SD-LABEL: test_usub_i128:
122+
; CHECK-SD: // %bb.0: // %entry
123+
; CHECK-SD-NEXT: subs x0, x0, x2
124+
; CHECK-SD-NEXT: sbcs x1, x1, x3
125+
; CHECK-SD-NEXT: cset w8, lo
126+
; CHECK-SD-NEXT: cmp w8, #1
127+
; CHECK-SD-NEXT: b.ne .LBB2_2
128+
; CHECK-SD-NEXT: // %bb.1: // %if.then
129+
; CHECK-SD-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
130+
; CHECK-SD-NEXT: .cfi_def_cfa_offset 16
131+
; CHECK-SD-NEXT: .cfi_offset w30, -16
132+
; CHECK-SD-NEXT: bl error
133+
; CHECK-SD-NEXT: // kill: def $w0 killed $w0 def $x0
134+
; CHECK-SD-NEXT: sxtw x0, w0
135+
; CHECK-SD-NEXT: asr x1, x0, #63
136+
; CHECK-SD-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
137+
; CHECK-SD-NEXT: .LBB2_2: // %cleanup
138+
; CHECK-SD-NEXT: ret
139+
;
140+
; CHECK-GI-LABEL: test_usub_i128:
141+
; CHECK-GI: // %bb.0: // %entry
142+
; CHECK-GI-NEXT: subs x0, x0, x2
143+
; CHECK-GI-NEXT: sbcs x1, x1, x3
144+
; CHECK-GI-NEXT: b.hs .LBB2_2
145+
; CHECK-GI-NEXT: // %bb.1: // %if.then
146+
; CHECK-GI-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
147+
; CHECK-GI-NEXT: .cfi_def_cfa_offset 16
148+
; CHECK-GI-NEXT: .cfi_offset w30, -16
149+
; CHECK-GI-NEXT: bl error
150+
; CHECK-GI-NEXT: // kill: def $w0 killed $w0 def $x0
151+
; CHECK-GI-NEXT: asr w1, w0, #31
152+
; CHECK-GI-NEXT: bfi x0, x1, #32, #32
153+
; CHECK-GI-NEXT: bfi x1, x1, #32, #32
154+
; CHECK-GI-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
155+
; CHECK-GI-NEXT: .LBB2_2: // %cleanup
156+
; CHECK-GI-NEXT: ret
157+
entry:
158+
%0 = tail call { i128, i1 } @llvm.usub.with.overflow.i128(i128 %x, i128 %y)
159+
%1 = extractvalue { i128, i1 } %0, 1
160+
br i1 %1, label %if.then, label %if.end
161+
162+
if.then:
163+
%call = tail call i32 @error()
164+
%conv1 = sext i32 %call to i128
165+
br label %cleanup
166+
167+
if.end:
168+
%2 = extractvalue { i128, i1 } %0, 0
169+
br label %cleanup
170+
171+
cleanup:
172+
%retval.0 = phi i128 [ %conv1, %if.then ], [ %2, %if.end ]
173+
ret i128 %retval.0
174+
}
175+
176+
define i128 @test_ssub_i128(i128 noundef %x, i128 noundef %y) {
177+
; CHECK-SD-LABEL: test_ssub_i128:
178+
; CHECK-SD: // %bb.0: // %entry
179+
; CHECK-SD-NEXT: subs x0, x0, x2
180+
; CHECK-SD-NEXT: sbcs x1, x1, x3
181+
; CHECK-SD-NEXT: cset w8, vs
182+
; CHECK-SD-NEXT: cmp w8, #1
183+
; CHECK-SD-NEXT: b.ne .LBB3_2
184+
; CHECK-SD-NEXT: // %bb.1: // %if.then
185+
; CHECK-SD-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
186+
; CHECK-SD-NEXT: .cfi_def_cfa_offset 16
187+
; CHECK-SD-NEXT: .cfi_offset w30, -16
188+
; CHECK-SD-NEXT: bl error
189+
; CHECK-SD-NEXT: // kill: def $w0 killed $w0 def $x0
190+
; CHECK-SD-NEXT: sxtw x0, w0
191+
; CHECK-SD-NEXT: asr x1, x0, #63
192+
; CHECK-SD-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
193+
; CHECK-SD-NEXT: .LBB3_2: // %cleanup
194+
; CHECK-SD-NEXT: ret
195+
;
196+
; CHECK-GI-LABEL: test_ssub_i128:
197+
; CHECK-GI: // %bb.0: // %entry
198+
; CHECK-GI-NEXT: subs x0, x0, x2
199+
; CHECK-GI-NEXT: sbcs x1, x1, x3
200+
; CHECK-GI-NEXT: b.vc .LBB3_2
201+
; CHECK-GI-NEXT: // %bb.1: // %if.then
202+
; CHECK-GI-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
203+
; CHECK-GI-NEXT: .cfi_def_cfa_offset 16
204+
; CHECK-GI-NEXT: .cfi_offset w30, -16
205+
; CHECK-GI-NEXT: bl error
206+
; CHECK-GI-NEXT: // kill: def $w0 killed $w0 def $x0
207+
; CHECK-GI-NEXT: asr w1, w0, #31
208+
; CHECK-GI-NEXT: bfi x0, x1, #32, #32
209+
; CHECK-GI-NEXT: bfi x1, x1, #32, #32
210+
; CHECK-GI-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
211+
; CHECK-GI-NEXT: .LBB3_2: // %cleanup
212+
; CHECK-GI-NEXT: ret
213+
entry:
214+
%0 = tail call { i128, i1 } @llvm.ssub.with.overflow.i128(i128 %x, i128 %y)
215+
%1 = extractvalue { i128, i1 } %0, 1
216+
br i1 %1, label %if.then, label %if.end
217+
218+
if.then:
219+
%call = tail call i32 @error()
220+
%conv1 = sext i32 %call to i128
221+
br label %cleanup
222+
223+
if.end:
224+
%2 = extractvalue { i128, i1 } %0, 0
225+
br label %cleanup
226+
227+
cleanup:
228+
%retval.0 = phi i128 [ %conv1, %if.then ], [ %2, %if.end ]
229+
ret i128 %retval.0
230+
}
231+
232+
define i128 @test_umul_i128(i128 noundef %x, i128 noundef %y) {
233+
; CHECK-LABEL: test_umul_i128:
234+
; CHECK: // %bb.0: // %entry
235+
; CHECK-NEXT: mul x9, x3, x0
236+
; CHECK-NEXT: cmp x1, #0
237+
; CHECK-NEXT: ccmp x3, #0, #4, ne
238+
; CHECK-NEXT: umulh x8, x1, x2
239+
; CHECK-NEXT: umulh x10, x3, x0
240+
; CHECK-NEXT: madd x9, x1, x2, x9
241+
; CHECK-NEXT: ccmp xzr, x8, #0, eq
242+
; CHECK-NEXT: umulh x11, x0, x2
243+
; CHECK-NEXT: ccmp xzr, x10, #0, eq
244+
; CHECK-NEXT: cset w8, ne
245+
; CHECK-NEXT: adds x1, x11, x9
246+
; CHECK-NEXT: csinc w8, w8, wzr, lo
247+
; CHECK-NEXT: cmp w8, #1
248+
; CHECK-NEXT: b.ne .LBB4_2
249+
; CHECK-NEXT: // %bb.1: // %if.then
250+
; CHECK-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
251+
; CHECK-NEXT: .cfi_def_cfa_offset 16
252+
; CHECK-NEXT: .cfi_offset w30, -16
253+
; CHECK-NEXT: bl error
254+
; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
255+
; CHECK-NEXT: sxtw x0, w0
256+
; CHECK-NEXT: asr x1, x0, #63
257+
; CHECK-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
258+
; CHECK-NEXT: ret
259+
; CHECK-NEXT: .LBB4_2: // %if.end
260+
; CHECK-NEXT: mul x0, x0, x2
261+
; CHECK-NEXT: ret
262+
entry:
263+
%0 = tail call { i128, i1 } @llvm.umul.with.overflow.i128(i128 %x, i128 %y)
264+
%1 = extractvalue { i128, i1 } %0, 1
265+
br i1 %1, label %if.then, label %if.end
266+
267+
if.then:
268+
%call = tail call i32 @error()
269+
%conv1 = sext i32 %call to i128
270+
br label %cleanup
271+
272+
if.end:
273+
%2 = extractvalue { i128, i1 } %0, 0
274+
br label %cleanup
275+
276+
cleanup:
277+
%retval.0 = phi i128 [ %conv1, %if.then ], [ %2, %if.end ]
278+
ret i128 %retval.0
279+
}
280+
281+
define i128 @test_smul_i128(i128 noundef %x, i128 noundef %y) {
282+
; CHECK-LABEL: test_smul_i128:
283+
; CHECK: // %bb.0: // %entry
284+
; CHECK-NEXT: asr x10, x1, #63
285+
; CHECK-NEXT: umulh x11, x0, x2
286+
; CHECK-NEXT: asr x14, x3, #63
287+
; CHECK-NEXT: mov x8, x1
288+
; CHECK-NEXT: mul x12, x1, x2
289+
; CHECK-NEXT: umulh x9, x1, x2
290+
; CHECK-NEXT: mul x10, x10, x2
291+
; CHECK-NEXT: adds x11, x12, x11
292+
; CHECK-NEXT: mul x15, x0, x3
293+
; CHECK-NEXT: umulh x13, x0, x3
294+
; CHECK-NEXT: adc x9, x9, x10
295+
; CHECK-NEXT: mul x14, x0, x14
296+
; CHECK-NEXT: mul x16, x1, x3
297+
; CHECK-NEXT: adds x1, x15, x11
298+
; CHECK-NEXT: asr x11, x9, #63
299+
; CHECK-NEXT: smulh x8, x8, x3
300+
; CHECK-NEXT: adc x10, x13, x14
301+
; CHECK-NEXT: asr x12, x10, #63
302+
; CHECK-NEXT: adds x9, x9, x10
303+
; CHECK-NEXT: adc x10, x11, x12
304+
; CHECK-NEXT: adds x9, x16, x9
305+
; CHECK-NEXT: asr x11, x1, #63
306+
; CHECK-NEXT: adc x8, x8, x10
307+
; CHECK-NEXT: eor x8, x8, x11
308+
; CHECK-NEXT: eor x9, x9, x11
309+
; CHECK-NEXT: orr x8, x9, x8
310+
; CHECK-NEXT: cbz x8, .LBB5_2
311+
; CHECK-NEXT: // %bb.1: // %if.then
312+
; CHECK-NEXT: str x30, [sp, #-16]! // 8-byte Folded Spill
313+
; CHECK-NEXT: .cfi_def_cfa_offset 16
314+
; CHECK-NEXT: .cfi_offset w30, -16
315+
; CHECK-NEXT: bl error
316+
; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
317+
; CHECK-NEXT: sxtw x0, w0
318+
; CHECK-NEXT: asr x1, x0, #63
319+
; CHECK-NEXT: ldr x30, [sp], #16 // 8-byte Folded Reload
320+
; CHECK-NEXT: ret
321+
; CHECK-NEXT: .LBB5_2: // %if.end
322+
; CHECK-NEXT: mul x0, x0, x2
323+
; CHECK-NEXT: ret
324+
entry:
325+
%0 = tail call { i128, i1 } @llvm.smul.with.overflow.i128(i128 %x, i128 %y)
326+
%1 = extractvalue { i128, i1 } %0, 1
327+
br i1 %1, label %if.then, label %if.end
328+
329+
if.then:
330+
%call = tail call i32 @error()
331+
%conv1 = sext i32 %call to i128
332+
br label %cleanup
333+
334+
if.end:
335+
%2 = extractvalue { i128, i1 } %0, 0
336+
br label %cleanup
337+
338+
cleanup:
339+
%retval.0 = phi i128 [ %conv1, %if.then ], [ %2, %if.end ]
340+
ret i128 %retval.0
341+
}
342+
343+
declare i32 @error()

0 commit comments

Comments
 (0)