[AMDGPU] Update compute program resource registers for GFX12 (#75911)
[llvm-project.git] / llvm / test / MC / AMDGPU / hsa-gfx12-v4.s
blob186d98f78b986c1eb7aad63a65c585a4a8a018a9
1 // RUN: llvm-mc -triple amdgcn-amd-amdhsa -mcpu=gfx1200 --amdhsa-code-object-version=4 < %s | FileCheck --check-prefix=ASM %s
2 // RUN: llvm-mc -triple amdgcn-amd-amdhsa -mcpu=gfx1200 --amdhsa-code-object-version=4 -filetype=obj < %s > %t
3 // RUN: llvm-readelf -S -r -s %t | FileCheck --check-prefix=READOBJ %s
4 // RUN: llvm-objdump -s -j .rodata %t | FileCheck --check-prefix=OBJDUMP %s
6 // READOBJ: Section Headers
7 // READOBJ: .text PROGBITS {{[0-9a-f]+}} {{[0-9a-f]+}} {{[0-9a-f]+}} {{[0-9]+}} AX {{[0-9]+}} {{[0-9]+}} 256
8 // READOBJ: .rodata PROGBITS {{[0-9a-f]+}} {{[0-9a-f]+}} 000100 {{[0-9]+}} A {{[0-9]+}} {{[0-9]+}} 64
10 // READOBJ: Relocation section '.rela.rodata' at offset
11 // READOBJ: 0000000000000010 {{[0-9a-f]+}}00000005 R_AMDGPU_REL64 0000000000000000 .text + 10
12 // READOBJ: 0000000000000050 {{[0-9a-f]+}}00000005 R_AMDGPU_REL64 0000000000000000 .text + 110
13 // READOBJ: 0000000000000090 {{[0-9a-f]+}}00000005 R_AMDGPU_REL64 0000000000000000 .text + 210
14 // READOBJ: 00000000000000d0 {{[0-9a-f]+}}00000005 R_AMDGPU_REL64 0000000000000000 .text + 310
16 // READOBJ: Symbol table '.symtab' contains {{[0-9]+}} entries:
17 // READOBJ: 0000000000000000 0 FUNC LOCAL PROTECTED 2 minimal
18 // READOBJ-NEXT: 0000000000000100 0 FUNC LOCAL PROTECTED 2 complete
19 // READOBJ-NEXT: 0000000000000200 0 FUNC LOCAL PROTECTED 2 special_sgpr
20 // READOBJ-NEXT: 0000000000000300 0 FUNC LOCAL PROTECTED 2 disabled_user_sgpr
21 // READOBJ-NEXT: 0000000000000000 64 OBJECT LOCAL DEFAULT 3 minimal.kd
22 // READOBJ-NEXT: 0000000000000040 64 OBJECT LOCAL DEFAULT 3 complete.kd
23 // READOBJ-NEXT: 0000000000000080 64 OBJECT LOCAL DEFAULT 3 special_sgpr.kd
24 // READOBJ-NEXT: 00000000000000c0 64 OBJECT LOCAL DEFAULT 3 disabled_user_sgpr.kd
26 // OBJDUMP: Contents of section .rodata
27 // Note, relocation for KERNEL_CODE_ENTRY_BYTE_OFFSET is not resolved here.
28 // minimal
29 // OBJDUMP-NEXT: 0000 00000000 00000000 00000000 00000000
30 // OBJDUMP-NEXT: 0010 00000000 00000000 00000000 00000000
31 // OBJDUMP-NEXT: 0020 00000000 00000000 00000000 00000000
32 // OBJDUMP-NEXT: 0030 00000c60 80000000 00000000 00000000
33 // complete
34 // OBJDUMP-NEXT: 0040 01000000 01000000 08000000 00000000
35 // OBJDUMP-NEXT: 0050 00000000 00000000 00000000 00000000
36 // OBJDUMP-NEXT: 0060 00000000 00000000 00000000 00000000
37 // OBJDUMP-NEXT: 0070 015021e4 1f0f007f 5e040000 00000000
38 // special_sgpr
39 // OBJDUMP-NEXT: 0080 00000000 00000000 00000000 00000000
40 // OBJDUMP-NEXT: 0090 00000000 00000000 00000000 00000000
41 // OBJDUMP-NEXT: 00a0 00000000 00000000 00000000 00000000
42 // OBJDUMP-NEXT: 00b0 00000060 80000000 00000000 00000000
43 // disabled_user_sgpr
44 // OBJDUMP-NEXT: 00c0 00000000 00000000 00000000 00000000
45 // OBJDUMP-NEXT: 00d0 00000000 00000000 00000000 00000000
46 // OBJDUMP-NEXT: 00e0 00000000 00000000 00000000 00000000
47 // OBJDUMP-NEXT: 00f0 00000c60 80000000 00000000 00000000
49 .text
50 // ASM: .text
52 .amdgcn_target "amdgcn-amd-amdhsa--gfx1200"
53 // ASM: .amdgcn_target "amdgcn-amd-amdhsa--gfx1200"
55 .p2align 8
56 .type minimal,@function
57 minimal:
58 s_endpgm
60 .p2align 8
61 .type complete,@function
62 complete:
63 s_endpgm
65 .p2align 8
66 .type special_sgpr,@function
67 special_sgpr:
68 s_endpgm
70 .p2align 8
71 .type disabled_user_sgpr,@function
72 disabled_user_sgpr:
73 s_endpgm
75 .rodata
76 // ASM: .rodata
78 // Test that only specifying required directives is allowed, and that defaulted
79 // values are omitted.
80 .p2align 6
81 .amdhsa_kernel minimal
82 .amdhsa_next_free_vgpr 0
83 .amdhsa_next_free_sgpr 0
84 .end_amdhsa_kernel
86 // ASM: .amdhsa_kernel minimal
87 // ASM: .amdhsa_next_free_vgpr 0
88 // ASM-NEXT: .amdhsa_next_free_sgpr 0
89 // ASM: .end_amdhsa_kernel
91 // Test that we can specify all available directives with non-default values.
92 .p2align 6
93 .amdhsa_kernel complete
94 .amdhsa_group_segment_fixed_size 1
95 .amdhsa_private_segment_fixed_size 1
96 .amdhsa_kernarg_size 8
97 .amdhsa_user_sgpr_count 15
98 .amdhsa_user_sgpr_dispatch_ptr 1
99 .amdhsa_user_sgpr_queue_ptr 1
100 .amdhsa_user_sgpr_kernarg_segment_ptr 1
101 .amdhsa_user_sgpr_dispatch_id 1
102 .amdhsa_user_sgpr_private_segment_size 1
103 .amdhsa_wavefront_size32 1
104 .amdhsa_enable_private_segment 1
105 .amdhsa_system_sgpr_workgroup_id_x 0
106 .amdhsa_system_sgpr_workgroup_id_y 1
107 .amdhsa_system_sgpr_workgroup_id_z 1
108 .amdhsa_system_sgpr_workgroup_info 1
109 .amdhsa_system_vgpr_workitem_id 1
110 .amdhsa_next_free_vgpr 9
111 .amdhsa_next_free_sgpr 27
112 .amdhsa_reserve_vcc 0
113 .amdhsa_float_round_mode_32 1
114 .amdhsa_float_round_mode_16_64 1
115 .amdhsa_float_denorm_mode_32 1
116 .amdhsa_float_denorm_mode_16_64 0
117 .amdhsa_fp16_overflow 1
118 .amdhsa_workgroup_processor_mode 1
119 .amdhsa_memory_ordered 1
120 .amdhsa_forward_progress 1
121 .amdhsa_round_robin_scheduling 1
122 .amdhsa_exception_fp_ieee_invalid_op 1
123 .amdhsa_exception_fp_denorm_src 1
124 .amdhsa_exception_fp_ieee_div_zero 1
125 .amdhsa_exception_fp_ieee_overflow 1
126 .amdhsa_exception_fp_ieee_underflow 1
127 .amdhsa_exception_fp_ieee_inexact 1
128 .amdhsa_exception_int_div_zero 1
129 .end_amdhsa_kernel
131 // ASM: .amdhsa_kernel complete
132 // ASM-NEXT: .amdhsa_group_segment_fixed_size 1
133 // ASM-NEXT: .amdhsa_private_segment_fixed_size 1
134 // ASM-NEXT: .amdhsa_kernarg_size 8
135 // ASM-NEXT: .amdhsa_user_sgpr_count 15
136 // ASM-NEXT: .amdhsa_user_sgpr_dispatch_ptr 1
137 // ASM-NEXT: .amdhsa_user_sgpr_queue_ptr 1
138 // ASM-NEXT: .amdhsa_user_sgpr_kernarg_segment_ptr 1
139 // ASM-NEXT: .amdhsa_user_sgpr_dispatch_id 1
140 // ASM-NEXT: .amdhsa_user_sgpr_private_segment_size 1
141 // ASM-NEXT: .amdhsa_wavefront_size32 1
142 // ASM-NEXT: .amdhsa_enable_private_segment 1
143 // ASM-NEXT: .amdhsa_system_sgpr_workgroup_id_x 0
144 // ASM-NEXT: .amdhsa_system_sgpr_workgroup_id_y 1
145 // ASM-NEXT: .amdhsa_system_sgpr_workgroup_id_z 1
146 // ASM-NEXT: .amdhsa_system_sgpr_workgroup_info 1
147 // ASM-NEXT: .amdhsa_system_vgpr_workitem_id 1
148 // ASM-NEXT: .amdhsa_next_free_vgpr 9
149 // ASM-NEXT: .amdhsa_next_free_sgpr 27
150 // ASM-NEXT: .amdhsa_reserve_vcc 0
151 // ASM-NEXT: .amdhsa_float_round_mode_32 1
152 // ASM-NEXT: .amdhsa_float_round_mode_16_64 1
153 // ASM-NEXT: .amdhsa_float_denorm_mode_32 1
154 // ASM-NEXT: .amdhsa_float_denorm_mode_16_64 0
155 // ASM-NEXT: .amdhsa_fp16_overflow 1
156 // ASM-NEXT: .amdhsa_workgroup_processor_mode 1
157 // ASM-NEXT: .amdhsa_memory_ordered 1
158 // ASM-NEXT: .amdhsa_forward_progress 1
159 // ASM-NEXT: .amdhsa_round_robin_scheduling 1
160 // ASM-NEXT: .amdhsa_exception_fp_ieee_invalid_op 1
161 // ASM-NEXT: .amdhsa_exception_fp_denorm_src 1
162 // ASM-NEXT: .amdhsa_exception_fp_ieee_div_zero 1
163 // ASM-NEXT: .amdhsa_exception_fp_ieee_overflow 1
164 // ASM-NEXT: .amdhsa_exception_fp_ieee_underflow 1
165 // ASM-NEXT: .amdhsa_exception_fp_ieee_inexact 1
166 // ASM-NEXT: .amdhsa_exception_int_div_zero 1
167 // ASM-NEXT: .end_amdhsa_kernel
169 // Test that we are including special SGPR usage in the granulated count.
170 .p2align 6
171 .amdhsa_kernel special_sgpr
172 .amdhsa_next_free_sgpr 27
174 .amdhsa_reserve_vcc 0
176 .amdhsa_float_denorm_mode_16_64 0
177 .amdhsa_next_free_vgpr 0
178 .end_amdhsa_kernel
180 // ASM: .amdhsa_kernel special_sgpr
181 // ASM: .amdhsa_next_free_vgpr 0
182 // ASM-NEXT: .amdhsa_next_free_sgpr 27
183 // ASM-NEXT: .amdhsa_reserve_vcc 0
184 // ASM: .amdhsa_float_denorm_mode_16_64 0
185 // ASM: .end_amdhsa_kernel
187 // Test that explicitly disabling user_sgpr's does not affect the user_sgpr
188 // count, i.e. this should produce the same descriptor as minimal.
189 .p2align 6
190 .amdhsa_kernel disabled_user_sgpr
191 .amdhsa_next_free_vgpr 0
192 .amdhsa_next_free_sgpr 0
193 .end_amdhsa_kernel
195 // ASM: .amdhsa_kernel disabled_user_sgpr
196 // ASM: .amdhsa_next_free_vgpr 0
197 // ASM-NEXT: .amdhsa_next_free_sgpr 0
198 // ASM: .end_amdhsa_kernel
200 .section .foo
202 .byte .amdgcn.gfx_generation_number
203 // ASM: .byte 12
205 .byte .amdgcn.gfx_generation_minor
206 // ASM: .byte 0
208 .byte .amdgcn.gfx_generation_stepping
209 // ASM: .byte 0
211 .byte .amdgcn.next_free_vgpr
212 // ASM: .byte 0
213 .byte .amdgcn.next_free_sgpr
214 // ASM: .byte 0
216 v_mov_b32_e32 v7, s10
218 .byte .amdgcn.next_free_vgpr
219 // ASM: .byte 8
220 .byte .amdgcn.next_free_sgpr
221 // ASM: .byte 11
223 .set .amdgcn.next_free_vgpr, 0
224 .set .amdgcn.next_free_sgpr, 0
226 .byte .amdgcn.next_free_vgpr
227 // ASM: .byte 0
228 .byte .amdgcn.next_free_sgpr
229 // ASM: .byte 0
231 v_mov_b32_e32 v16, s3
233 .byte .amdgcn.next_free_vgpr
234 // ASM: .byte 17
235 .byte .amdgcn.next_free_sgpr
236 // ASM: .byte 4
238 // Metadata
240 .amdgpu_metadata
241 amdhsa.version:
244 amdhsa.kernels:
245 - .name: amd_kernel_code_t_test_all
246 .symbol: amd_kernel_code_t_test_all@kd
247 .kernarg_segment_size: 8
248 .group_segment_fixed_size: 16
249 .private_segment_fixed_size: 32
250 .kernarg_segment_align: 64
251 .wavefront_size: 128
252 .sgpr_count: 14
253 .vgpr_count: 40
254 .max_flat_workgroup_size: 256
255 - .name: amd_kernel_code_t_minimal
256 .symbol: amd_kernel_code_t_minimal@kd
257 .kernarg_segment_size: 8
258 .group_segment_fixed_size: 16
259 .private_segment_fixed_size: 32
260 .kernarg_segment_align: 64
261 .wavefront_size: 128
262 .sgpr_count: 14
263 .vgpr_count: 40
264 .max_flat_workgroup_size: 256
265 .end_amdgpu_metadata
267 // ASM: .amdgpu_metadata
268 // ASM: amdhsa.kernels:
269 // ASM: - .group_segment_fixed_size: 16
270 // ASM: .kernarg_segment_align: 64
271 // ASM: .kernarg_segment_size: 8
272 // ASM: .max_flat_workgroup_size: 256
273 // ASM: .name: amd_kernel_code_t_test_all
274 // ASM: .private_segment_fixed_size: 32
275 // ASM: .sgpr_count: 14
276 // ASM: .symbol: 'amd_kernel_code_t_test_all@kd'
277 // ASM: .vgpr_count: 40
278 // ASM: .wavefront_size: 128
279 // ASM: - .group_segment_fixed_size: 16
280 // ASM: .kernarg_segment_align: 64
281 // ASM: .kernarg_segment_size: 8
282 // ASM: .max_flat_workgroup_size: 256
283 // ASM: .name: amd_kernel_code_t_minimal
284 // ASM: .private_segment_fixed_size: 32
285 // ASM: .sgpr_count: 14
286 // ASM: .symbol: 'amd_kernel_code_t_minimal@kd'
287 // ASM: .vgpr_count: 40
288 // ASM: .wavefront_size: 128
289 // ASM: amdhsa.version:
290 // ASM-NEXT: - 3
291 // ASM-NEXT: - 0
292 // ASM: .end_amdgpu_metadata