1 ; RUN: llc -mtriple=aarch64-none-linux-gnu -verify-machineinstrs < %s | FileCheck %s
2 ; RUN: llc -mtriple=aarch64-none-linux-gnu -mattr=-fp-armv8 -verify-machineinstrs < %s | FileCheck --check-prefix=CHECK-NOFP %s
4 declare void @use_addr(i8*)
6 define void @test_simple_alloca(i64 %n) {
7 ; CHECK-LABEL: test_simple_alloca:
9 %buf = alloca i8, i64 %n
10 ; Make sure we align the stack change to 16 bytes:
11 ; CHECK-DAG: add [[SPDELTA:x[0-9]+]], x0, #15
12 ; CHECK-DAG: and x0, [[SPDELTA]], #0xfffffffffffffff0
14 ; Make sure we change SP. It would be surprising if anything but x0 were used
15 ; for the final sp, but it could be if it was then moved into x0.
16 ; CHECK-DAG: mov [[TMP:x[0-9]+]], sp
17 ; CHECK-DAG: sub x0, [[TMP]], [[SPDELTA]]
20 call void @use_addr(i8* %buf)
24 ; Make sure epilogue restores sp from fp
25 ; CHECK: sub sp, x29, #16
26 ; CHECK: ldp x29, x30, [sp, #16]
27 ; CHECK: add sp, sp, #32
31 declare void @use_addr_loc(i8*, i64*)
33 define i64 @test_alloca_with_local(i64 %n) {
34 ; CHECK-LABEL: test_alloca_with_local:
35 ; CHECK: sub sp, sp, #32
36 ; CHECK: stp x29, x30, [sp, #16]
39 %buf = alloca i8, i64 %n
40 ; Make sure we align the stack change to 16 bytes:
41 ; CHECK-DAG: add [[SPDELTA:x[0-9]+]], x0, #15
42 ; CHECK-DAG: and x0, [[SPDELTA]], #0xfffffffffffffff0
44 ; Make sure we change SP. It would be surprising if anything but x0 were used
45 ; for the final sp, but it could be if it was then moved into x0.
46 ; CHECK-DAG: mov [[TMP:x[0-9]+]], sp
47 ; CHECK-DAG: sub x0, [[TMP]], [[SPDELTA]]
50 ; Obviously suboptimal code here, but it to get &local in x1
51 ; CHECK: sub [[TMP:x[0-9]+]], x29, [[LOC_FROM_FP:#[0-9]+]]
52 ; CHECK: add x1, [[TMP]], #0
54 call void @use_addr_loc(i8* %buf, i64* %loc)
58 ; CHECK: sub x[[TMP:[0-9]+]], x29, [[LOC_FROM_FP]]
59 ; CHECK: ldr x0, [x[[TMP]]]
62 ; Make sure epilogue restores sp from fp
63 ; CHECK: sub sp, x29, #16
64 ; CHECK: ldp x29, x30, [sp, #16]
65 ; CHECK: add sp, sp, #32
69 define void @test_variadic_alloca(i64 %n, ...) {
70 ; CHECK: test_variadic_alloca:
72 ; CHECK: sub sp, sp, #208
73 ; CHECK: stp x29, x30, [sp, #192]
74 ; CHECK: add x29, sp, #192
75 ; CHECK: sub [[TMP:x[0-9]+]], x29, #192
76 ; CHECK: add x8, [[TMP]], #0
77 ; CHECK-FP: str q7, [x8, #112]
79 ; CHECK-FP: str q1, [x8, #16]
81 ; CHECK-NOFP: sub sp, sp, #80
82 ; CHECK-NOFP: stp x29, x30, [sp, #64]
83 ; CHECK-NOFP: add x29, sp, #64
84 ; CHECK-NOFP: sub [[TMP:x[0-9]+]], x29, #64
85 ; CHECK-NOFP: add x8, [[TMP]], #0
87 %addr = alloca i8, i64 %n
89 call void @use_addr(i8* %addr)
93 ; CHECK: sub sp, x29, #192
94 ; CHECK: ldp x29, x30, [sp, #192]
95 ; CHECK: add sp, sp, #208
97 ; CHECK-NOFP: sub sp, x29, #64
98 ; CHECK-NOFP: ldp x29, x30, [sp, #64]
99 ; CHECK-NOFP: add sp, sp, #80
102 define void @test_alloca_large_frame(i64 %n) {
103 ; CHECK-LABEL: test_alloca_large_frame:
105 ; CHECK: sub sp, sp, #496
106 ; CHECK: stp x29, x30, [sp, #480]
107 ; CHECK: add x29, sp, #480
108 ; CHECK: sub sp, sp, #48
109 ; CHECK: sub sp, sp, #1953, lsl #12
111 %addr1 = alloca i8, i64 %n
112 %addr2 = alloca i64, i64 1000000
114 call void @use_addr_loc(i8* %addr1, i64* %addr2)
117 ; CHECK: sub sp, x29, #480
118 ; CHECK: ldp x29, x30, [sp, #480]
119 ; CHECK: add sp, sp, #496
122 declare i8* @llvm.stacksave()
123 declare void @llvm.stackrestore(i8*)
125 define void @test_scoped_alloca(i64 %n) {
126 ; CHECK-LABEL: test_scoped_alloca:
127 ; CHECK: sub sp, sp, #32
129 %sp = call i8* @llvm.stacksave()
130 ; CHECK: mov [[SAVED_SP:x[0-9]+]], sp
131 ; CHECK: mov [[OLDSP:x[0-9]+]], sp
133 %addr = alloca i8, i64 %n
134 ; CHECK: and [[SPDELTA:x[0-9]+]], {{x[0-9]+}}, #0xfffffffffffffff0
135 ; CHECK-DAG: sub [[NEWSP:x[0-9]+]], [[OLDSP]], [[SPDELTA]]
136 ; CHECK: mov sp, [[NEWSP]]
138 call void @use_addr(i8* %addr)
141 call void @llvm.stackrestore(i8* %sp)
142 ; CHECK: mov sp, [[SAVED_SP]]