1; RUN: llc -march=amdgcn -mtriple=amdgcn-unknown-amdhsa --amdhsa-code-object-version=2 -mcpu=kaveri -verify-machineinstrs < %s | FileCheck --check-prefixes=ALL,CO-V2  %s
2; RUN: llc -march=amdgcn -mtriple=amdgcn-unknown-amdhsa --amdhsa-code-object-version=2 -mcpu=carrizo -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck --check-prefixes=ALL,CO-V2  %s
3; RUN: llc -march=amdgcn -mcpu=tahiti -verify-machineinstrs < %s | FileCheck --check-prefixes=ALL,MESA %s
4; RUN: llc -march=amdgcn -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck --check-prefixes=ALL,MESA %s
5; RUN: llc -mtriple=amdgcn-unknown-mesa3d -mcpu=tahiti -verify-machineinstrs < %s | FileCheck -check-prefixes=ALL,CO-V2 %s
6; RUN: llc -mtriple=amdgcn-unknown-mesa3d -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefixes=ALL,CO-V2 %s
7; RUN: llc -march=amdgcn -mtriple=amdgcn-unknown-amdhsa -mcpu=gfx90a -verify-machineinstrs < %s | FileCheck -check-prefixes=ALL,PACKED-TID %s
8
9declare i32 @llvm.amdgcn.workitem.id.x() #0
10declare i32 @llvm.amdgcn.workitem.id.y() #0
11declare i32 @llvm.amdgcn.workitem.id.z() #0
12
13; MESA: .section .AMDGPU.config
14; MESA: .long 47180
15; MESA-NEXT: .long 132{{$}}
16
17; ALL-LABEL: {{^}}test_workitem_id_x:
18; CO-V2: enable_vgpr_workitem_id = 0
19
20; ALL-NOT: v0
21; ALL: {{buffer|flat|global}}_store_dword {{.*}}v0
22
23; CO-V3: .amdhsa_system_vgpr_workitem_id 0
24; PACKED-TID: .amdhsa_system_vgpr_workitem_id 0
25define amdgpu_kernel void @test_workitem_id_x(i32 addrspace(1)* %out) #1 {
26  %id = call i32 @llvm.amdgcn.workitem.id.x()
27  store i32 %id, i32 addrspace(1)* %out
28  ret void
29}
30
31; MESA: .section .AMDGPU.config
32; MESA: .long 47180
33; MESA-NEXT: .long 2180{{$}}
34
35; ALL-LABEL: {{^}}test_workitem_id_y:
36; CO-V2: enable_vgpr_workitem_id = 1
37
38; PACKED-TID: v_bfe_u32 [[ID:v[0-9]+]], v0, 10, 10
39; PACKED-TID: {{buffer|flat|global}}_store_dword {{.*}}[[ID]]
40
41; UNPACKED-TID-NOT: v1
42; UNPACKED-TID: {{buffer|flat}}_store_dword {{.*}}v1
43
44; CO-V3: .amdhsa_system_vgpr_workitem_id 1
45; PACKED-TID: .amdhsa_system_vgpr_workitem_id 1
46define amdgpu_kernel void @test_workitem_id_y(i32 addrspace(1)* %out) #1 {
47  %id = call i32 @llvm.amdgcn.workitem.id.y()
48  store i32 %id, i32 addrspace(1)* %out
49  ret void
50}
51
52; MESA: .section .AMDGPU.config
53; MESA: .long 47180
54; MESA-NEXT: .long 4228{{$}}
55
56; ALL-LABEL: {{^}}test_workitem_id_z:
57; CO-V2: enable_vgpr_workitem_id = 2
58
59; PACKED-TID: v_bfe_u32 [[ID:v[0-9]+]], v0, 20, 10
60; PACKED-TID: {{buffer|flat|global}}_store_dword {{.*}}[[ID]]
61
62; UNPACKED-TID-NOT: v2
63; UNPACKED-TID: {{buffer|flat}}_store_dword {{.*}}v2
64
65; CO-V3: .amdhsa_system_vgpr_workitem_id 2
66; PACKED-TID: .amdhsa_system_vgpr_workitem_id 2
67define amdgpu_kernel void @test_workitem_id_z(i32 addrspace(1)* %out) #1 {
68  %id = call i32 @llvm.amdgcn.workitem.id.z()
69  store i32 %id, i32 addrspace(1)* %out
70  ret void
71}
72
73attributes #0 = { nounwind readnone }
74attributes #1 = { nounwind }
75