xref: /llvm-project/llvm/test/CodeGen/AMDGPU/use_restore_frame_reg.mir (revision 11b040192640ef3b1f481124c440f464ed6ec86a) 
1# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
2# RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx900 -verify-machineinstrs -run-pass=prologepilog %s -o - | FileCheck --check-prefix=MUBUF %s
3# RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx900 -verify-machineinstrs -run-pass=prologepilog -mattr=+enable-flat-scratch %s -o - | FileCheck --check-prefix=FLATSCR %s
4
5---
6name: use_restore_frame_reg
7tracksRegLiveness: true
8
9stack:
10  - { id: 0, type: default, offset: 0, size: 4, alignment: 8192 }
11  - { id: 1, type: default, offset: 0, size: 4, alignment: 8192 }
12  - { id: 2, type: default, offset: 0, size: 4, alignment: 8192 }
13  - { id: 3, type: default, offset: 0, size: 4, alignment: 8192 }
14  - { id: 4, type: default, offset: 0, size: 4, alignment: 8192 }
15  - { id: 5, type: default, offset: 0, size: 4, alignment: 8192 }
16  - { id: 6, type: default, offset: 0, size: 4, alignment: 8192 }
17  - { id: 7, type: default, offset: 0, size: 4, alignment: 8192 }
18  - { id: 8, type: default, offset: 0, size: 4, alignment: 8192 }
19  - { id: 9, type: default, offset: 0, size: 4, alignment: 8192 }
20  - { id: 10, type: default, offset: 0, size: 4, alignment: 8192 }
21  - { id: 11, type: default, offset: 0, size: 4, alignment: 8192 }
22  - { id: 12, type: default, offset: 0, size: 4, alignment: 8192 }
23  - { id: 13, type: default, offset: 0, size: 4, alignment: 8192 }
24  - { id: 14, type: default, offset: 0, size: 4, alignment: 8192 }
25  - { id: 15, type: default, offset: 0, size: 4, alignment: 8192 }
26  - { id: 16, type: default, offset: 0, size: 4, alignment: 8192 }
27  - { id: 17, type: default, offset: 0, size: 4, alignment: 8192 }
28  - { id: 18, type: default, offset: 0, size: 4, alignment: 8192 }
29
30machineFunctionInfo:
31  isEntryFunction: false
32  scratchRSrcReg:  $sgpr0_sgpr1_sgpr2_sgpr3
33  frameOffsetReg:  $sgpr33
34  stackPtrOffsetReg:  $sgpr32
35
36body:             |
37  ; MUBUF-LABEL: name: use_restore_frame_reg
38  ; MUBUF: bb.0:
39  ; MUBUF-NEXT:   successors: %bb.2(0x40000000), %bb.1(0x40000000)
40  ; MUBUF-NEXT:   liveins: $vgpr1, $vgpr2
41  ; MUBUF-NEXT: {{  $}}
42  ; MUBUF-NEXT:   $sgpr4 = COPY $sgpr33
43  ; MUBUF-NEXT:   $sgpr33 = frame-setup S_ADD_I32 $sgpr32, 524224, implicit-def $scc
44  ; MUBUF-NEXT:   $sgpr33 = frame-setup S_AND_B32 killed $sgpr33, 4294443008, implicit-def dead $scc
45  ; MUBUF-NEXT:   $sgpr6_sgpr7 = S_XOR_SAVEEXEC_B64 -1, implicit-def $exec, implicit-def dead $scc, implicit $exec
46  ; MUBUF-NEXT:   $sgpr5 = S_ADD_I32 $sgpr33, 9961728, implicit-def dead $scc
47  ; MUBUF-NEXT:   BUFFER_STORE_DWORD_OFFSET $vgpr2, $sgpr0_sgpr1_sgpr2_sgpr3, killed $sgpr5, 0, 0, 0, implicit $exec :: (store (s32) into %stack.20, addrspace 5)
48  ; MUBUF-NEXT:   $exec = S_MOV_B64 killed $sgpr6_sgpr7
49  ; MUBUF-NEXT:   $vgpr2 = SI_SPILL_S32_TO_VGPR $sgpr4, 0, undef $vgpr2
50  ; MUBUF-NEXT:   $vgpr2 = SI_SPILL_S32_TO_VGPR $sgpr34, 1, undef $vgpr2
51  ; MUBUF-NEXT:   $sgpr34 = frame-setup COPY $sgpr32
52  ; MUBUF-NEXT:   $sgpr32 = frame-setup S_ADD_I32 $sgpr32, 11010048, implicit-def dead $scc
53  ; MUBUF-NEXT:   S_CMP_EQ_U32 0, 0, implicit-def $scc
54  ; MUBUF-NEXT:   S_NOP 0, implicit-def $sgpr4, implicit-def $sgpr5, implicit-def $sgpr6, implicit-def $sgpr7, implicit-def $sgpr8, implicit-def $sgpr9, implicit-def $sgpr10, implicit-def $sgpr11, implicit-def $sgpr12, implicit-def $sgpr13, implicit-def $sgpr14, implicit-def $sgpr15, implicit-def $sgpr16, implicit-def $sgpr17, implicit-def $sgpr18, implicit-def $sgpr19, implicit-def $sgpr20, implicit-def $sgpr21, implicit-def $sgpr22, implicit-def $sgpr23, implicit-def $sgpr24, implicit-def $sgpr25, implicit-def $sgpr26, implicit-def $sgpr27, implicit-def $sgpr28, implicit-def $sgpr29, implicit-def $sgpr30, implicit-def $sgpr31, implicit-def $vcc
55  ; MUBUF-NEXT:   $vgpr0 = V_LSHRREV_B32_e64 6, $sgpr33, implicit $exec
56  ; MUBUF-NEXT:   $vgpr0 = V_ADD_U32_e32 8192, killed $vgpr0, implicit $exec
57  ; MUBUF-NEXT:   $vgpr3 = V_LSHRREV_B32_e64 6, $sgpr33, implicit $exec
58  ; MUBUF-NEXT:   $vgpr3 = V_ADD_U32_e32 155648, killed $vgpr3, implicit $exec
59  ; MUBUF-NEXT:   $vgpr0 = V_OR_B32_e32 killed $vgpr3, $vgpr1, implicit $exec, implicit $sgpr4, implicit $sgpr5, implicit $sgpr6, implicit $sgpr7, implicit $sgpr8, implicit $sgpr9, implicit $sgpr10, implicit $sgpr11, implicit $sgpr12, implicit $sgpr13, implicit $sgpr14, implicit $sgpr15, implicit $sgpr16, implicit $sgpr17, implicit $sgpr18, implicit $sgpr19, implicit $sgpr20, implicit $sgpr21, implicit $sgpr22, implicit $sgpr23, implicit $sgpr24, implicit $sgpr25, implicit $sgpr26, implicit $sgpr27, implicit $sgpr28, implicit $sgpr29, implicit $sgpr30, implicit $sgpr31
60  ; MUBUF-NEXT:   S_CBRANCH_VCCNZ %bb.2, implicit $vcc
61  ; MUBUF-NEXT:   S_CBRANCH_SCC1 %bb.2, implicit $scc
62  ; MUBUF-NEXT: {{  $}}
63  ; MUBUF-NEXT: bb.1:
64  ; MUBUF-NEXT:   successors: %bb.2(0x80000000)
65  ; MUBUF-NEXT:   liveins: $vgpr2
66  ; MUBUF-NEXT: {{  $}}
67  ; MUBUF-NEXT:   S_NOP 0
68  ; MUBUF-NEXT: {{  $}}
69  ; MUBUF-NEXT: bb.2:
70  ; MUBUF-NEXT:   liveins: $vgpr2
71  ; MUBUF-NEXT: {{  $}}
72  ; MUBUF-NEXT:   $sgpr32 = frame-destroy COPY $sgpr34
73  ; MUBUF-NEXT:   $sgpr4 = SI_RESTORE_S32_FROM_VGPR $vgpr2, 0
74  ; MUBUF-NEXT:   $sgpr34 = SI_RESTORE_S32_FROM_VGPR $vgpr2, 1
75  ; MUBUF-NEXT:   $sgpr6_sgpr7 = S_XOR_SAVEEXEC_B64 -1, implicit-def $exec, implicit-def dead $scc, implicit $exec
76  ; MUBUF-NEXT:   $sgpr5 = S_ADD_I32 $sgpr33, 9961728, implicit-def dead $scc
77  ; MUBUF-NEXT:   $vgpr2 = BUFFER_LOAD_DWORD_OFFSET $sgpr0_sgpr1_sgpr2_sgpr3, killed $sgpr5, 0, 0, 0, implicit $exec :: (load (s32) from %stack.20, addrspace 5)
78  ; MUBUF-NEXT:   $exec = S_MOV_B64 killed $sgpr6_sgpr7
79  ; MUBUF-NEXT:   $sgpr33 = COPY $sgpr4
80  ; MUBUF-NEXT:   S_ENDPGM 0
81  ;
82  ; FLATSCR-LABEL: name: use_restore_frame_reg
83  ; FLATSCR: bb.0:
84  ; FLATSCR-NEXT:   successors: %bb.2(0x40000000), %bb.1(0x40000000)
85  ; FLATSCR-NEXT:   liveins: $vgpr1, $vgpr2
86  ; FLATSCR-NEXT: {{  $}}
87  ; FLATSCR-NEXT:   $sgpr4 = COPY $sgpr33
88  ; FLATSCR-NEXT:   $sgpr33 = frame-setup S_ADD_I32 $sgpr32, 8191, implicit-def $scc
89  ; FLATSCR-NEXT:   $sgpr33 = frame-setup S_AND_B32 killed $sgpr33, 4294959104, implicit-def dead $scc
90  ; FLATSCR-NEXT:   $sgpr6_sgpr7 = S_XOR_SAVEEXEC_B64 -1, implicit-def $exec, implicit-def dead $scc, implicit $exec
91  ; FLATSCR-NEXT:   $sgpr5 = S_ADD_I32 $sgpr33, 155652, implicit-def dead $scc
92  ; FLATSCR-NEXT:   SCRATCH_STORE_DWORD_SADDR $vgpr2, killed $sgpr5, 0, 0, implicit $exec, implicit $flat_scr :: (store (s32) into %stack.20, addrspace 5)
93  ; FLATSCR-NEXT:   $exec = S_MOV_B64 killed $sgpr6_sgpr7
94  ; FLATSCR-NEXT:   $vgpr2 = SI_SPILL_S32_TO_VGPR $sgpr4, 0, undef $vgpr2
95  ; FLATSCR-NEXT:   $vgpr2 = SI_SPILL_S32_TO_VGPR $sgpr34, 1, undef $vgpr2
96  ; FLATSCR-NEXT:   $sgpr34 = frame-setup COPY $sgpr32
97  ; FLATSCR-NEXT:   $sgpr32 = frame-setup S_ADD_I32 $sgpr32, 172032, implicit-def dead $scc
98  ; FLATSCR-NEXT:   S_CMP_EQ_U32 0, 0, implicit-def $scc
99  ; FLATSCR-NEXT:   S_NOP 0, implicit-def $sgpr4, implicit-def $sgpr5, implicit-def $sgpr6, implicit-def $sgpr7, implicit-def $sgpr8, implicit-def $sgpr9, implicit-def $sgpr10, implicit-def $sgpr11, implicit-def $sgpr12, implicit-def $sgpr13, implicit-def $sgpr14, implicit-def $sgpr15, implicit-def $sgpr16, implicit-def $sgpr17, implicit-def $sgpr18, implicit-def $sgpr19, implicit-def $sgpr20, implicit-def $sgpr21, implicit-def $sgpr22, implicit-def $sgpr23, implicit-def $sgpr24, implicit-def $sgpr25, implicit-def $sgpr26, implicit-def $sgpr27, implicit-def $sgpr28, implicit-def $sgpr29, implicit-def $sgpr30, implicit-def $sgpr31, implicit-def $vcc
100  ; FLATSCR-NEXT:   $sgpr33 = S_ADDC_U32 $sgpr33, 8192, implicit-def $scc, implicit $scc
101  ; FLATSCR-NEXT:   S_BITCMP1_B32 $sgpr33, 0, implicit-def $scc
102  ; FLATSCR-NEXT:   $sgpr33 = S_BITSET0_B32 0, $sgpr33
103  ; FLATSCR-NEXT:   $vgpr0 = V_MOV_B32_e32 $sgpr33, implicit $exec
104  ; FLATSCR-NEXT:   $sgpr33 = S_ADDC_U32 $sgpr33, -8192, implicit-def $scc, implicit $scc
105  ; FLATSCR-NEXT:   S_BITCMP1_B32 $sgpr33, 0, implicit-def $scc
106  ; FLATSCR-NEXT:   $sgpr33 = S_BITSET0_B32 0, $sgpr33
107  ; FLATSCR-NEXT:   $sgpr33 = S_ADDC_U32 $sgpr33, 155648, implicit-def $scc, implicit $scc
108  ; FLATSCR-NEXT:   S_BITCMP1_B32 $sgpr33, 0, implicit-def $scc
109  ; FLATSCR-NEXT:   $sgpr33 = S_BITSET0_B32 0, $sgpr33
110  ; FLATSCR-NEXT:   $vgpr0 = V_OR_B32_e32 $sgpr33, $vgpr1, implicit $exec, implicit $sgpr4, implicit $sgpr5, implicit $sgpr6, implicit $sgpr7, implicit $sgpr8, implicit $sgpr9, implicit $sgpr10, implicit $sgpr11, implicit $sgpr12, implicit $sgpr13, implicit $sgpr14, implicit $sgpr15, implicit $sgpr16, implicit $sgpr17, implicit $sgpr18, implicit $sgpr19, implicit $sgpr20, implicit $sgpr21, implicit $sgpr22, implicit $sgpr23, implicit $sgpr24, implicit $sgpr25, implicit $sgpr26, implicit $sgpr27, implicit $sgpr28, implicit $sgpr29, implicit $sgpr30, implicit $sgpr31
111  ; FLATSCR-NEXT:   $sgpr33 = S_ADDC_U32 $sgpr33, -155648, implicit-def $scc, implicit $scc
112  ; FLATSCR-NEXT:   S_BITCMP1_B32 $sgpr33, 0, implicit-def $scc
113  ; FLATSCR-NEXT:   $sgpr33 = S_BITSET0_B32 0, $sgpr33
114  ; FLATSCR-NEXT:   S_CBRANCH_VCCNZ %bb.2, implicit $vcc
115  ; FLATSCR-NEXT:   S_CBRANCH_SCC1 %bb.2, implicit $scc
116  ; FLATSCR-NEXT: {{  $}}
117  ; FLATSCR-NEXT: bb.1:
118  ; FLATSCR-NEXT:   successors: %bb.2(0x80000000)
119  ; FLATSCR-NEXT:   liveins: $vgpr2
120  ; FLATSCR-NEXT: {{  $}}
121  ; FLATSCR-NEXT:   S_NOP 0
122  ; FLATSCR-NEXT: {{  $}}
123  ; FLATSCR-NEXT: bb.2:
124  ; FLATSCR-NEXT:   liveins: $vgpr2
125  ; FLATSCR-NEXT: {{  $}}
126  ; FLATSCR-NEXT:   $sgpr32 = frame-destroy COPY $sgpr34
127  ; FLATSCR-NEXT:   $sgpr4 = SI_RESTORE_S32_FROM_VGPR $vgpr2, 0
128  ; FLATSCR-NEXT:   $sgpr34 = SI_RESTORE_S32_FROM_VGPR $vgpr2, 1
129  ; FLATSCR-NEXT:   $sgpr6_sgpr7 = S_XOR_SAVEEXEC_B64 -1, implicit-def $exec, implicit-def dead $scc, implicit $exec
130  ; FLATSCR-NEXT:   $sgpr5 = S_ADD_I32 $sgpr33, 155652, implicit-def dead $scc
131  ; FLATSCR-NEXT:   $vgpr2 = SCRATCH_LOAD_DWORD_SADDR killed $sgpr5, 0, 0, implicit $exec, implicit $flat_scr :: (load (s32) from %stack.20, addrspace 5)
132  ; FLATSCR-NEXT:   $exec = S_MOV_B64 killed $sgpr6_sgpr7
133  ; FLATSCR-NEXT:   $sgpr33 = COPY $sgpr4
134  ; FLATSCR-NEXT:   S_ENDPGM 0
135  bb.0:
136    liveins: $vgpr1
137
138    S_CMP_EQ_U32 0, 0, implicit-def $scc
139    S_NOP 0, implicit-def $sgpr4, implicit-def $sgpr5, implicit-def $sgpr6, implicit-def $sgpr7, implicit-def $sgpr8, implicit-def $sgpr9, implicit-def $sgpr10, implicit-def $sgpr11, implicit-def $sgpr12, implicit-def $sgpr13, implicit-def $sgpr14, implicit-def $sgpr15, implicit-def $sgpr16, implicit-def $sgpr17, implicit-def $sgpr18, implicit-def $sgpr19, implicit-def $sgpr20, implicit-def $sgpr21, implicit-def $sgpr22, implicit-def $sgpr23, implicit-def $sgpr24, implicit-def $sgpr25, implicit-def $sgpr26, implicit-def $sgpr27, implicit-def $sgpr28, implicit-def $sgpr29, implicit-def $sgpr30, implicit-def $sgpr31, implicit-def $vcc
140    $vgpr0 = V_MOV_B32_e32 %stack.0, implicit $exec
141    $vgpr0 = V_OR_B32_e32 %stack.18, $vgpr1, implicit $exec, implicit $sgpr4, implicit $sgpr5, implicit $sgpr6, implicit $sgpr7, implicit $sgpr8, implicit $sgpr9, implicit $sgpr10, implicit $sgpr11, implicit $sgpr12, implicit $sgpr13, implicit $sgpr14, implicit $sgpr15, implicit $sgpr16, implicit $sgpr17, implicit $sgpr18, implicit $sgpr19, implicit $sgpr20, implicit $sgpr21, implicit $sgpr22, implicit $sgpr23, implicit $sgpr24, implicit $sgpr25, implicit $sgpr26, implicit $sgpr27, implicit $sgpr28, implicit $sgpr29, implicit $sgpr30, implicit $sgpr31
142    S_CBRANCH_VCCNZ %bb.2, implicit $vcc
143    S_CBRANCH_SCC1 %bb.2, implicit $scc
144
145  bb.1:
146    S_NOP 0
147
148  bb.2:
149    S_ENDPGM 0
150
151...
152