xref: /llvm-project/llvm/test/CodeGen/AArch64/stack-probing-shrink-wrap.mir (revision 09bc6abba6e226ad5e9d18d4365690d6f04de21a)
1# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py UTC_ARGS: --version 4
2# RUN: llc %s --start-before=shrink-wrap --stop-after=prologepilog -o - | FileCheck %s
3--- |
4  target triple = "aarch64-linux"
5
6  define void @f(i32 %n) #0 {
7  entry:
8    %a = alloca i8, i32 150000, align 8
9    %c0 = icmp sle i32 %n, 1
10    br i1 %c0, label %if.then1, label %exit
11
12  if.then1:                                         ; preds = %entry
13    %0 = icmp sle i32 %n, 1
14    %v = select i1 %0, i32 0, i32 1
15    call void @g(ptr %a, i32 %v)
16    br label %exit
17
18  exit:                                             ; preds = %if.then1, %entry
19    ret void
20  }
21
22  declare void @g(...)
23
24  attributes #0 = { nounwind "probe-stack"="inline-asm" "stack-probe-size"="4096" }
25
26...
27---
28name:            f
29alignment:       4
30tracksRegLiveness: true
31liveins:
32  - { reg: '$w0', virtual-reg: '' }
33frameInfo:
34  adjustsStack:    true
35  localFrameSize:  150000
36stack:
37  - { id: 0, name: a, type: default, offset: 0, size: 150000, alignment: 8,
38      stack-id: default, callee-saved-register: '', callee-saved-restored: true,
39      local-offset: -150000, debug-info-variable: '', debug-info-expression: '',
40      debug-info-location: '' }
41entry_values:    []
42callSites:       []
43debugValueSubstitutions: []
44constants:       []
45machineFunctionInfo: {}
46body:             |
47  ; CHECK-LABEL: name: f
48  ; CHECK: bb.0.entry:
49  ; CHECK-NEXT:   successors: %bb.3(0x80000000)
50  ; CHECK-NEXT:   liveins: $w0, $lr
51  ; CHECK-NEXT: {{  $}}
52  ; CHECK-NEXT:   early-clobber $sp = frame-setup STPXpre killed $fp, killed $lr, $sp, -2 :: (store (s64) into %stack.2), (store (s64) into %stack.1)
53  ; CHECK-NEXT:   $x9 = frame-setup SUBXri $sp, 36, 12
54  ; CHECK-NEXT: {{  $}}
55  ; CHECK-NEXT: bb.3.entry:
56  ; CHECK-NEXT:   successors: %bb.4(0x40000000), %bb.3(0x40000000)
57  ; CHECK-NEXT:   liveins: $w0, $x9
58  ; CHECK-NEXT: {{  $}}
59  ; CHECK-NEXT:   $sp = frame-setup SUBXri $sp, 1, 12
60  ; CHECK-NEXT:   frame-setup STRXui $xzr, $sp, 0
61  ; CHECK-NEXT:   $xzr = frame-setup SUBSXrx64 $sp, $x9, 24, implicit-def $nzcv
62  ; CHECK-NEXT:   frame-setup Bcc 1, %bb.3, implicit $nzcv
63  ; CHECK-NEXT: {{  $}}
64  ; CHECK-NEXT: bb.4.entry:
65  ; CHECK-NEXT:   successors: %bb.1(0x40000000), %bb.2(0x40000000)
66  ; CHECK-NEXT:   liveins: $w0
67  ; CHECK-NEXT: {{  $}}
68  ; CHECK-NEXT:   $sp = frame-setup SUBXri $sp, 2544, 0
69  ; CHECK-NEXT:   frame-setup STRXui $xzr, $sp, 0
70  ; CHECK-NEXT:   dead $wzr = SUBSWri killed renamable $w0, 1, 0, implicit-def $nzcv
71  ; CHECK-NEXT:   Bcc 12, %bb.2, implicit $nzcv
72  ; CHECK-NEXT:   B %bb.1
73  ; CHECK-NEXT: {{  $}}
74  ; CHECK-NEXT: bb.1.if.then1:
75  ; CHECK-NEXT:   successors: %bb.2(0x80000000)
76  ; CHECK-NEXT:   liveins: $nzcv
77  ; CHECK-NEXT: {{  $}}
78  ; CHECK-NEXT:   renamable $w1 = CSINCWr $wzr, $wzr, 13, implicit killed $nzcv
79  ; CHECK-NEXT:   $x0 = ADDXri $sp, 0, 0
80  ; CHECK-NEXT:   BL @g, csr_aarch64_aapcs, implicit-def dead $lr, implicit $sp, implicit $x0, implicit $w1, implicit-def $sp
81  ; CHECK-NEXT: {{  $}}
82  ; CHECK-NEXT: bb.2.exit:
83  ; CHECK-NEXT:   $sp = frame-destroy ADDXri $sp, 36, 12
84  ; CHECK-NEXT:   $sp = frame-destroy ADDXri $sp, 2544, 0
85  ; CHECK-NEXT:   early-clobber $sp, $fp, $lr = frame-destroy LDPXpost $sp, 2 :: (load (s64) from %stack.2), (load (s64) from %stack.1)
86  ; CHECK-NEXT:   RET_ReallyLR
87  bb.0.entry:
88    successors: %bb.1(0x40000000), %bb.2(0x40000000)
89    liveins: $w0
90
91    dead $wzr = SUBSWri killed renamable $w0, 1, 0, implicit-def $nzcv
92    Bcc 12, %bb.2, implicit $nzcv
93    B %bb.1
94
95  bb.1.if.then1:
96    successors: %bb.2(0x80000000)
97    liveins: $nzcv
98
99    renamable $w1 = CSINCWr $wzr, $wzr, 13, implicit killed $nzcv
100    ADJCALLSTACKDOWN 0, 0, implicit-def dead $sp, implicit $sp
101    $x0 = ADDXri %stack.0.a, 0, 0
102    BL @g, csr_aarch64_aapcs, implicit-def dead $lr, implicit $sp, implicit $x0, implicit $w1, implicit-def $sp
103    ADJCALLSTACKUP 0, 0, implicit-def dead $sp, implicit $sp
104
105  bb.2.exit:
106    RET_ReallyLR
107
108...
109