xref: /aosp_15_r20/external/llvm/test/CodeGen/AMDGPU/amdgcn.work-item-intrinsics.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; RUN: llc -march=amdgcn -mcpu=SI -verify-machineinstrs < %s | FileCheck -check-prefix=SI -check-prefix=GCN -check-prefix=SI-NOHSA -check-prefix=GCN-NOHSA -check-prefix=FUNC %s
2*9880d681SAndroid Build Coastguard Worker; RUN: llc -march=amdgcn -mcpu=tonga -verify-machineinstrs < %s | FileCheck -check-prefix=VI  -check-prefix=VI-NOHSA -check-prefix=GCN -check-prefix=GCN-NOHSA -check-prefix=FUNC %s
3*9880d681SAndroid Build Coastguard Worker
4*9880d681SAndroid Build Coastguard Worker
5*9880d681SAndroid Build Coastguard Worker; FUNC-LABEL: {{^}}workdim:
6*9880d681SAndroid Build Coastguard Worker
7*9880d681SAndroid Build Coastguard Worker; SI-NOHSA: s_load_dword [[VAL:s[0-9]+]], s[0:1], 0xb
8*9880d681SAndroid Build Coastguard Worker; VI-NOHSA: s_load_dword [[VAL:s[0-9]+]], s[0:1], 0x2c
9*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: v_mov_b32_e32 [[VVAL:v[0-9]+]], [[VAL]]
10*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: buffer_store_dword [[VVAL]]
11*9880d681SAndroid Build Coastguard Worker
12*9880d681SAndroid Build Coastguard Workerdefine void @workdim (i32 addrspace(1)* %out) {
13*9880d681SAndroid Build Coastguard Workerentry:
14*9880d681SAndroid Build Coastguard Worker  %0 = call i32 @llvm.amdgcn.read.workdim() #0
15*9880d681SAndroid Build Coastguard Worker  store i32 %0, i32 addrspace(1)* %out
16*9880d681SAndroid Build Coastguard Worker  ret void
17*9880d681SAndroid Build Coastguard Worker}
18*9880d681SAndroid Build Coastguard Worker
19*9880d681SAndroid Build Coastguard Worker; The workgroup.id values are stored in sgprs offset by the number of user
20*9880d681SAndroid Build Coastguard Worker; sgprs.
21*9880d681SAndroid Build Coastguard Worker
22*9880d681SAndroid Build Coastguard Worker; FUNC-LABEL: {{^}}workgroup_id_x:
23*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: v_mov_b32_e32 [[VVAL:v[0-9]+]], s2{{$}}
24*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: buffer_store_dword [[VVAL]]
25*9880d681SAndroid Build Coastguard Worker
26*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: COMPUTE_PGM_RSRC2:USER_SGPR: 2
27*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TGID_X_EN: 1
28*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TGID_Y_EN: 0
29*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TGID_Z_EN: 0
30*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TIDIG_COMP_CNT: 0
31*9880d681SAndroid Build Coastguard Workerdefine void @workgroup_id_x(i32 addrspace(1)* %out) {
32*9880d681SAndroid Build Coastguard Workerentry:
33*9880d681SAndroid Build Coastguard Worker  %0 = call i32 @llvm.amdgcn.workgroup.id.x() #0
34*9880d681SAndroid Build Coastguard Worker  store i32 %0, i32 addrspace(1)* %out
35*9880d681SAndroid Build Coastguard Worker  ret void
36*9880d681SAndroid Build Coastguard Worker}
37*9880d681SAndroid Build Coastguard Worker
38*9880d681SAndroid Build Coastguard Worker; FUNC-LABEL: {{^}}workgroup_id_y:
39*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: v_mov_b32_e32 [[VVAL:v[0-9]+]], s3
40*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: buffer_store_dword [[VVAL]]
41*9880d681SAndroid Build Coastguard Worker
42*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: COMPUTE_PGM_RSRC2:USER_SGPR: 2
43*9880d681SAndroid Build Coastguard Workerdefine void @workgroup_id_y(i32 addrspace(1)* %out) {
44*9880d681SAndroid Build Coastguard Workerentry:
45*9880d681SAndroid Build Coastguard Worker  %0 = call i32 @llvm.amdgcn.workgroup.id.y() #0
46*9880d681SAndroid Build Coastguard Worker  store i32 %0, i32 addrspace(1)* %out
47*9880d681SAndroid Build Coastguard Worker  ret void
48*9880d681SAndroid Build Coastguard Worker}
49*9880d681SAndroid Build Coastguard Worker
50*9880d681SAndroid Build Coastguard Worker; FUNC-LABEL: {{^}}workgroup_id_z:
51*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: v_mov_b32_e32 [[VVAL:v[0-9]+]], s3{{$}}
52*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: buffer_store_dword [[VVAL]]
53*9880d681SAndroid Build Coastguard Worker
54*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: COMPUTE_PGM_RSRC2:USER_SGPR: 2
55*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TGID_X_EN: 1
56*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TGID_Y_EN: 0
57*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TGID_Z_EN: 1
58*9880d681SAndroid Build Coastguard Worker; GCN: COMPUTE_PGM_RSRC2:TIDIG_COMP_CNT: 0
59*9880d681SAndroid Build Coastguard Workerdefine void @workgroup_id_z(i32 addrspace(1)* %out) {
60*9880d681SAndroid Build Coastguard Workerentry:
61*9880d681SAndroid Build Coastguard Worker  %0 = call i32 @llvm.amdgcn.workgroup.id.z() #0
62*9880d681SAndroid Build Coastguard Worker  store i32 %0, i32 addrspace(1)* %out
63*9880d681SAndroid Build Coastguard Worker  ret void
64*9880d681SAndroid Build Coastguard Worker}
65*9880d681SAndroid Build Coastguard Worker
66*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: .section .AMDGPU.config
67*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: .long 47180
68*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA-NEXT: .long 132{{$}}
69*9880d681SAndroid Build Coastguard Worker
70*9880d681SAndroid Build Coastguard Worker; FUNC-LABEL: {{^}}workitem_id_x:
71*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: buffer_store_dword v0
72*9880d681SAndroid Build Coastguard Workerdefine void @workitem_id_x(i32 addrspace(1)* %out) {
73*9880d681SAndroid Build Coastguard Workerentry:
74*9880d681SAndroid Build Coastguard Worker  %0 = call i32 @llvm.amdgcn.workitem.id.x() #0
75*9880d681SAndroid Build Coastguard Worker  store i32 %0, i32 addrspace(1)* %out
76*9880d681SAndroid Build Coastguard Worker  ret void
77*9880d681SAndroid Build Coastguard Worker}
78*9880d681SAndroid Build Coastguard Worker
79*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: .section .AMDGPU.config
80*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: .long 47180
81*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA-NEXT: .long 2180{{$}}
82*9880d681SAndroid Build Coastguard Worker
83*9880d681SAndroid Build Coastguard Worker; FUNC-LABEL: {{^}}workitem_id_y:
84*9880d681SAndroid Build Coastguard Worker
85*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: buffer_store_dword v1
86*9880d681SAndroid Build Coastguard Workerdefine void @workitem_id_y(i32 addrspace(1)* %out) {
87*9880d681SAndroid Build Coastguard Workerentry:
88*9880d681SAndroid Build Coastguard Worker  %0 = call i32 @llvm.amdgcn.workitem.id.y() #0
89*9880d681SAndroid Build Coastguard Worker  store i32 %0, i32 addrspace(1)* %out
90*9880d681SAndroid Build Coastguard Worker  ret void
91*9880d681SAndroid Build Coastguard Worker}
92*9880d681SAndroid Build Coastguard Worker
93*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: .section .AMDGPU.config
94*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: .long 47180
95*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA-NEXT: .long 4228{{$}}
96*9880d681SAndroid Build Coastguard Worker
97*9880d681SAndroid Build Coastguard Worker; FUNC-LABEL: {{^}}workitem_id_z:
98*9880d681SAndroid Build Coastguard Worker; GCN-NOHSA: buffer_store_dword v2
99*9880d681SAndroid Build Coastguard Workerdefine void @workitem_id_z(i32 addrspace(1)* %out) {
100*9880d681SAndroid Build Coastguard Workerentry:
101*9880d681SAndroid Build Coastguard Worker  %0 = call i32 @llvm.amdgcn.workitem.id.z() #0
102*9880d681SAndroid Build Coastguard Worker  store i32 %0, i32 addrspace(1)* %out
103*9880d681SAndroid Build Coastguard Worker  ret void
104*9880d681SAndroid Build Coastguard Worker}
105*9880d681SAndroid Build Coastguard Worker
106*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.workgroup.id.x() #0
107*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.workgroup.id.y() #0
108*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.workgroup.id.z() #0
109*9880d681SAndroid Build Coastguard Worker
110*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.workitem.id.x() #0
111*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.workitem.id.y() #0
112*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.workitem.id.z() #0
113*9880d681SAndroid Build Coastguard Worker
114*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.read.workdim() #0
115