llvm/test/CodeGen/AMDGPU/llvm.amdgcn.dispatch.ptr.ll

   1 ; RUN: llc -mtriple=amdgcn--amdhsa --amdhsa-code-object-version=2 -mcpu=kaveri -verify-machineinstrs < %s | FileCheck -check-prefix=GCN %s
   2 ; RUN: not llc -mtriple=amdgcn-unknown-unknown -mcpu=kaveri -verify-machineinstrs < %s 2>&1 | FileCheck -check-prefix=ERROR %s
   3
   4 ; ERROR: in function test{{.*}}: unsupported hsa intrinsic without hsa target
   5
   6 ; GCN-LABEL: {{^}}test:
   7 ; GCN: enable_sgpr_dispatch_ptr = 1
   8 ; GCN: s_load_dword s{{[0-9]+}}, s[4:5], 0x0
   9 define amdgpu_kernel void @test(i32 addrspace(1)* %out) {
  10   %dispatch_ptr = call noalias i8 addrspace(4)* @llvm.amdgcn.dispatch.ptr() #0
  11   %header_ptr = bitcast i8 addrspace(4)* %dispatch_ptr to i32 addrspace(4)*
  12   %value = load i32, i32 addrspace(4)* %header_ptr
  13   store i32 %value, i32 addrspace(1)* %out
  14   ret void
  15 }
  16
  17 ; GCN-LABEL: {{^}}test2
  18 ; GCN: enable_sgpr_dispatch_ptr = 1
  19 ; GCN: s_load_dword s[[REG:[0-9]+]], s[4:5], 0x1
  20 ; GCN: s_lshr_b32 s{{[0-9]+}}, s[[REG]], 16
  21 ; GCN-NOT: load_ushort
  22 ; GCN: s_endpgm
  23 define amdgpu_kernel void @test2(i32 addrspace(1)* %out) {
  24   %dispatch_ptr = call noalias i8 addrspace(4)* @llvm.amdgcn.dispatch.ptr() #0
  25   %d1 = getelementptr inbounds i8, i8 addrspace(4)* %dispatch_ptr, i64 6
  26   %h1 = bitcast i8 addrspace(4)* %d1 to i16 addrspace(4)*
  27   %v1 = load i16, i16 addrspace(4)* %h1
  28   %e1 = zext i16 %v1 to i32
  29   store i32 %e1, i32 addrspace(1)* %out
  30   ret void
  31 }
  32
  33 declare noalias i8 addrspace(4)* @llvm.amdgcn.dispatch.ptr() #0
  34
  35 attributes #0 = { readnone }