1; RUN: llc < %s -asm-verbose=false -disable-wasm-fallthrough-return-opt | FileCheck %s
2
3target datalayout = "e-m:e-p:32:32-i64:64-n32:64-S128"
4target triple = "wasm32-unknown-unknown"
5
6declare void @ext_func(i64* %ptr)
7declare void @ext_func_i32(i32* %ptr)
8
9; CHECK-LABEL: alloca32:
10; Check that there is an extra local for the stack pointer.
11; CHECK: .local i32{{$}}
12define void @alloca32() noredzone {
13 ; CHECK: i32.const $push[[L4:.+]]=, 0{{$}}
14 ; CHECK: i32.const $push[[L1:.+]]=, 0{{$}}
15 ; CHECK-NEXT: i32.load $push[[L2:.+]]=, __stack_pointer($pop[[L1]])
16 ; CHECK-NEXT: i32.const $push[[L3:.+]]=, 16
17 ; CHECK-NEXT: i32.sub $push[[L9:.+]]=, $pop[[L2]], $pop[[L3]]
18 ; CHECK-NEXT: tee_local $push[[L8:.+]]=, $[[SP:.+]]=, $pop[[L9]]{{$}}
19 ; CHECK-NEXT: i32.store $drop=, __stack_pointer($pop[[L4]]), $pop[[L8]]{{$}}
20 %retval = alloca i32
21 ; CHECK: i32.const $push[[L0:.+]]=, 0
22 ; CHECK: i32.store $drop=, 12($[[SP]]), $pop[[L0]]
23 store i32 0, i32* %retval
24 ; CHECK: i32.const $push[[L6:.+]]=, 0
25 ; CHECK-NEXT: i32.const $push[[L5:.+]]=, 16
26 ; CHECK-NEXT: i32.add $push[[L7:.+]]=, $[[SP]], $pop[[L5]]
27 ; CHECK-NEXT: i32.store $drop=, __stack_pointer($pop[[L6]]), $pop[[L7]]
28 ret void
29}
30
31; CHECK-LABEL: alloca3264:
32; CHECK: .local i32{{$}}
33define void @alloca3264() {
34 ; CHECK: i32.const $push[[L2:.+]]=, 0{{$}}
35 ; CHECK-NEXT: i32.load $push[[L3:.+]]=, __stack_pointer($pop[[L2]])
36 ; CHECK-NEXT: i32.const $push[[L4:.+]]=, 16
37 ; CHECK-NEXT: i32.sub $push[[L6:.+]]=, $pop[[L3]], $pop[[L4]]
38 ; CHECK-NEXT: tee_local $push[[L5:.+]]=, $[[SP:.+]]=, $pop[[L6]]
39 %r1 = alloca i32
40 %r2 = alloca double
41 ; CHECK-NEXT: i32.const $push[[L0:.+]]=, 0
42 ; CHECK-NEXT: i32.store $drop=, 12($pop[[L5]]), $pop[[L0]]
43 store i32 0, i32* %r1
44 ; CHECK-NEXT: i64.const $push[[L1:.+]]=, 0
45 ; CHECK-NEXT: i64.store $drop=, 0($[[SP]]), $pop[[L1]]
46 store double 0.0, double* %r2
47 ; CHECK-NEXT: return
48 ret void
49}
50
51; CHECK-LABEL: allocarray:
52; CHECK: .local i32{{$}}
53define void @allocarray() {
54 ; CHECK: i32.const $push[[L6:.+]]=, 0{{$}}
55 ; CHECK: i32.const $push[[L3:.+]]=, 0{{$}}
56 ; CHECK-NEXT: i32.load $push[[L4:.+]]=, __stack_pointer($pop[[L3]])
57 ; CHECK-NEXT: i32.const $push[[L5:.+]]=, 144{{$}}
58 ; CHECK-NEXT: i32.sub $push[[L12:.+]]=, $pop[[L4]], $pop[[L5]]
59 ; CHECK-NEXT: tee_local $push[[L11:.+]]=, $0=, $pop[[L12]]
60 ; CHECK-NEXT: i32.store $drop=, __stack_pointer($pop[[L6]]), $pop[[L11]]
61 %r = alloca [33 x i32]
62
63 ; CHECK:      i32.const $push{{.+}}=, 24
64 ; CHECK-NEXT: i32.add $push[[L3:.+]]=, $[[SP]], $pop{{.+}}
65 ; CHECK-NEXT: i32.const $push[[L1:.+]]=, 1{{$}}
66 ; CHECK-NEXT: i32.store $drop=, 0($pop[[L3]]), $pop[[L1]]{{$}}
67 ; CHECK-NEXT: i32.const $push[[L10:.+]]=, 1{{$}}
68 ; CHECK-NEXT: i32.store $drop=, 12(${{.+}}), $pop[[L10]]{{$}}
69 %p = getelementptr [33 x i32], [33 x i32]* %r, i32 0, i32 0
70 store i32 1, i32* %p
71 %p2 = getelementptr [33 x i32], [33 x i32]* %r, i32 0, i32 3
72 store i32 1, i32* %p2
73
74 ; CHECK: i32.const $push[[L9:.+]]=, 0{{$}}
75 ; CHECK-NEXT: i32.const $push[[L7:.+]]=, 144
76 ; CHECK-NEXT: i32.add $push[[L8:.+]]=, $[[SP]], $pop[[L7]]
77 ; CHECK-NEXT: i32.store $drop=, __stack_pointer($pop[[L9]]), $pop[[L8]]
78 ret void
79}
80
81; CHECK-LABEL: non_mem_use
82define void @non_mem_use(i8** %addr) {
83 ; CHECK: i32.const $push[[L2:.+]]=, 48
84 ; CHECK-NEXT: i32.sub $push[[L12:.+]]=, {{.+}}, $pop[[L2]]
85 ; CHECK-NEXT: tee_local $push[[L11:.+]]=, $[[SP:.+]]=, $pop[[L12]]
86 ; CHECK-NEXT: i32.store $drop=, {{.+}}, $pop[[L11]]
87 %buf = alloca [27 x i8], align 16
88 %r = alloca i64
89 %r2 = alloca i64
90 ; %r is at SP+8
91 ; CHECK: i32.const $push[[OFF:.+]]=, 8
92 ; CHECK-NEXT: i32.add $push[[ARG1:.+]]=, $[[SP]], $pop[[OFF]]
93 ; CHECK-NEXT: call ext_func@FUNCTION, $pop[[ARG1]]
94 call void @ext_func(i64* %r)
95 ; %r2 is at SP+0, no add needed
96 ; CHECK-NEXT: call ext_func@FUNCTION, $[[SP]]
97 call void @ext_func(i64* %r2)
98 ; Use as a value, but in a store
99 ; %buf is at SP+16
100 ; CHECK: i32.const $push[[OFF:.+]]=, 16
101 ; CHECK-NEXT: i32.add $push[[VAL:.+]]=, $[[SP]], $pop[[OFF]]
102 ; CHECK-NEXT: i32.store $drop=, 0($0), $pop[[VAL]]
103 %gep = getelementptr inbounds [27 x i8], [27 x i8]* %buf, i32 0, i32 0
104 store i8* %gep, i8** %addr
105 ret void
106}
107
108; CHECK-LABEL: allocarray_inbounds:
109; CHECK: .local i32{{$}}
110define void @allocarray_inbounds() {
111 ; CHECK: i32.const $push[[L5:.+]]=, 0{{$}}
112 ; CHECK: i32.const $push[[L2:.+]]=, 0{{$}}
113 ; CHECK-NEXT: i32.load $push[[L3:.+]]=, __stack_pointer($pop[[L2]])
114 ; CHECK-NEXT: i32.const $push[[L4:.+]]=, 32{{$}}
115 ; CHECK-NEXT: i32.sub $push[[L11:.+]]=, $pop[[L3]], $pop[[L4]]
116 ; CHECK-NEXT: tee_local $push[[L10:.+]]=, $[[SP:.+]]=, $pop[[L11]]
117 ; CHECK-NEXT: i32.store $drop=, __stack_pointer($pop[[L5]]), $pop[[L10]]{{$}}
118 %r = alloca [5 x i32]
119 ; CHECK: i32.const $push[[L3:.+]]=, 1
120 ; CHECK-DAG: i32.store $drop=, 24(${{.+}}), $pop[[L3]]
121 %p = getelementptr inbounds [5 x i32], [5 x i32]* %r, i32 0, i32 0
122 store i32 1, i32* %p
123 ; This store should have both the GEP and the FI folded into it.
124 ; CHECK-DAG: i32.store {{.*}}=, 12(${{.+}}), $pop
125 %p2 = getelementptr inbounds [5 x i32], [5 x i32]* %r, i32 0, i32 3
126 store i32 1, i32* %p2
127 call void @ext_func(i64* null);
128 ; CHECK: call ext_func
129 ; CHECK: i32.const $push[[L6:.+]]=, 0{{$}}
130 ; CHECK-NEXT: i32.const $push[[L5:.+]]=, 32{{$}}
131 ; CHECK-NEXT: i32.add $push[[L7:.+]]=, ${{.+}}, $pop[[L5]]
132 ; CHECK-NEXT: i32.store $drop=, __stack_pointer($pop[[L6]]), $pop[[L7]]
133 ret void
134}
135
136; CHECK-LABEL: dynamic_alloca:
137define void @dynamic_alloca(i32 %alloc) {
138 ; CHECK: i32.const $push[[L7:.+]]=, 0{{$}}
139 ; CHECK: i32.const $push[[L1:.+]]=, 0{{$}}
140 ; CHECK-NEXT: i32.load $push[[L13:.+]]=, __stack_pointer($pop[[L1]])
141 ; CHECK-NEXT: tee_local $push[[L12:.+]]=, [[SP:.+]], $pop[[L13]]{{$}}
142 ; Target independent codegen bumps the stack pointer.
143 ; CHECK: i32.sub
144 ; Check that SP is written back to memory after decrement
145 ; CHECK: i32.store $drop=, __stack_pointer($pop{{.+}}),
146 %r = alloca i32, i32 %alloc
147 ; Target-independent codegen also calculates the store addr
148 ; CHECK: call ext_func_i32@FUNCTION
149 call void @ext_func_i32(i32* %r)
150 ; CHECK: i32.const $push[[L3:.+]]=, 0{{$}}
151 ; CHECK: i32.store $drop=, __stack_pointer($pop[[L3]]), $pop{{.+}}
152 ret void
153}
154
155; CHECK-LABEL: dynamic_alloca_redzone:
156define void @dynamic_alloca_redzone(i32 %alloc) {
157 ; CHECK: i32.const $push[[L8:.+]]=, 0{{$}}
158 ; CHECK-NEXT: i32.load $push[[L13:.+]]=, __stack_pointer($pop[[L1]])
159 ; CHECK-NEXT: tee_local $push[[L12:.+]]=, [[SP:.+]], $pop[[L13]]{{$}}
160 ; CHECK-NEXT: copy_local [[FP:.+]]=, $pop[[L12]]{{$}}
161 ; Target independent codegen bumps the stack pointer
162 ; CHECK: i32.sub
163 %r = alloca i32, i32 %alloc
164 ; CHECK-NEXT: tee_local       $push[[L8:.+]]=, $0=, $pop
165 ; CHECK-NEXT: copy_local      $drop=, $pop[[L8]]{{$}}
166 ; CHECK-NEXT: i32.const       $push[[L6:.+]]=, 0{{$}}
167 ; CHECK-NEXT: i32.store       $drop=, 0($0), $pop[[L6]]{{$}}
168 store i32 0, i32* %r
169 ; CHECK-NEXT: return
170 ret void
171}
172
173; CHECK-LABEL: dynamic_static_alloca:
174define void @dynamic_static_alloca(i32 %alloc) noredzone {
175 ; Decrement SP in the prolog by the static amount and writeback to memory.
176 ; CHECK: i32.const $push[[L11:.+]]=, 0{{$}}
177 ; CHECK: i32.const $push[[L8:.+]]=, 0{{$}}
178 ; CHECK-NEXT: i32.load $push[[L9:.+]]=, __stack_pointer($pop[[L8]])
179 ; CHECK-NEXT: i32.const $push[[L10:.+]]=, 16
180 ; CHECK-NEXT: i32.sub $push[[L20:.+]]=, $pop[[L9]], $pop[[L10]]
181 ; CHECK-NEXT: tee_local $push[[L19:.+]]=, $[[FP:.+]]=, $pop[[L20]]
182 ; CHECK:      i32.store $drop=, __stack_pointer($pop{{.+}}), $pop{{.+}}
183 ; Decrement SP in the body by the dynamic amount.
184 ; CHECK: i32.sub
185 ; Writeback to memory.
186 ; CHECK: i32.store $drop=, __stack_pointer($pop{{.+}}), $pop{{.+}}
187 %r1 = alloca i32
188 %r = alloca i32, i32 %alloc
189 store i32 0, i32* %r
190 ; CHEC: i32.store $drop=, 0($pop{{.+}}), $pop{{.+}}
191 ret void
192}
193
194; The use of the alloca in a phi causes a CopyToReg DAG node to be generated,
195; which has to have special handling because CopyToReg can't have a FI operand
196; CHECK-LABEL: copytoreg_fi:
197define void @copytoreg_fi(i1 %cond, i32* %b) {
198entry:
199 ; CHECK: i32.const $push[[L1:.+]]=, 16
200 ; CHECK-NEXT: i32.sub $push[[L3:.+]]=, {{.+}}, $pop[[L1]]
201 %addr = alloca i32
202 ; CHECK: i32.const $push[[OFF:.+]]=, 12
203 ; CHECK-NEXT: i32.add $push[[ADDR:.+]]=, $pop[[L3]], $pop[[OFF]]
204 ; CHECK-NEXT: copy_local [[COPY:.+]]=, $pop[[ADDR]]
205 br label %body
206body:
207 %a = phi i32* [%addr, %entry], [%b, %body]
208 store i32 1, i32* %a
209 ; CHECK: i32.store {{.*}}, 0([[COPY]]),
210 br i1 %cond, label %body, label %exit
211exit:
212 ret void
213}
214
215declare void @use_i8_star(i8*)
216declare i8* @llvm.frameaddress(i32)
217
218; Test __builtin_frame_address(0).
219; CHECK-LABEL: frameaddress_0:
220; CHECK: i32.const $push[[L0:.+]]=, 0{{$}}
221; CHECK-NEXT: i32.load $push[[L3:.+]]=, __stack_pointer($pop[[L0]])
222; CHECK-NEXT: copy_local $push[[L4:.+]]=, $pop[[L3]]{{$}}
223; CHECK-NEXT: tee_local $push[[L2:.+]]=, $[[FP:.+]]=, $pop[[L4]]{{$}}
224; CHECK-NEXT: call use_i8_star@FUNCTION, $pop[[L2]]
225; CHECK-NEXT: i32.const $push[[L1:.+]]=, 0{{$}}
226; CHECK-NEXT: i32.store $drop=, __stack_pointer($pop[[L1]]), $[[FP]]
227define void @frameaddress_0() {
228  %t = call i8* @llvm.frameaddress(i32 0)
229  call void @use_i8_star(i8* %t)
230  ret void
231}
232
233; Test __builtin_frame_address(1).
234
235; CHECK-LABEL: frameaddress_1:
236; CHECK-NEXT: i32.const $push0=, 0{{$}}
237; CHECK-NEXT: call use_i8_star@FUNCTION, $pop0{{$}}
238; CHECK-NEXT: return{{$}}
239define void @frameaddress_1() {
240  %t = call i8* @llvm.frameaddress(i32 1)
241  call void @use_i8_star(i8* %t)
242  ret void
243}
244
245; Test a stack address passed to an inline asm.
246; CHECK-LABEL: inline_asm:
247; CHECK:       __stack_pointer
248; CHECK:       #APP
249; CHECK-NEXT:  # %{{[0-9]+}}{{$}}
250; CHECK-NEXT:  #NO_APP
251define void @inline_asm() {
252  %tmp = alloca i8
253  call void asm sideeffect "# %0", "r"(i8* %tmp)
254  ret void
255}
256
257; TODO: test over-aligned alloca
258