176 lines · plain
1; RUN: llc -mtriple=aarch64-linux-gnu -disable-post-ra -verify-machineinstrs -o - %s | FileCheck %s2; RUN: llc -mtriple=arm64-apple-ios -disable-post-ra -verify-machineinstrs -o - %s | FileCheck %s --check-prefix=CHECK-MACHO3; RUN: llc -mtriple=aarch64-none-linux-gnu -disable-post-ra -mattr=-fp-armv8 -verify-machineinstrs < %s | FileCheck --check-prefix=CHECK-NOFP-ARM64 %s4 5declare void @use_addr(ptr)6 7define void @test_simple_alloca(i64 %n) {8; CHECK-LABEL: test_simple_alloca:9 10 %buf = alloca i8, i64 %n11 ; Make sure we align the stack change to 16 bytes:12; CHECK: {{mov|add}} x2913; CHECK: mov [[TMP:x[0-9]+]], sp14; CHECK: add [[SPDELTA_TMP:x[0-9]+]], x0, #1515; CHECK: and [[SPDELTA:x[0-9]+]], [[SPDELTA_TMP]], #0xfffffffffffffff016 17 ; Make sure we change SP. It would be surprising if anything but x0 were used18 ; for the final sp, but it could be if it was then moved into x0.19; CHECK: sub [[NEWSP:x[0-9]+]], [[TMP]], [[SPDELTA]]20; CHECK: mov sp, [[NEWSP]]21 22 call void @use_addr(ptr %buf)23; CHECK: bl use_addr24 25 ret void26 ; Make sure epilogue restores sp from fp27; CHECK: {{sub|mov}} sp, x2928; CHECK: ret29}30 31declare void @use_addr_loc(ptr, ptr)32 33define i64 @test_alloca_with_local(i64 %n) {34; CHECK-LABEL: test_alloca_with_local:35; CHECK-DAG: sub sp, sp, [[LOCAL_STACK:#[0-9]+]]36; CHECK-DAG: {{mov|add}} x29, sp37 38 %loc = alloca i6439 %buf = alloca i8, i64 %n40 ; Make sure we align the stack change to 16 bytes:41; CHECK: mov [[TMP:x[0-9]+]], sp42; CHECK: add [[SPDELTA_TMP:x[0-9]+]], x0, #1543; CHECK: and [[SPDELTA:x[0-9]+]], [[SPDELTA_TMP]], #0xfffffffffffffff044 45 ; Make sure we change SP. It would be surprising if anything but x0 were used46 ; for the final sp, but it could be if it was then moved into x0.47; CHECK: sub [[NEWSP:x[0-9]+]], [[TMP]], [[SPDELTA]]48; CHECK: mov sp, [[NEWSP]]49 50; CHECK: sub {{x[0-9]+}}, x29, #[[LOC_FROM_FP:[0-9]+]]51 52 call void @use_addr_loc(ptr %buf, ptr %loc)53; CHECK: bl use_addr54 55 %val = load i64, ptr %loc56 57; CHECK: ldur x0, [x29, #-[[LOC_FROM_FP]]]58 59 ret i64 %val60 ; Make sure epilogue restores sp from fp61; CHECK: {{sub|mov}} sp, x2962; CHECK: ret63}64 65%struct.__va_list = type { ptr, ptr, ptr, i32, i32 }66 67define void @test_variadic_alloca(i64 %n, ...) {68; CHECK-LABEL: test_variadic_alloca:69 70; [...]71 72 73; CHECK-NOFP-AARCH64: sub sp, sp, #8074; CHECK-NOFP-AARCH64: stp x29, x30, [sp, #64]75; CHECK-NOFP-AARCH64: add x29, sp, #6476; CHECK-NOFP-AARCH64: sub [[TMP:x[0-9]+]], x29, #6477; CHECK-NOFP-AARCH64: add x8, [[TMP]], #078 79 80; CHECK: stp x29, x30, [sp, #-16]!81; CHECK: mov x29, sp82; CHECK: sub sp, sp, #22483; CHECK-DAG: stp q6, q7, [x29, #-128]84; [...]85; CHECK-DAG: stp q2, q3, [x29, #-192]86 87; CHECK-DAG: stp x5, x6, [x29, #-56]88; [...]89; CHECK-DAG: stp x1, x2, [x29, #-88]90 91; CHECK-NOFP-ARM64: stp x29, x30, [sp, #-16]!92; CHECK-NOFP-ARM64: mov x29, sp93; CHECK-NOFP-ARM64: sub sp, sp, #1694; CHECK-NOFP-ARM64-DAG: stp x5, x6, [x29, #-56]95; [...]96; CHECK-NOFP-ARM64-DAG: stp x3, x4, [x29, #-72]97; [...]98; CHECK-NOFP-ARM64-DAG: stp x1, x2, [x29, #-88]99; [...]100; CHECK-NOFP-ARM64: mov x8, sp101 %valist = alloca %struct.__va_list102 call void @llvm.va_start(ptr %valist)103 %addr = alloca i8, i64 %n104 105 call void @use_addr(ptr %addr)106; CHECK: bl use_addr107 108 call void @llvm.va_end(ptr %valist)109 ret void110 111; CHECK-NOFP-AARCH64: sub sp, x29, #64112; CHECK-NOFP-AARCH64: ldp x29, x30, [sp, #64]113; CHECK-NOFP-AARCH64: add sp, sp, #80114 115; CHECK-NOFP-ARM64: mov sp, x29116; CHECK-NOFP-ARM64: ldp x29, x30, [sp], #16117}118 119define void @test_alloca_large_frame(i64 %n) {120; CHECK-LABEL: test_alloca_large_frame:121; CHECK-MACHO-LABEL: test_alloca_large_frame:122 123 124; CHECK: stp x29, x30, [sp, #-32]!125; CHECK: stp x28, x19, [sp, #16]126; CHECK: mov x29, sp127; CHECK: sub sp, sp, #1953, lsl #12128; CHECK: sub sp, sp, #512129 130; CHECK-MACHO: stp x20, x19, [sp, #-32]!131; CHECK-MACHO: stp x29, x30, [sp, #16]132; CHECK-MACHO: add x29, sp, #16133; CHECK-MACHO: sub sp, sp, #1953, lsl #12134; CHECK-MACHO: sub sp, sp, #512135 136 %addr1 = alloca i8, i64 %n137 %addr2 = alloca i64, i64 1000000138 139 call void @use_addr_loc(ptr %addr1, ptr %addr2)140 141 ret void142 143; CHECK: mov sp, x29144; CHECK: ldp x28, x19, [sp, #16]145; CHECK: ldp x29, x30, [sp], #32146 147; CHECK-MACHO: sub sp, x29, #16148; CHECK-MACHO: ldp x29, x30, [sp, #16]149; CHECK-MACHO: ldp x20, x19, [sp], #32150}151 152declare ptr @llvm.stacksave()153declare void @llvm.stackrestore(ptr)154 155define void @test_scoped_alloca(i64 %n) {156; CHECK-LABEL: test_scoped_alloca:157 158 %sp = call ptr @llvm.stacksave()159; CHECK: mov x29, sp160; CHECK: mov [[SAVED_SP:x[0-9]+]], sp161; CHECK: mov [[OLDSP:x[0-9]+]], sp162 163 %addr = alloca i8, i64 %n164; CHECK: and [[SPDELTA:x[0-9]+]], {{x[0-9]+}}, #0xfffffffffffffff0165; CHECK-DAG: sub [[NEWSP:x[0-9]+]], [[OLDSP]], [[SPDELTA]]166; CHECK: mov sp, [[NEWSP]]167 168 call void @use_addr(ptr %addr)169; CHECK: bl use_addr170 171 call void @llvm.stackrestore(ptr %sp)172; CHECK: mov sp, [[SAVED_SP]]173 174 ret void175}176