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 -check-prefix=GFX9
   3 ; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx1200 < %s | FileCheck %s -check-prefix=GFX12
   4
   5 ; We have an indirect call with a known set of callees, which are
   6 ; known to not need any special inputs. The ABI still needs to use the
   7 ; register
   8
   9 ; FIXME: Passing real values for workitem ID, and 0s that can be undef
  10
  11 define amdgpu_kernel void @indirect_call_known_no_special_inputs() {
  12 ; GFX9-LABEL: indirect_call_known_no_special_inputs:
  13 ; GFX9:       ; %bb.0: ; %bb
  14 ; GFX9-NEXT:    s_add_u32 flat_scratch_lo, s12, s17
  15 ; GFX9-NEXT:    s_addc_u32 flat_scratch_hi, s13, 0
  16 ; GFX9-NEXT:    s_add_u32 s0, s0, s17
  17 ; GFX9-NEXT:    s_addc_u32 s1, s1, 0
  18 ; GFX9-NEXT:    s_mov_b32 s13, s15
  19 ; GFX9-NEXT:    s_mov_b32 s12, s14
  20 ; GFX9-NEXT:    s_mov_b64 s[14:15], 0
  21 ; GFX9-NEXT:    s_load_dword s17, s[14:15], 0x0
  22 ; GFX9-NEXT:    s_getpc_b64 s[14:15]
  23 ; GFX9-NEXT:    s_add_u32 s14, s14, wobble@gotpcrel32@lo+4
  24 ; GFX9-NEXT:    s_addc_u32 s15, s15, wobble@gotpcrel32@hi+12
  25 ; GFX9-NEXT:    s_getpc_b64 s[18:19]
  26 ; GFX9-NEXT:    s_add_u32 s18, s18, snork@gotpcrel32@lo+4
  27 ; GFX9-NEXT:    s_addc_u32 s19, s19, snork@gotpcrel32@hi+12
  28 ; GFX9-NEXT:    s_load_dwordx2 s[20:21], s[18:19], 0x0
  29 ; GFX9-NEXT:    s_load_dwordx2 s[22:23], s[14:15], 0x0
  30 ; GFX9-NEXT:    v_lshlrev_b32_e32 v2, 20, v2
  31 ; GFX9-NEXT:    s_waitcnt lgkmcnt(0)
  32 ; GFX9-NEXT:    s_and_b32 s14, 1, s17
  33 ; GFX9-NEXT:    s_cmp_eq_u32 s14, 1
  34 ; GFX9-NEXT:    v_lshlrev_b32_e32 v1, 10, v1
  35 ; GFX9-NEXT:    s_cselect_b32 s19, s23, s21
  36 ; GFX9-NEXT:    s_cselect_b32 s18, s22, s20
  37 ; GFX9-NEXT:    v_or3_b32 v31, v0, v1, v2
  38 ; GFX9-NEXT:    s_mov_b32 s14, s16
  39 ; GFX9-NEXT:    v_mov_b32_e32 v1, 0
  40 ; GFX9-NEXT:    v_mov_b32_e32 v4, 0
  41 ; GFX9-NEXT:    s_mov_b32 s32, 0
  42 ; GFX9-NEXT:    s_swappc_b64 s[30:31], s[18:19]
  43 ; GFX9-NEXT:    s_endpgm
  44 ;
  45 ; GFX12-LABEL: indirect_call_known_no_special_inputs:
  46 ; GFX12:       ; %bb.0: ; %bb
  47 ; GFX12-NEXT:    s_mov_b64 s[10:11], s[6:7]
  48 ; GFX12-NEXT:    s_getpc_b64 s[6:7]
  49 ; GFX12-NEXT:    s_wait_alu 0xfffe
  50 ; GFX12-NEXT:    s_sext_i32_i16 s7, s7
  51 ; GFX12-NEXT:    s_add_co_u32 s6, s6, snork@gotpcrel32@lo+12
  52 ; GFX12-NEXT:    s_wait_alu 0xfffe
  53 ; GFX12-NEXT:    s_add_co_ci_u32 s7, s7, snork@gotpcrel32@hi+24
  54 ; GFX12-NEXT:    s_mov_b64 s[8:9], s[4:5]
  55 ; GFX12-NEXT:    s_mov_b64 s[4:5], 0
  56 ; GFX12-NEXT:    s_getpc_b64 s[12:13]
  57 ; GFX12-NEXT:    s_wait_alu 0xfffe
  58 ; GFX12-NEXT:    s_sext_i32_i16 s13, s13
  59 ; GFX12-NEXT:    s_add_co_u32 s12, s12, wobble@gotpcrel32@lo+12
  60 ; GFX12-NEXT:    s_wait_alu 0xfffe
  61 ; GFX12-NEXT:    s_add_co_ci_u32 s13, s13, wobble@gotpcrel32@hi+24
  62 ; GFX12-NEXT:    s_load_u8 s14, s[4:5], 0x0
  63 ; GFX12-NEXT:    s_load_b64 s[4:5], s[6:7], 0x0
  64 ; GFX12-NEXT:    s_load_b64 s[6:7], s[12:13], 0x0
  65 ; GFX12-NEXT:    v_dual_mov_b32 v1, 0 :: v_dual_mov_b32 v4, 0
  66 ; GFX12-NEXT:    v_mov_b32_e32 v31, v0
  67 ; GFX12-NEXT:    s_mov_b32 s32, 0
  68 ; GFX12-NEXT:    s_wait_kmcnt 0x0
  69 ; GFX12-NEXT:    s_and_b32 s12, 1, s14
  70 ; GFX12-NEXT:    s_wait_alu 0xfffe
  71 ; GFX12-NEXT:    s_cmp_eq_u32 s12, 1
  72 ; GFX12-NEXT:    s_cselect_b32 s13, s7, s5
  73 ; GFX12-NEXT:    s_cselect_b32 s12, s6, s4
  74 ; GFX12-NEXT:    s_mov_b64 s[4:5], s[0:1]
  75 ; GFX12-NEXT:    s_mov_b64 s[6:7], s[2:3]
  76 ; GFX12-NEXT:    s_wait_alu 0xfffe
  77 ; GFX12-NEXT:    s_swappc_b64 s[30:31], s[12:13]
  78 ; GFX12-NEXT:    s_endpgm
  79
  80 bb:
  81   %cond = load i1, ptr addrspace(4) null
  82   %tmp = select i1 %cond, ptr @wobble, ptr @snork
  83   call void %tmp(ptr undef, i32 undef, ptr undef)
  84   ret void
  85 }
  86
  87 define void @wobble() {
  88 ; GFX9-LABEL: wobble:
  89 ; GFX9:       ; %bb.0: ; %bb
  90 ; GFX9-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
  91 ; GFX9-NEXT:    s_setpc_b64 s[30:31]
  92 ;
  93 ; GFX12-LABEL: wobble:
  94 ; GFX12:       ; %bb.0: ; %bb
  95 ; GFX12-NEXT:    s_wait_loadcnt_dscnt 0x0
  96 ; GFX12-NEXT:    s_wait_expcnt 0x0
  97 ; GFX12-NEXT:    s_wait_samplecnt 0x0
  98 ; GFX12-NEXT:    s_wait_bvhcnt 0x0
  99 ; GFX12-NEXT:    s_wait_kmcnt 0x0
 100 ; GFX12-NEXT:    s_setpc_b64 s[30:31]
 101 bb:
 102   ret void
 103 }
 104
 105 define void @snork() {
 106 ; GFX9-LABEL: snork:
 107 ; GFX9:       ; %bb.0: ; %bb
 108 ; GFX9-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
 109 ; GFX9-NEXT:    s_setpc_b64 s[30:31]
 110 ;
 111 ; GFX12-LABEL: snork:
 112 ; GFX12:       ; %bb.0: ; %bb
 113 ; GFX12-NEXT:    s_wait_loadcnt_dscnt 0x0
 114 ; GFX12-NEXT:    s_wait_expcnt 0x0
 115 ; GFX12-NEXT:    s_wait_samplecnt 0x0
 116 ; GFX12-NEXT:    s_wait_bvhcnt 0x0
 117 ; GFX12-NEXT:    s_wait_kmcnt 0x0
 118 ; GFX12-NEXT:    s_setpc_b64 s[30:31]
 119 bb:
 120   ret void
 121 }