llvm/test/CodeGen/AMDGPU/promote-alloca-globals.ll

   1 ; RUN: opt -data-layout=A5 -S -mtriple=amdgcn-unknown-unknown -passes=amdgpu-promote-alloca < %s | FileCheck -check-prefix=IR %s
   2 ; RUN: llc -march=amdgcn -mcpu=tonga < %s | FileCheck -check-prefix=ASM %s
   3
   4
   5 @global_array0 = internal unnamed_addr addrspace(3) global [750 x [10 x i32]] undef, align 4
   6 @global_array1 = internal unnamed_addr addrspace(3) global [750 x [10 x i32]] undef, align 4
   7
   8 ; IR-LABEL: define amdgpu_kernel void @promote_alloca_size_256(ptr addrspace(1) nocapture %out, ptr addrspace(1) nocapture %in) {
   9 ; IR-NOT: alloca [10 x i32]
  10 ; ASM-LABEL: {{^}}promote_alloca_size_256:
  11 ; ASM: .amdgpu_lds llvm.amdgcn.kernel.promote_alloca_size_256.lds, 60000, 16
  12 ; ASM-NOT: .amdgpu_lds
  13
  14 define amdgpu_kernel void @promote_alloca_size_256(ptr addrspace(1) nocapture %out, ptr addrspace(1) nocapture %in) {
  15 entry:
  16   %stack = alloca [10 x i32], align 4, addrspace(5)
  17   %tmp = load i32, ptr addrspace(1) %in, align 4
  18   %arrayidx1 = getelementptr inbounds [10 x i32], ptr addrspace(5) %stack, i32 0, i32 %tmp
  19   store i32 4, ptr addrspace(5) %arrayidx1, align 4
  20   %arrayidx2 = getelementptr inbounds i32, ptr addrspace(1) %in, i32 1
  21   %tmp1 = load i32, ptr addrspace(1) %arrayidx2, align 4
  22   %arrayidx3 = getelementptr inbounds [10 x i32], ptr addrspace(5) %stack, i32 0, i32 %tmp1
  23   store i32 5, ptr addrspace(5) %arrayidx3, align 4
  24   %tmp2 = load i32, ptr addrspace(5) %stack, align 4
  25   store i32 %tmp2, ptr addrspace(1) %out, align 4
  26   %arrayidx12 = getelementptr inbounds [10 x i32], ptr addrspace(5) %stack, i32 0, i32 1
  27   %tmp3 = load i32, ptr addrspace(5) %arrayidx12
  28   %arrayidx13 = getelementptr inbounds i32, ptr addrspace(1) %out, i32 1
  29   store i32 %tmp3, ptr addrspace(1) %arrayidx13
  30   store i32 %tmp3, ptr addrspace(3) @global_array0
  31   store i32 %tmp3, ptr addrspace(3) @global_array1
  32   ret void
  33 }