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) 8define amdgpu_kernel void @array_alloca(ptr addrspace(1) nocapture %out, ptr addrspace(1) nocapture %in) #0 { 9entry: 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) 29define amdgpu_kernel void @array_alloca_dynamic(ptr addrspace(1) nocapture %out, ptr addrspace(1) nocapture %in, i32 %size) #0 { 30entry: 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 48attributes #0 = { nounwind } 49