1; RUN: llc -verify-machineinstrs -march=amdgcn -mattr=-promote-alloca < %s | FileCheck -check-prefix=SI-ALLOCA -check-prefix=SI %s
2; RUN: llc -verify-machineinstrs -march=amdgcn -mattr=+promote-alloca < %s | FileCheck -check-prefix=SI-PROMOTE -check-prefix=SI %s
3
4declare i32 @llvm.amdgcn.mbcnt.lo(i32, i32) #1
5declare i32 @llvm.amdgcn.mbcnt.hi(i32, i32) #1
6declare void @llvm.amdgcn.s.barrier() #2
7
8; The required pointer calculations for the alloca'd actually requires
9; an add and won't be folded into the addressing, which fails with a
10; 64-bit pointer add. This should work since private pointers should
11; be 32-bits.
12
13; SI-LABEL: {{^}}test_private_array_ptr_calc:
14
15; FIXME: We end up with zero argument for ADD, because
16; SIRegisterInfo::eliminateFrameIndex() blindly replaces the frame index
17; with the appropriate offset.  We should fold this into the store.
18; SI-ALLOCA: v_add_i32_e32 [[PTRREG:v[0-9]+]], vcc, 0, v{{[0-9]+}}
19; SI-ALLOCA: buffer_store_dword {{v[0-9]+}}, [[PTRREG]], s[{{[0-9]+:[0-9]+}}]
20;
21; FIXME: The AMDGPUPromoteAlloca pass should be able to convert this
22; alloca to a vector.  It currently fails because it does not know how
23; to interpret:
24; getelementptr inbounds [4 x i32], [4 x i32]* %alloca, i32 1, i32 %b
25
26; SI-PROMOTE: v_add_i32_e32 [[PTRREG:v[0-9]+]], vcc, 16
27; SI-PROMOTE: ds_write_b32 [[PTRREG]]
28define void @test_private_array_ptr_calc(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %inA, i32 addrspace(1)* noalias %inB) #0 {
29  %alloca = alloca [4 x i32], i32 4, align 16
30  %mbcnt.lo = call i32 @llvm.amdgcn.mbcnt.lo(i32 -1, i32 0);
31  %tid = call i32 @llvm.amdgcn.mbcnt.hi(i32 -1, i32 %mbcnt.lo)
32  %a_ptr = getelementptr inbounds i32, i32 addrspace(1)* %inA, i32 %tid
33  %b_ptr = getelementptr inbounds i32, i32 addrspace(1)* %inB, i32 %tid
34  %a = load i32, i32 addrspace(1)* %a_ptr
35  %b = load i32, i32 addrspace(1)* %b_ptr
36  %result = add i32 %a, %b
37  %alloca_ptr = getelementptr inbounds [4 x i32], [4 x i32]* %alloca, i32 1, i32 %b
38  store i32 %result, i32* %alloca_ptr, align 4
39  ; Dummy call
40  call void @llvm.amdgcn.s.barrier()
41  %reload = load i32, i32* %alloca_ptr, align 4
42  %out_ptr = getelementptr inbounds i32, i32 addrspace(1)* %out, i32 %tid
43  store i32 %reload, i32 addrspace(1)* %out_ptr, align 4
44  ret void
45}
46
47attributes #0 = { nounwind }
48attributes #1 = { nounwind readnone }
49attributes #2 = { nounwind convergent }
50