1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc < %s -mtriple=aarch64-unknown-unknown | FileCheck %s
3; RUN: llc < %s -mtriple=aarch64-windows         | FileCheck %s -check-prefix=CHECK-WIN
4; RUN: llc < %s -mtriple=aarch64-apple-darwin    | FileCheck %s -check-prefix=CHECK-DARWIN
5
6; The Windows runtime doesn't have these.
7; CHECK-WIN-NOT: __ashlti3
8; CHECK-WIN-NOT: __ashrti3
9
10; Darwin compiler-rt excludes these.
11; CHECK-DARWIN-NOT: __ashlti3
12; CHECK-DARWIN-NOT: __ashrti3
13
14define i64 @f0(i64 %val, i64 %amt) minsize optsize {
15; CHECK-LABEL: f0:
16; CHECK:       // %bb.0:
17; CHECK-NEXT:    lsl x0, x0, x1
18; CHECK-NEXT:    ret
19;
20; CHECK-DARWIN-LABEL: f0:
21; CHECK-DARWIN:       ; %bb.0:
22; CHECK-DARWIN-NEXT:    lsl x0, x0, x1
23; CHECK-DARWIN-NEXT:    ret
24  %res = shl i64 %val, %amt
25  ret i64 %res
26}
27
28define i32 @f1(i64 %x, i64 %y) minsize optsize {
29; CHECK-LABEL: f1:
30; CHECK:       // %bb.0:
31; CHECK-NEXT:    lsl x0, x0, x1
32; CHECK-NEXT:    // kill: def $w0 killed $w0 killed $x0
33; CHECK-NEXT:    ret
34;
35; CHECK-DARWIN-LABEL: f1:
36; CHECK-DARWIN:       ; %bb.0:
37; CHECK-DARWIN-NEXT:    lsl x0, x0, x1
38; CHECK-DARWIN-NEXT:    ; kill: def $w0 killed $w0 killed $x0
39; CHECK-DARWIN-NEXT:    ret
40	%a = shl i64 %x, %y
41	%b = trunc i64 %a to i32
42	ret i32 %b
43}
44
45define i32 @f2(i64 %x, i64 %y) minsize optsize {
46; CHECK-LABEL: f2:
47; CHECK:       // %bb.0:
48; CHECK-NEXT:    asr x0, x0, x1
49; CHECK-NEXT:    // kill: def $w0 killed $w0 killed $x0
50; CHECK-NEXT:    ret
51;
52; CHECK-DARWIN-LABEL: f2:
53; CHECK-DARWIN:       ; %bb.0:
54; CHECK-DARWIN-NEXT:    asr x0, x0, x1
55; CHECK-DARWIN-NEXT:    ; kill: def $w0 killed $w0 killed $x0
56; CHECK-DARWIN-NEXT:    ret
57	%a = ashr i64 %x, %y
58	%b = trunc i64 %a to i32
59	ret i32 %b
60}
61
62define i32 @f3(i64 %x, i64 %y) minsize optsize {
63; CHECK-LABEL: f3:
64; CHECK:       // %bb.0:
65; CHECK-NEXT:    lsr x0, x0, x1
66; CHECK-NEXT:    // kill: def $w0 killed $w0 killed $x0
67; CHECK-NEXT:    ret
68;
69; CHECK-DARWIN-LABEL: f3:
70; CHECK-DARWIN:       ; %bb.0:
71; CHECK-DARWIN-NEXT:    lsr x0, x0, x1
72; CHECK-DARWIN-NEXT:    ; kill: def $w0 killed $w0 killed $x0
73; CHECK-DARWIN-NEXT:    ret
74	%a = lshr i64 %x, %y
75	%b = trunc i64 %a to i32
76	ret i32 %b
77}
78
79define dso_local { i64, i64 } @shl128(i64 %x.coerce0, i64 %x.coerce1, i8 signext %y) minsize optsize {
80; CHECK-LABEL: shl128:
81; CHECK:       // %bb.0: // %entry
82; CHECK-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
83; CHECK-NEXT:    .cfi_def_cfa_offset 16
84; CHECK-NEXT:    .cfi_offset w30, -16
85; CHECK-NEXT:    bl __ashlti3
86; CHECK-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
87; CHECK-NEXT:    ret
88;
89; CHECK-DARWIN-LABEL: shl128:
90; CHECK-DARWIN:       ; %bb.0: ; %entry
91; CHECK-DARWIN-NEXT:    mvn w8, w2
92; CHECK-DARWIN-NEXT:    mov w9, w2
93; CHECK-DARWIN-NEXT:    lsr x10, x0, #1
94; CHECK-DARWIN-NEXT:    tst x9, #0x40
95; CHECK-DARWIN-NEXT:    lsr x8, x10, x8
96; CHECK-DARWIN-NEXT:    lsl x10, x1, x9
97; CHECK-DARWIN-NEXT:    orr x8, x10, x8
98; CHECK-DARWIN-NEXT:    lsl x10, x0, x9
99; CHECK-DARWIN-NEXT:    csel x1, x10, x8, ne
100; CHECK-DARWIN-NEXT:    csel x0, xzr, x10, ne
101; CHECK-DARWIN-NEXT:    ret
102
103entry:
104  %x.sroa.2.0.insert.ext = zext i64 %x.coerce1 to i128
105  %x.sroa.2.0.insert.shift = shl nuw i128 %x.sroa.2.0.insert.ext, 64
106  %x.sroa.0.0.insert.ext = zext i64 %x.coerce0 to i128
107  %x.sroa.0.0.insert.insert = or i128 %x.sroa.2.0.insert.shift, %x.sroa.0.0.insert.ext
108  %conv = sext i8 %y to i32
109  %sh_prom = zext i32 %conv to i128
110  %shl = shl i128 %x.sroa.0.0.insert.insert, %sh_prom
111  %retval.sroa.0.0.extract.trunc = trunc i128 %shl to i64
112  %retval.sroa.2.0.extract.shift = lshr i128 %shl, 64
113  %retval.sroa.2.0.extract.trunc = trunc i128 %retval.sroa.2.0.extract.shift to i64
114  %.fca.0.insert = insertvalue { i64, i64 } undef, i64 %retval.sroa.0.0.extract.trunc, 0
115  %.fca.1.insert = insertvalue { i64, i64 } %.fca.0.insert, i64 %retval.sroa.2.0.extract.trunc, 1
116  ret { i64, i64 } %.fca.1.insert
117}
118
119define dso_local { i64, i64 } @ashr128(i64 %x.coerce0, i64 %x.coerce1, i8 signext %y) minsize optsize {
120; CHECK-LABEL: ashr128:
121; CHECK:       // %bb.0: // %entry
122; CHECK-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
123; CHECK-NEXT:    .cfi_def_cfa_offset 16
124; CHECK-NEXT:    .cfi_offset w30, -16
125; CHECK-NEXT:    bl __ashrti3
126; CHECK-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
127; CHECK-NEXT:    ret
128;
129; CHECK-DARWIN-LABEL: ashr128:
130; CHECK-DARWIN:       ; %bb.0: ; %entry
131; CHECK-DARWIN-NEXT:    mov w8, w2
132; CHECK-DARWIN-NEXT:    mvn w9, w2
133; CHECK-DARWIN-NEXT:    lsl x10, x1, #1
134; CHECK-DARWIN-NEXT:    tst x8, #0x40
135; CHECK-DARWIN-NEXT:    lsr x11, x0, x8
136; CHECK-DARWIN-NEXT:    lsl x9, x10, x9
137; CHECK-DARWIN-NEXT:    asr x10, x1, x8
138; CHECK-DARWIN-NEXT:    orr x9, x9, x11
139; CHECK-DARWIN-NEXT:    asr x8, x1, #63
140; CHECK-DARWIN-NEXT:    csel x0, x10, x9, ne
141; CHECK-DARWIN-NEXT:    csel x1, x8, x10, ne
142; CHECK-DARWIN-NEXT:    ret
143entry:
144  %x.sroa.2.0.insert.ext = zext i64 %x.coerce1 to i128
145  %x.sroa.2.0.insert.shift = shl nuw i128 %x.sroa.2.0.insert.ext, 64
146  %x.sroa.0.0.insert.ext = zext i64 %x.coerce0 to i128
147  %x.sroa.0.0.insert.insert = or i128 %x.sroa.2.0.insert.shift, %x.sroa.0.0.insert.ext
148  %conv = sext i8 %y to i32
149  %sh_prom = zext i32 %conv to i128
150  %shr = ashr i128 %x.sroa.0.0.insert.insert, %sh_prom
151  %retval.sroa.0.0.extract.trunc = trunc i128 %shr to i64
152  %retval.sroa.2.0.extract.shift = lshr i128 %shr, 64
153  %retval.sroa.2.0.extract.trunc = trunc i128 %retval.sroa.2.0.extract.shift to i64
154  %.fca.0.insert = insertvalue { i64, i64 } undef, i64 %retval.sroa.0.0.extract.trunc, 0
155  %.fca.1.insert = insertvalue { i64, i64 } %.fca.0.insert, i64 %retval.sroa.2.0.extract.trunc, 1
156  ret { i64, i64 } %.fca.1.insert
157}
158
159define dso_local { i64, i64 } @lshr128(i64 %x.coerce0, i64 %x.coerce1, i8 signext %y) minsize optsize {
160; CHECK-LABEL: lshr128:
161; CHECK:       // %bb.0: // %entry
162; CHECK-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
163; CHECK-NEXT:    .cfi_def_cfa_offset 16
164; CHECK-NEXT:    .cfi_offset w30, -16
165; CHECK-NEXT:    bl __lshrti3
166; CHECK-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
167; CHECK-NEXT:    ret
168;
169; CHECK-DARWIN-LABEL: lshr128:
170; CHECK-DARWIN:       ; %bb.0: ; %entry
171; CHECK-DARWIN-NEXT:    mov w8, w2
172; CHECK-DARWIN-NEXT:    mvn w9, w2
173; CHECK-DARWIN-NEXT:    lsl x10, x1, #1
174; CHECK-DARWIN-NEXT:    tst x8, #0x40
175; CHECK-DARWIN-NEXT:    lsr x11, x0, x8
176; CHECK-DARWIN-NEXT:    lsl x9, x10, x9
177; CHECK-DARWIN-NEXT:    orr x9, x9, x11
178; CHECK-DARWIN-NEXT:    lsr x10, x1, x8
179; CHECK-DARWIN-NEXT:    csel x0, x10, x9, ne
180; CHECK-DARWIN-NEXT:    csel x1, xzr, x10, ne
181; CHECK-DARWIN-NEXT:    ret
182entry:
183  %x.sroa.2.0.insert.ext = zext i64 %x.coerce1 to i128
184  %x.sroa.2.0.insert.shift = shl nuw i128 %x.sroa.2.0.insert.ext, 64
185  %x.sroa.0.0.insert.ext = zext i64 %x.coerce0 to i128
186  %x.sroa.0.0.insert.insert = or i128 %x.sroa.2.0.insert.shift, %x.sroa.0.0.insert.ext
187  %conv = sext i8 %y to i32
188  %sh_prom = zext i32 %conv to i128
189  %shr = lshr i128 %x.sroa.0.0.insert.insert, %sh_prom
190  %retval.sroa.0.0.extract.trunc = trunc i128 %shr to i64
191  %retval.sroa.2.0.extract.shift = lshr i128 %shr, 64
192  %retval.sroa.2.0.extract.trunc = trunc i128 %retval.sroa.2.0.extract.shift to i64
193  %.fca.0.insert = insertvalue { i64, i64 } undef, i64 %retval.sroa.0.0.extract.trunc, 0
194  %.fca.1.insert = insertvalue { i64, i64 } %.fca.0.insert, i64 %retval.sroa.2.0.extract.trunc, 1
195  ret { i64, i64 } %.fca.1.insert
196}
197