llvm/test/CodeGen/AMDGPU/promote-alloca-array-allocation.ll

   1 ; RUN: opt -S -mtriple=amdgcn-amd-amdhsa -passes=amdgpu-promote-alloca < %s | FileCheck %s
   2
   3 ; Make sure this allocates the correct size if the alloca has a non-0
   4 ; number of elements.
   5
   6 ; CHECK-LABEL: @array_alloca(
   7 ; CHECK: %stack = alloca i32, i32 5, align 4, addrspace(5)
   8 define amdgpu_kernel void @array_alloca(ptr addrspace(1) nocapture %out, ptr addrspace(1) nocapture %in) #0 {
   9 entry:
  10   %stack = alloca i32, i32 5, align 4, addrspace(5)
  11   %ld0 = load i32, ptr addrspace(1) %in, align 4
  12   %arrayidx1 = getelementptr inbounds i32, ptr addrspace(5) %stack, i32 %ld0
  13   store i32 4, ptr addrspace(5) %arrayidx1, align 4
  14   %arrayidx2 = getelementptr inbounds i32, ptr addrspace(1) %in, i32 1
  15   %ld1 = load i32, ptr addrspace(1) %arrayidx2, align 4
  16   %arrayidx3 = getelementptr inbounds i32, ptr addrspace(5) %stack, i32 %ld1
  17   store i32 5, ptr addrspace(5) %arrayidx3, align 4
  18   %ld2 = load i32, ptr addrspace(5) %stack, align 4
  19   store i32 %ld2, ptr addrspace(1) %out, align 4
  20   %arrayidx12 = getelementptr inbounds i32, ptr addrspace(5) %stack, i32 1
  21   %ld3 = load i32, ptr addrspace(5) %arrayidx12
  22   %arrayidx13 = getelementptr inbounds i32, ptr addrspace(1) %out, i32 1
  23   store i32 %ld3, ptr addrspace(1) %arrayidx13
  24   ret void
  25 }
  26
  27 ; CHECK-LABEL: @array_alloca_dynamic(
  28 ; CHECK: %stack = alloca i32, i32 %size, align 4, addrspace(5)
  29 define amdgpu_kernel void @array_alloca_dynamic(ptr addrspace(1) nocapture %out, ptr addrspace(1) nocapture %in, i32 %size) #0 {
  30 entry:
  31   %stack = alloca i32, i32 %size, align 4, addrspace(5)
  32   %ld0 = load i32, ptr addrspace(1) %in, align 4
  33   %arrayidx1 = getelementptr inbounds i32, ptr addrspace(5) %stack, i32 %ld0
  34   store i32 4, ptr addrspace(5) %arrayidx1, align 4
  35   %arrayidx2 = getelementptr inbounds i32, ptr addrspace(1) %in, i32 1
  36   %ld1 = load i32, ptr addrspace(1) %arrayidx2, align 4
  37   %arrayidx3 = getelementptr inbounds i32, ptr addrspace(5) %stack, i32 %ld1
  38   store i32 5, ptr addrspace(5) %arrayidx3, align 4
  39   %ld2 = load i32, ptr addrspace(5) %stack, align 4
  40   store i32 %ld2, ptr addrspace(1) %out, align 4
  41   %arrayidx12 = getelementptr inbounds i32, ptr addrspace(5) %stack, i32 1
  42   %ld3 = load i32, ptr addrspace(5) %arrayidx12
  43   %arrayidx13 = getelementptr inbounds i32, ptr addrspace(1) %out, i32 1
  44   store i32 %ld3, ptr addrspace(1) %arrayidx13
  45   ret void
  46 }
  47
  48 attributes #0 = { nounwind }