xref: /llvm-project/llvm/test/CodeGen/AMDGPU/llvm.amdgcn.lds.kernel.id.ll (revision 6548b6354d1d990e1c98736f5e7c3de876bedc8e)
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc -mtriple=amdgcn--amdhsa -verify-machineinstrs < %s | FileCheck -check-prefix=GCN %s
3
4declare i32 @llvm.amdgcn.lds.kernel.id()
5declare i32 @llvm.amdgcn.workgroup.id.x()
6
7define void @function_lds_id(ptr addrspace(1) %out) {
8; GCN-LABEL: function_lds_id:
9; GCN:       ; %bb.0:
10; GCN-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
11; GCN-NEXT:    s_add_i32 s4, s15, s12
12; GCN-NEXT:    v_mov_b32_e32 v2, s4
13; GCN-NEXT:    flat_store_dword v[0:1], v2
14; GCN-NEXT:    s_waitcnt vmcnt(0)
15; GCN-NEXT:    s_setpc_b64 s[30:31]
16  %tmp0 = call i32 @llvm.amdgcn.lds.kernel.id()
17  %help = call i32 @llvm.amdgcn.workgroup.id.x()
18  %both = add i32 %tmp0, %help
19  store i32 %both, ptr addrspace(1) %out
20  ret void
21}
22
23define amdgpu_kernel void @kernel_lds_id(ptr addrspace(1) %out) !llvm.amdgcn.lds.kernel.id !0 {
24; GCN-LABEL: kernel_lds_id:
25; GCN:       ; %bb.0:
26; GCN-NEXT:    s_load_dwordx2 s[0:1], s[8:9], 0x0
27; GCN-NEXT:    s_add_i32 s2, s12, 42
28; GCN-NEXT:    v_mov_b32_e32 v2, s2
29; GCN-NEXT:    s_waitcnt lgkmcnt(0)
30; GCN-NEXT:    v_mov_b32_e32 v0, s0
31; GCN-NEXT:    v_mov_b32_e32 v1, s1
32; GCN-NEXT:    flat_store_dword v[0:1], v2
33; GCN-NEXT:    s_endpgm
34  %tmp0 = call i32 @llvm.amdgcn.lds.kernel.id()
35  %help = call i32 @llvm.amdgcn.workgroup.id.x()
36  %both = add i32 %tmp0, %help
37  store i32 %both, ptr addrspace(1) %out
38  ret void
39}
40
41define amdgpu_kernel void @indirect_lds_id(ptr addrspace(1) %out) !llvm.amdgcn.lds.kernel.id !1 {
42; GCN-LABEL: indirect_lds_id:
43; GCN:       ; %bb.0:
44; GCN-NEXT:    s_mov_b32 s32, 0
45; GCN-NEXT:    s_mov_b32 flat_scratch_lo, s13
46; GCN-NEXT:    s_add_i32 s12, s12, s17
47; GCN-NEXT:    s_lshr_b32 flat_scratch_hi, s12, 8
48; GCN-NEXT:    s_add_u32 s0, s0, s17
49; GCN-NEXT:    s_addc_u32 s1, s1, 0
50; GCN-NEXT:    s_mov_b32 s13, s15
51; GCN-NEXT:    s_mov_b32 s12, s14
52; GCN-NEXT:    s_load_dwordx2 s[18:19], s[8:9], 0x0
53; GCN-NEXT:    s_add_u32 s8, s8, 8
54; GCN-NEXT:    v_lshlrev_b32_e32 v2, 20, v2
55; GCN-NEXT:    s_addc_u32 s9, s9, 0
56; GCN-NEXT:    s_getpc_b64 s[14:15]
57; GCN-NEXT:    s_add_u32 s14, s14, function_lds_id@gotpcrel32@lo+4
58; GCN-NEXT:    s_addc_u32 s15, s15, function_lds_id@gotpcrel32@hi+12
59; GCN-NEXT:    s_load_dwordx2 s[20:21], s[14:15], 0x0
60; GCN-NEXT:    v_lshlrev_b32_e32 v1, 10, v1
61; GCN-NEXT:    v_or_b32_e32 v0, v0, v1
62; GCN-NEXT:    v_or_b32_e32 v31, v0, v2
63; GCN-NEXT:    s_mov_b32 s15, 21
64; GCN-NEXT:    s_mov_b32 s14, s16
65; GCN-NEXT:    s_waitcnt lgkmcnt(0)
66; GCN-NEXT:    v_mov_b32_e32 v0, s18
67; GCN-NEXT:    v_mov_b32_e32 v1, s19
68; GCN-NEXT:    s_swappc_b64 s[30:31], s[20:21]
69; GCN-NEXT:    s_endpgm
70  call void @function_lds_id(ptr addrspace(1) %out)
71  ret void
72}
73
74define amdgpu_kernel void @doesnt_use_it(ptr addrspace(1) %out) !llvm.amdgcn.lds.kernel.id !0 {
75; GCN-LABEL: doesnt_use_it:
76; GCN:       ; %bb.0:
77; GCN-NEXT:    s_load_dwordx2 s[0:1], s[8:9], 0x0
78; GCN-NEXT:    v_mov_b32_e32 v2, 0x64
79; GCN-NEXT:    s_waitcnt lgkmcnt(0)
80; GCN-NEXT:    v_mov_b32_e32 v0, s0
81; GCN-NEXT:    v_mov_b32_e32 v1, s1
82; GCN-NEXT:    flat_store_dword v[0:1], v2
83; GCN-NEXT:    s_endpgm
84  store i32 100, ptr addrspace(1) %out
85  ret void
86}
87
88
89!0 = !{i32 42}
90!1 = !{i32 21}
91