[AVR] Fix an issue of writing 16-bit ports
[llvm-project.git] / llvm / test / CodeGen / AVR / dynalloca.ll
blob28c743c5bdc6766f4f67579fccf224616ea4be6b
1 ; RUN: llc < %s -march=avr | FileCheck %s
3 declare void @foo(i16*, i16*, i8*)
5 define void @test1(i16 %x) {
6 ; CHECK-LABEL: test1:
7 ; Frame setup, with frame pointer
8 ; CHECK: in r28, 61
9 ; CHECK: in r29, 62
10 ; CHECK: out 61, r28
11 ; allocate first dynalloca
12 ; CHECK: in {{.*}}, 61
13 ; CHECK: in {{.*}}, 62
14 ; CHECK: sub
15 ; CHECK: sbc
16 ; CHECK: in r0, 63
17 ; CHECK-NEXT: cli
18 ; CHECK-NEXT: out 62, {{.*}}
19 ; CHECK-NEXT: out 63, r0
20 ; CHECK-NEXT: out 61, {{.*}}
21 ; Test writes
22 ; CHECK: std Z+13, {{.*}}
23 ; CHECK: std Z+12, {{.*}}
24 ; CHECK: std Z+7, {{.*}}
25 ; CHECK-NOT: std
26 ; Test SP restore
27 ; CHECK: in r0, 63
28 ; CHECK-NEXT: cli
29 ; CHECK-NEXT: out 62, r29
30 ; CHECK-NEXT: out 63, r0
31 ; CHECK-NEXT: out 61, r28
32   %a = alloca [8 x i16]
33   %vla = alloca i16, i16 %x
34   %add = shl nsw i16 %x, 1
35   %vla1 = alloca i8, i16 %add
36   %arrayidx = getelementptr inbounds [8 x i16], [8 x i16]* %a, i16 0, i16 2
37   store i16 3, i16* %arrayidx
38   %arrayidx2 = getelementptr inbounds i16, i16* %vla, i16 6
39   store i16 4, i16* %arrayidx2
40   %arrayidx3 = getelementptr inbounds i8, i8* %vla1, i16 7
41   store i8 44, i8* %arrayidx3
42   %arraydecay = getelementptr inbounds [8 x i16], [8 x i16]* %a, i16 0, i16 0
43   call void @foo(i16* %arraydecay, i16* %vla, i8* %vla1)
44   ret void
47 declare void @foo2(i16*, i64, i64, i64)
49 ; Test that arguments are passed through pushes into the call instead of
50 ; allocating the call frame space in the prologue. Also test that SP is restored
51 ; after the call frame is restored and not before.
52 define void @dynalloca2(i16 %x) {
53 ; CHECK-LABEL: dynalloca2:
54 ; CHECK: in r28, 61
55 ; CHECK: in r29, 62
56 ; Allocate stack space for call
57 ; CHECK: in {{.*}}, 61
58 ; CHECK: in {{.*}}, 62
59 ; CHECK: subi
60 ; CHECK: sbci
61 ; CHECK: in r0, 63
62 ; CHECK-NEXT: cli
63 ; CHECK-NEXT: out 62, {{.*}}
64 ; CHECK-NEXT: out 63, r0
65 ; CHECK-NEXT: out 61, {{.*}}
66 ; Store values on the stack
67 ; CHECK: ldi r16, 0
68 ; CHECK: ldi r17, 0
69 ; CHECK: std Z+8, r17
70 ; CHECK: std Z+7, r16
71 ; CHECK: std Z+6, r17
72 ; CHECK: std Z+5, r16
73 ; CHECK: std Z+4, r17
74 ; CHECK: std Z+3, r16
75 ; CHECK: std Z+2, r17
76 ; CHECK: std Z+1, r16
77 ; CHECK: call
78 ; Call frame restore
79 ; CHECK-NEXT: in r30, 61
80 ; CHECK-NEXT: in r31, 62
81 ; CHECK-NEXT: adiw r30, 8
82 ; CHECK-NEXT: in r0, 63
83 ; CHECK-NEXT: cli
84 ; CHECK-NEXT: out 62, r31
85 ; CHECK-NEXT: out 63, r0
86 ; CHECK-NEXT: out 61, r30
87 ; SP restore
88 ; CHECK: in r0, 63
89 ; CHECK-NEXT: cli
90 ; CHECK-NEXT: out 62, r29
91 ; CHECK-NEXT: out 63, r0
92 ; CHECK-NEXT: out 61, r28
93   %vla = alloca i16, i16 %x
94   call void @foo2(i16* %vla, i64 0, i64 0, i64 0)
95   ret void
98 ; Test a function with a variable sized object but without any other need for a
99 ; frame pointer.
100 ; Allocas that are not placed in the entry block are considered variable sized
101 ; (they could be in a loop).
102 define void @dynalloca3() {
103 ; CHECK-LABEL: dynalloca3:
104 ; Read frame pointer
105 ; CHECK:      in r28, 61
106 ; CHECK-NEXT: in r29, 62
107 ; Allocate memory for the alloca
108 ; CHECK-NEXT: in r24, 61
109 ; CHECK-NEXT: in r25, 62
110 ; CHECK-NEXT: sbiw r24, 8
111 ; CHECK-NEXT: in r0, 63
112 ; CHECK-NEXT: cli
113 ; CHECK-NEXT: out 62, r25
114 ; CHECK-NEXT: out 63, r0
115 ; CHECK-NEXT: out 61, r24
116 ; Restore frame pointer
117 ; CHECK-NEXT: in r0, 63
118 ; CHECK-NEXT: cli
119 ; CHECK-NEXT: out 62, r29
120 ; CHECK-NEXT: out 63, r0
121 ; CHECK-NEXT: out 61, r28
122   br label %1
124   %a = alloca i64
125   ret void