llvm/test/CodeGen/AMDGPU/call-graph-register-usage.ll

   1 ; RUN: sed 's/CODE_OBJECT_VERSION/400/g' %s | llc -mtriple=amdgcn-amd-amdhsa -enable-ipra=0 -verify-machineinstrs | FileCheck -check-prefixes=GCN,CI %s
   2 ; RUN: sed 's/CODE_OBJECT_VERSION/500/g' %s | llc -mtriple=amdgcn-amd-amdhsa -enable-ipra=0 -verify-machineinstrs | FileCheck -check-prefixes=GCN-V5 %s
   3 ; RUN: sed 's/CODE_OBJECT_VERSION/400/g' %s | llc -mtriple=amdgcn-amd-amdhsa -mcpu=fiji -enable-ipra=0 -verify-machineinstrs | FileCheck -check-prefixes=GCN,VI,VI-NOBUG %s
   4 ; RUN: sed 's/CODE_OBJECT_VERSION/400/g' %s | llc -mtriple=amdgcn-amd-amdhsa -mcpu=iceland -enable-ipra=0 -verify-machineinstrs | FileCheck -check-prefixes=GCN,VI,VI-BUG %s
   5
   6 ; Make sure to run a GPU with the SGPR allocation bug.
   7
   8 ; GCN-LABEL: {{^}}use_vcc:
   9 ; GCN: ; NumSgprs: 34
  10 ; GCN: ; NumVgprs: 0
  11 define void @use_vcc() #1 {
  12   call void asm sideeffect "", "~{vcc}" () #0
  13   ret void
  14 }
  15
  16 ; GCN-LABEL: {{^}}indirect_use_vcc:
  17 ; GCN: s_mov_b32 s4, s33
  18 ; GCN: v_writelane_b32 v40, s4, 2
  19 ; GCN: v_writelane_b32 v40, s30, 0
  20 ; GCN: v_writelane_b32 v40, s31, 1
  21 ; GCN: s_swappc_b64
  22 ; GCN: v_readlane_b32 s31, v40, 1
  23 ; GCN: v_readlane_b32 s30, v40, 0
  24 ; GCN: v_readlane_b32 s4, v40, 2
  25 ; GCN: s_mov_b32 s33, s4
  26 ; GCN: s_setpc_b64 s[30:31]
  27 ; GCN: ; NumSgprs: 36
  28 ; GCN: ; NumVgprs: 41
  29 define void @indirect_use_vcc() #1 {
  30   call void @use_vcc()
  31   ret void
  32 }
  33
  34 ; GCN-LABEL: {{^}}indirect_2level_use_vcc_kernel:
  35 ; CI: ; NumSgprs: 38
  36 ; VI-NOBUG: ; NumSgprs: 40
  37 ; VI-BUG: ; NumSgprs: 96
  38 ; GCN: ; NumVgprs: 41
  39 define amdgpu_kernel void @indirect_2level_use_vcc_kernel(ptr addrspace(1) %out) #0 {
  40   call void @indirect_use_vcc()
  41   ret void
  42 }
  43
  44 ; GCN-LABEL: {{^}}use_flat_scratch:
  45 ; CI: ; NumSgprs: 36
  46 ; VI: ; NumSgprs: 38
  47 ; GCN: ; NumVgprs: 0
  48 define void @use_flat_scratch() #1 {
  49   call void asm sideeffect "", "~{flat_scratch}" () #0
  50   ret void
  51 }
  52
  53 ; GCN-LABEL: {{^}}indirect_use_flat_scratch:
  54 ; CI: ; NumSgprs: 38
  55 ; VI: ; NumSgprs: 40
  56 ; GCN: ; NumVgprs: 41
  57 define void @indirect_use_flat_scratch() #1 {
  58   call void @use_flat_scratch()
  59   ret void
  60 }
  61
  62 ; GCN-LABEL: {{^}}indirect_2level_use_flat_scratch_kernel:
  63 ; CI: ; NumSgprs: 38
  64 ; VI-NOBUG: ; NumSgprs: 40
  65 ; VI-BUG: ; NumSgprs: 96
  66 ; GCN: ; NumVgprs: 41
  67 define amdgpu_kernel void @indirect_2level_use_flat_scratch_kernel(ptr addrspace(1) %out) #0 {
  68   call void @indirect_use_flat_scratch()
  69   ret void
  70 }
  71
  72 ; GCN-LABEL: {{^}}use_10_vgpr:
  73 ; GCN: ; NumVgprs: 10
  74 define void @use_10_vgpr() #1 {
  75   call void asm sideeffect "", "~{v0},~{v1},~{v2},~{v3},~{v4}"() #0
  76   call void asm sideeffect "", "~{v5},~{v6},~{v7},~{v8},~{v9}"() #0
  77   ret void
  78 }
  79
  80 ; GCN-LABEL: {{^}}indirect_use_10_vgpr:
  81 ; GCN: ; NumVgprs: 41
  82 define void @indirect_use_10_vgpr() #0 {
  83   call void @use_10_vgpr()
  84   ret void
  85 }
  86
  87 ; GCN-LABEL: {{^}}indirect_2_level_use_10_vgpr:
  88 ; GCN: ; NumVgprs: 41
  89 define amdgpu_kernel void @indirect_2_level_use_10_vgpr() #0 {
  90   call void @indirect_use_10_vgpr()
  91   ret void
  92 }
  93
  94 ; GCN-LABEL: {{^}}use_50_vgpr:
  95 ; GCN: ; NumVgprs: 50
  96 define void @use_50_vgpr() #1 {
  97   call void asm sideeffect "", "~{v49}"() #0
  98   ret void
  99 }
 100
 101 ; GCN-LABEL: {{^}}indirect_use_50_vgpr:
 102 ; GCN: ; NumVgprs: 50
 103 define void @indirect_use_50_vgpr() #0 {
 104   call void @use_50_vgpr()
 105   ret void
 106 }
 107
 108 ; GCN-LABEL: {{^}}use_80_sgpr:
 109 ; GCN: ; NumSgprs: 80
 110 define void @use_80_sgpr() #1 {
 111   call void asm sideeffect "", "~{s79}"() #0
 112   ret void
 113 }
 114
 115 ; GCN-LABEL: {{^}}indirect_use_80_sgpr:
 116 ; GCN: ; NumSgprs: 82
 117 define void @indirect_use_80_sgpr() #1 {
 118   call void @use_80_sgpr()
 119   ret void
 120 }
 121
 122 ; GCN-LABEL: {{^}}indirect_2_level_use_80_sgpr:
 123 ; CI: ; NumSgprs: 84
 124 ; VI-NOBUG: ; NumSgprs: 86
 125 ; VI-BUG: ; NumSgprs: 96
 126 define amdgpu_kernel void @indirect_2_level_use_80_sgpr() #0 {
 127   call void @indirect_use_80_sgpr()
 128   ret void
 129 }
 130
 131
 132 ; GCN-LABEL: {{^}}use_stack0:
 133 ; GCN: ScratchSize: 2052
 134 define void @use_stack0() #1 {
 135   %alloca = alloca [512 x i32], align 4, addrspace(5)
 136   call void asm sideeffect "; use $0", "v"(ptr addrspace(5) %alloca) #0
 137   ret void
 138 }
 139
 140 ; GCN-LABEL: {{^}}use_stack1:
 141 ; GCN: ScratchSize: 404
 142 define void @use_stack1() #1 {
 143   %alloca = alloca [100 x i32], align 4, addrspace(5)
 144   call void asm sideeffect "; use $0", "v"(ptr addrspace(5) %alloca) #0
 145   ret void
 146 }
 147
 148 ; GCN-LABEL: {{^}}indirect_use_stack:
 149 ; GCN: ScratchSize: 2132
 150 define void @indirect_use_stack() #1 {
 151   %alloca = alloca [16 x i32], align 4, addrspace(5)
 152   call void asm sideeffect "; use $0", "v"(ptr addrspace(5) %alloca) #0
 153   call void @use_stack0()
 154   ret void
 155 }
 156
 157 ; GCN-LABEL: {{^}}indirect_2_level_use_stack:
 158 ; GCN: ScratchSize: 2132
 159 define amdgpu_kernel void @indirect_2_level_use_stack() #0 {
 160   call void @indirect_use_stack()
 161   ret void
 162 }
 163
 164
 165 ; Should be maximum of callee usage
 166 ; GCN-LABEL: {{^}}multi_call_use_use_stack:
 167 ; GCN: ScratchSize: 2052
 168 define amdgpu_kernel void @multi_call_use_use_stack() #0 {
 169   call void @use_stack0()
 170   call void @use_stack1()
 171   ret void
 172 }
 173
 174
 175 declare void @external() #0
 176
 177 ; GCN-LABEL: {{^}}usage_external:
 178 ; NumSgprs: 48
 179 ; NumVgprs: 24
 180 ; GCN: ScratchSize: 16384
 181 ;
 182 ; GCN-V5-LABEL: {{^}}usage_external:
 183 ; GCN-V5: ScratchSize: 0
 184 define amdgpu_kernel void @usage_external() #0 {
 185   call void @external()
 186   ret void
 187 }
 188
 189 declare void @external_recurse() #2
 190
 191 ; GCN-LABEL: {{^}}usage_external_recurse:
 192 ; NumSgprs: 48
 193 ; NumVgprs: 24
 194 ; GCN: ScratchSize: 16384
 195 ;
 196 ; GCN-V5-LABEL: {{^}}usage_external_recurse:
 197 ; GCN-V5: ScratchSize: 0
 198 define amdgpu_kernel void @usage_external_recurse() #0 {
 199   call void @external_recurse()
 200   ret void
 201 }
 202
 203 ; GCN-LABEL: {{^}}direct_recursion_use_stack:
 204 ; GCN: ScratchSize: 18448{{$}}
 205 ;
 206 ; GCN-V5-LABEL: {{^}}direct_recursion_use_stack:
 207 ; GCN-V5: ScratchSize: 2064{{$}}
 208 define void @direct_recursion_use_stack(i32 %val) #2 {
 209   %alloca = alloca [512 x i32], align 4, addrspace(5)
 210   call void asm sideeffect "; use $0", "v"(ptr addrspace(5) %alloca) #0
 211   %cmp = icmp eq i32 %val, 0
 212   br i1 %cmp, label %ret, label %call
 213
 214 call:
 215   %val.sub1 = sub i32 %val, 1
 216   call void @direct_recursion_use_stack(i32 %val.sub1)
 217   br label %ret
 218
 219 ret:
 220   ret void
 221 }
 222
 223 ; GCN-LABEL: {{^}}usage_direct_recursion:
 224 ; GCN: .amdhsa_private_segment_fixed_size 18448
 225 ;
 226 ; GCN-V5-LABEL: {{^}}usage_direct_recursion:
 227 ; GCN-V5: .amdhsa_private_segment_fixed_size 2064{{$}}
 228 define amdgpu_kernel void @usage_direct_recursion(i32 %n) #0 {
 229   call void @direct_recursion_use_stack(i32 %n)
 230   ret void
 231 }
 232
 233 ; Make sure there's no assert when a sgpr96 is used.
 234 ; GCN-LABEL: {{^}}count_use_sgpr96_external_call
 235 ; GCN: ; sgpr96 s[{{[0-9]+}}:{{[0-9]+}}]
 236 ; CI: NumSgprs: 84
 237 ; VI-NOBUG: NumSgprs: 86
 238 ; VI-BUG: NumSgprs: 96
 239 ; GCN: NumVgprs: 50
 240 define amdgpu_kernel void @count_use_sgpr96_external_call()  {
 241 entry:
 242   tail call void asm sideeffect "; sgpr96 $0", "s"(<3 x i32> <i32 10, i32 11, i32 12>) #1
 243   call void @external()
 244   ret void
 245 }
 246
 247 ; Make sure there's no assert when a sgpr160 is used.
 248 ; GCN-LABEL: {{^}}count_use_sgpr160_external_call
 249 ; GCN: ; sgpr160 s[{{[0-9]+}}:{{[0-9]+}}]
 250 ; CI: NumSgprs: 84
 251 ; VI-NOBUG: NumSgprs: 86
 252 ; VI-BUG: NumSgprs: 96
 253 ; GCN: NumVgprs: 50
 254 define amdgpu_kernel void @count_use_sgpr160_external_call()  {
 255 entry:
 256   tail call void asm sideeffect "; sgpr160 $0", "s"(<5 x i32> <i32 10, i32 11, i32 12, i32 13, i32 14>) #1
 257   call void @external()
 258   ret void
 259 }
 260
 261 ; Make sure there's no assert when a vgpr160 is used.
 262 ; GCN-LABEL: {{^}}count_use_vgpr160_external_call
 263 ; GCN: ; vgpr160 v[{{[0-9]+}}:{{[0-9]+}}]
 264 ; CI: NumSgprs: 84
 265 ; VI-NOBUG: NumSgprs: 86
 266 ; VI-BUG: NumSgprs: 96
 267 ; GCN: NumVgprs: 50
 268 define amdgpu_kernel void @count_use_vgpr160_external_call()  {
 269 entry:
 270   tail call void asm sideeffect "; vgpr160 $0", "v"(<5 x i32> <i32 10, i32 11, i32 12, i32 13, i32 14>) #1
 271   call void @external()
 272   ret void
 273 }
 274
 275 attributes #0 = { nounwind noinline norecurse }
 276 attributes #1 = { nounwind noinline norecurse }
 277 attributes #2 = { nounwind noinline }
 278
 279 !llvm.module.flags = !{!0}
 280 !0 = !{i32 1, !"amdgpu_code_object_version", i32 CODE_OBJECT_VERSION}