1; RUN: llc -mtriple=aarch64-linux-gnu -verify-machineinstrs -o - %s | FileCheck %s --check-prefix=CHECK 2; RUN: llc -mtriple=aarch64-none-linux-gnu -mattr=-fp-armv8 -verify-machineinstrs < %s | FileCheck --check-prefix=CHECK-NOFP-ARM64 %s 3 4declare void @use_addr(i8*) 5 6define void @test_simple_alloca(i64 %n) { 7; CHECK-LABEL: test_simple_alloca: 8 9 %buf = alloca i8, i64 %n 10 ; Make sure we align the stack change to 16 bytes: 11; CHECK: {{mov|add}} x29 12; CHECK: mov [[TMP:x[0-9]+]], sp 13; CHECK: add [[SPDELTA_TMP:x[0-9]+]], x0, #15 14; CHECK: and [[SPDELTA:x[0-9]+]], [[SPDELTA_TMP]], #0xfffffffffffffff0 15 16 ; Make sure we change SP. It would be surprising if anything but x0 were used 17 ; for the final sp, but it could be if it was then moved into x0. 18; CHECK: sub [[NEWSP:x[0-9]+]], [[TMP]], [[SPDELTA]] 19; CHECK: mov sp, [[NEWSP]] 20 21 call void @use_addr(i8* %buf) 22; CHECK: bl use_addr 23 24 ret void 25 ; Make sure epilogue restores sp from fp 26; CHECK: {{sub|mov}} sp, x29 27; CHECK: ret 28} 29 30declare void @use_addr_loc(i8*, i64*) 31 32define i64 @test_alloca_with_local(i64 %n) { 33; CHECK-LABEL: test_alloca_with_local: 34; CHECK-DAG: sub sp, sp, [[LOCAL_STACK:#[0-9]+]] 35; CHECK-DAG: {{mov|add}} x29, sp 36 37 %loc = alloca i64 38 %buf = alloca i8, i64 %n 39 ; Make sure we align the stack change to 16 bytes: 40; CHECK: mov [[TMP:x[0-9]+]], sp 41; CHECK: add [[SPDELTA_TMP:x[0-9]+]], x0, #15 42; CHECK: and [[SPDELTA:x[0-9]+]], [[SPDELTA_TMP]], #0xfffffffffffffff0 43 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: sub [[NEWSP:x[0-9]+]], [[TMP]], [[SPDELTA]] 47; CHECK: mov sp, [[NEWSP]] 48 49; CHECK: sub {{x[0-9]+}}, x29, #[[LOC_FROM_FP:[0-9]+]] 50 51 call void @use_addr_loc(i8* %buf, i64* %loc) 52; CHECK: bl use_addr 53 54 %val = load i64, i64* %loc 55 56; CHECK: ldur x0, [x29, #-[[LOC_FROM_FP]]] 57 58 ret i64 %val 59 ; Make sure epilogue restores sp from fp 60; CHECK: {{sub|mov}} sp, x29 61; CHECK: ret 62} 63 64define void @test_variadic_alloca(i64 %n, ...) { 65; CHECK-LABEL: test_variadic_alloca: 66 67; [...] 68 69 70; CHECK-NOFP-AARCH64: sub sp, sp, #80 71; CHECK-NOFP-AARCH64: stp x29, x30, [sp, #64] 72; CHECK-NOFP-AARCH64: add x29, sp, #64 73; CHECK-NOFP-AARCH64: sub [[TMP:x[0-9]+]], x29, #64 74; CHECK-NOFP-AARCH64: add x8, [[TMP]], #0 75 76 77; CHECK: stp x29, x30, [sp, #-16]! 78; CHECK: mov x29, sp 79; CHECK: sub sp, sp, #192 80; CHECK: stp q6, q7, [x29, #-96] 81; [...] 82; CHECK: stp q0, q1, [x29, #-192] 83 84; CHECK: stp x6, x7, [x29, #-16] 85; [...] 86; CHECK: stp x2, x3, [x29, #-48] 87 88; CHECK-NOFP-ARM64: stp x29, x30, [sp, #-16]! 89; CHECK-NOFP-ARM64: mov x29, sp 90; CHECK-NOFP-ARM64: sub sp, sp, #64 91; CHECK-NOFP-ARM64: stp x6, x7, [x29, #-16] 92; [...] 93; CHECK-NOFP-ARM64: stp x4, x5, [x29, #-32] 94; [...] 95; CHECK-NOFP-ARM64: stp x2, x3, [x29, #-48] 96; [...] 97; CHECK-NOFP-ARM64: mov x8, sp 98 99 %addr = alloca i8, i64 %n 100 101 call void @use_addr(i8* %addr) 102; CHECK: bl use_addr 103 104 ret void 105 106; CHECK-NOFP-AARCH64: sub sp, x29, #64 107; CHECK-NOFP-AARCH64: ldp x29, x30, [sp, #64] 108; CHECK-NOFP-AARCH64: add sp, sp, #80 109 110; CHECK-NOFP-ARM64: mov sp, x29 111; CHECK-NOFP-ARM64: ldp x29, x30, [sp], #16 112} 113 114define void @test_alloca_large_frame(i64 %n) { 115; CHECK-LABEL: test_alloca_large_frame: 116 117 118; CHECK: stp x20, x19, [sp, #-32]! 119; CHECK: stp x29, x30, [sp, #16] 120; CHECK: add x29, sp, #16 121; CHECK: sub sp, sp, #1953, lsl #12 122; CHECK: sub sp, sp, #512 123 124 %addr1 = alloca i8, i64 %n 125 %addr2 = alloca i64, i64 1000000 126 127 call void @use_addr_loc(i8* %addr1, i64* %addr2) 128 129 ret void 130 131; CHECK: sub sp, x29, #16 132; CHECK: ldp x29, x30, [sp, #16] 133; CHECK: ldp x20, x19, [sp], #32 134} 135 136declare i8* @llvm.stacksave() 137declare void @llvm.stackrestore(i8*) 138 139define void @test_scoped_alloca(i64 %n) { 140; CHECK-LABEL: test_scoped_alloca: 141 142 %sp = call i8* @llvm.stacksave() 143; CHECK: mov [[SAVED_SP:x[0-9]+]], sp 144; CHECK: mov [[OLDSP:x[0-9]+]], sp 145 146 %addr = alloca i8, i64 %n 147; CHECK: and [[SPDELTA:x[0-9]+]], {{x[0-9]+}}, #0xfffffffffffffff0 148; CHECK-DAG: sub [[NEWSP:x[0-9]+]], [[OLDSP]], [[SPDELTA]] 149; CHECK: mov sp, [[NEWSP]] 150 151 call void @use_addr(i8* %addr) 152; CHECK: bl use_addr 153 154 call void @llvm.stackrestore(i8* %sp) 155; CHECK: mov sp, [[SAVED_SP]] 156 157 ret void 158} 159