llvm/test/CodeGen/AMDGPU/llvm.amdgcn.lds.kernel.id.ll

   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
   4 declare i32 @llvm.amdgcn.lds.kernel.id()
   5 declare i32 @llvm.amdgcn.workgroup.id.x()
   6
   7 define 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
  23 define 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[4:5], 0x0
  27 ; GCN-NEXT:    s_add_i32 s2, s6, 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
  41 define 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, s7
  46 ; GCN-NEXT:    s_add_i32 s6, s6, s9
  47 ; GCN-NEXT:    s_lshr_b32 flat_scratch_hi, s6, 8
  48 ; GCN-NEXT:    s_add_u32 s0, s0, s9
  49 ; GCN-NEXT:    s_addc_u32 s1, s1, 0
  50 ; GCN-NEXT:    s_load_dwordx2 s[4:5], s[4:5], 0x0
  51 ; GCN-NEXT:    s_getpc_b64 s[6:7]
  52 ; GCN-NEXT:    s_add_u32 s6, s6, function_lds_id@gotpcrel32@lo+4
  53 ; GCN-NEXT:    s_addc_u32 s7, s7, function_lds_id@gotpcrel32@hi+12
  54 ; GCN-NEXT:    s_load_dwordx2 s[6:7], s[6:7], 0x0
  55 ; GCN-NEXT:    s_mov_b32 s15, 21
  56 ; GCN-NEXT:    s_mov_b32 s12, s8
  57 ; GCN-NEXT:    s_waitcnt lgkmcnt(0)
  58 ; GCN-NEXT:    v_mov_b32_e32 v0, s4
  59 ; GCN-NEXT:    v_mov_b32_e32 v1, s5
  60 ; GCN-NEXT:    s_swappc_b64 s[30:31], s[6:7]
  61 ; GCN-NEXT:    s_endpgm
  62   call void @function_lds_id(ptr addrspace(1) %out)
  63   ret void
  64 }
  65
  66 define amdgpu_kernel void @doesnt_use_it(ptr addrspace(1) %out) !llvm.amdgcn.lds.kernel.id !0 {
  67 ; GCN-LABEL: doesnt_use_it:
  68 ; GCN:       ; %bb.0:
  69 ; GCN-NEXT:    s_load_dwordx2 s[0:1], s[4:5], 0x0
  70 ; GCN-NEXT:    v_mov_b32_e32 v2, 0x64
  71 ; GCN-NEXT:    s_waitcnt lgkmcnt(0)
  72 ; GCN-NEXT:    v_mov_b32_e32 v0, s0
  73 ; GCN-NEXT:    v_mov_b32_e32 v1, s1
  74 ; GCN-NEXT:    flat_store_dword v[0:1], v2
  75 ; GCN-NEXT:    s_endpgm
  76   store i32 100, ptr addrspace(1) %out
  77   ret void
  78 }
  79
  80
  81 !0 = !{i32 42}
  82 !1 = !{i32 21}