llvm/test/CodeGen/AMDGPU/indirect-call-known-callees.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx900 < %s | FileCheck %s
   3
   4 ; We have an indirect call with a known set of callees, which are
   5 ; known to not need any special inputs. The ABI still needs to use the
   6 ; register
   7
   8 ; FIXME: Passing real values for workitem ID, and 0s that can be undef
   9
  10 define amdgpu_kernel void @indirect_call_known_no_special_inputs() {
  11 ; CHECK-LABEL: indirect_call_known_no_special_inputs:
  12 ; CHECK:       ; %bb.0: ; %bb
  13 ; CHECK-NEXT:    s_add_u32 flat_scratch_lo, s4, s7
  14 ; CHECK-NEXT:    s_addc_u32 flat_scratch_hi, s5, 0
  15 ; CHECK-NEXT:    s_mov_b64 s[4:5], 0
  16 ; CHECK-NEXT:    s_load_dword s4, s[4:5], 0x0
  17 ; CHECK-NEXT:    s_add_u32 s0, s0, s7
  18 ; CHECK-NEXT:    s_addc_u32 s1, s1, 0
  19 ; CHECK-NEXT:    s_mov_b32 s33, s6
  20 ; CHECK-NEXT:    v_mov_b32_e32 v31, v0
  21 ; CHECK-NEXT:    s_waitcnt lgkmcnt(0)
  22 ; CHECK-NEXT:    s_bitcmp1_b32 s4, 0
  23 ; CHECK-NEXT:    s_cselect_b64 vcc, -1, 0
  24 ; CHECK-NEXT:    s_getpc_b64 s[4:5]
  25 ; CHECK-NEXT:    s_add_u32 s4, s4, wobble@gotpcrel32@lo+4
  26 ; CHECK-NEXT:    s_addc_u32 s5, s5, wobble@gotpcrel32@hi+12
  27 ; CHECK-NEXT:    s_getpc_b64 s[6:7]
  28 ; CHECK-NEXT:    s_add_u32 s6, s6, snork@gotpcrel32@lo+4
  29 ; CHECK-NEXT:    s_addc_u32 s7, s7, snork@gotpcrel32@hi+12
  30 ; CHECK-NEXT:    s_load_dwordx2 s[8:9], s[6:7], 0x0
  31 ; CHECK-NEXT:    s_load_dwordx2 s[10:11], s[4:5], 0x0
  32 ; CHECK-NEXT:    s_mov_b32 s32, 0
  33 ; CHECK-NEXT:    s_mov_b64 s[4:5], exec
  34 ; CHECK-NEXT:    s_waitcnt lgkmcnt(0)
  35 ; CHECK-NEXT:    v_mov_b32_e32 v0, s9
  36 ; CHECK-NEXT:    v_mov_b32_e32 v1, s11
  37 ; CHECK-NEXT:    v_mov_b32_e32 v2, s8
  38 ; CHECK-NEXT:    v_mov_b32_e32 v4, s10
  39 ; CHECK-NEXT:    v_cndmask_b32_e32 v3, v0, v1, vcc
  40 ; CHECK-NEXT:    v_cndmask_b32_e32 v2, v2, v4, vcc
  41 ; CHECK-NEXT:    v_mov_b32_e32 v1, 0
  42 ; CHECK-NEXT:  .LBB0_1: ; =>This Inner Loop Header: Depth=1
  43 ; CHECK-NEXT:    v_readfirstlane_b32 s4, v2
  44 ; CHECK-NEXT:    v_readfirstlane_b32 s5, v3
  45 ; CHECK-NEXT:    v_cmp_eq_u64_e32 vcc, s[4:5], v[2:3]
  46 ; CHECK-NEXT:    s_and_saveexec_b64 s[34:35], vcc
  47 ; CHECK-NEXT:    s_mov_b64 s[8:9], 0
  48 ; CHECK-NEXT:    s_mov_b32 s12, s33
  49 ; CHECK-NEXT:    v_mov_b32_e32 v4, v1
  50 ; CHECK-NEXT:    s_swappc_b64 s[30:31], s[4:5]
  51 ; CHECK-NEXT:    ; implicit-def: $vgpr2_vgpr3
  52 ; CHECK-NEXT:    ; implicit-def: $vgpr31
  53 ; CHECK-NEXT:    ; implicit-def: $vgpr1
  54 ; CHECK-NEXT:    s_xor_b64 exec, exec, s[34:35]
  55 ; CHECK-NEXT:    s_cbranch_execnz .LBB0_1
  56 ; CHECK-NEXT:  ; %bb.2:
  57 ; CHECK-NEXT:    s_endpgm
  58
  59 ; CHECK: .amdhsa_kernarg_size 0
  60 ; CHECK-NEXT: .amdhsa_user_sgpr_count 6
  61 ; CHECK-NEXT: .amdhsa_user_sgpr_private_segment_buffer 1
  62 ; CHECK-NEXT: .amdhsa_user_sgpr_dispatch_ptr 0
  63 ; CHECK-NEXT: .amdhsa_user_sgpr_queue_ptr 0
  64 ; CHECK-NEXT: .amdhsa_user_sgpr_kernarg_segment_ptr 0
  65 ; CHECK-NEXT: .amdhsa_user_sgpr_dispatch_id 0
  66 ; CHECK-NEXT: .amdhsa_user_sgpr_flat_scratch_init 1
  67 ; CHECK-NEXT: .amdhsa_user_sgpr_private_segment_size 0
  68 ; CHECK-NEXT: .amdhsa_system_sgpr_private_segment_wavefront_offset 1
  69 ; CHECK-NEXT: .amdhsa_system_sgpr_workgroup_id_x 1
  70 ; CHECK-NEXT: .amdhsa_system_sgpr_workgroup_id_y 0
  71 ; CHECK-NEXT: .amdhsa_system_sgpr_workgroup_id_z 0
  72 ; CHECK-NEXT: .amdhsa_system_sgpr_workgroup_info 0
  73 ; CHECK-NEXT: .amdhsa_system_vgpr_workitem_id 0
  74 bb:
  75   %cond = load i1, i1 addrspace(4)* null
  76   %tmp = select i1 %cond, void (i8*, i32, i8*)* bitcast (void ()* @wobble to void (i8*, i32, i8*)*), void (i8*, i32, i8*)* bitcast (void ()* @snork to void (i8*, i32, i8*)*)
  77   call void %tmp(i8* undef, i32 undef, i8* undef)
  78   ret void
  79 }
  80
  81 define void @wobble() {
  82 ; CHECK-LABEL: wobble:
  83 ; CHECK:       ; %bb.0: ; %bb
  84 ; CHECK-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
  85 ; CHECK-NEXT:    s_setpc_b64 s[30:31]
  86 bb:
  87   ret void
  88 }
  89
  90 define void @snork() {
  91 ; CHECK-LABEL: snork:
  92 ; CHECK:       ; %bb.0: ; %bb
  93 ; CHECK-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
  94 ; CHECK-NEXT:    s_setpc_b64 s[30:31]
  95 bb:
  96   ret void
  97 }