xref: /aosp_15_r20/external/llvm/test/CodeGen/AMDGPU/amdgcn.private-memory.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; RUN: llc -mattr=+promote-alloca -verify-machineinstrs -march=amdgcn < %s | FileCheck -check-prefix=GCN -check-prefix=GCN-PROMOTE %s
2*9880d681SAndroid Build Coastguard Worker; RUN: llc -mattr=+promote-alloca,-flat-for-global -verify-machineinstrs -mtriple=amdgcn--amdhsa -mcpu=kaveri < %s | FileCheck -check-prefix=GCN -check-prefix=GCN-PROMOTE -check-prefix=HSA %s
3*9880d681SAndroid Build Coastguard Worker; RUN: llc -mattr=-promote-alloca -verify-machineinstrs -march=amdgcn < %s | FileCheck -check-prefix=GCN -check-prefix=GCN-ALLOCA %s
4*9880d681SAndroid Build Coastguard Worker; RUN: llc -mattr=-promote-alloca,-flat-for-global -verify-machineinstrs -mtriple=amdgcn-amdhsa -mcpu=kaveri < %s | FileCheck  -check-prefix=GCN -check-prefix=GCN-ALLOCA -check-prefix=HSA %s
5*9880d681SAndroid Build Coastguard Worker; RUN: llc -mattr=+promote-alloca -verify-machineinstrs -march=amdgcn -mcpu=tonga < %s | FileCheck -check-prefix=GCN -check-prefix=GCN-PROMOTE %s
6*9880d681SAndroid Build Coastguard Worker; RUN: llc -mattr=-promote-alloca -verify-machineinstrs -march=amdgcn -mcpu=tonga < %s | FileCheck -check-prefix=GCN  -check-prefix=GCN-ALLOCA %s
7*9880d681SAndroid Build Coastguard Worker
8*9880d681SAndroid Build Coastguard Worker
9*9880d681SAndroid Build Coastguard Workerdeclare i32 @llvm.amdgcn.workitem.id.x() nounwind readnone
10*9880d681SAndroid Build Coastguard Worker
11*9880d681SAndroid Build Coastguard Worker
12*9880d681SAndroid Build Coastguard Worker; Make sure we don't overwrite workitem information with private memory
13*9880d681SAndroid Build Coastguard Worker
14*9880d681SAndroid Build Coastguard Worker; GCN-LABEL: {{^}}work_item_info:
15*9880d681SAndroid Build Coastguard Worker; GCN-NOT: v0
16*9880d681SAndroid Build Coastguard Worker; GCN: v_add_i32_e32 [[RESULT:v[0-9]+]], vcc, v0, v{{[0-9]+}}
17*9880d681SAndroid Build Coastguard Worker; GCN: buffer_store_dword [[RESULT]]
18*9880d681SAndroid Build Coastguard Workerdefine void @work_item_info(i32 addrspace(1)* %out, i32 %in) {
19*9880d681SAndroid Build Coastguard Workerentry:
20*9880d681SAndroid Build Coastguard Worker  %0 = alloca [2 x i32]
21*9880d681SAndroid Build Coastguard Worker  %1 = getelementptr [2 x i32], [2 x i32]* %0, i32 0, i32 0
22*9880d681SAndroid Build Coastguard Worker  %2 = getelementptr [2 x i32], [2 x i32]* %0, i32 0, i32 1
23*9880d681SAndroid Build Coastguard Worker  store i32 0, i32* %1
24*9880d681SAndroid Build Coastguard Worker  store i32 1, i32* %2
25*9880d681SAndroid Build Coastguard Worker  %3 = getelementptr [2 x i32], [2 x i32]* %0, i32 0, i32 %in
26*9880d681SAndroid Build Coastguard Worker  %4 = load i32, i32* %3
27*9880d681SAndroid Build Coastguard Worker  %5 = call i32 @llvm.amdgcn.workitem.id.x()
28*9880d681SAndroid Build Coastguard Worker  %6 = add i32 %4, %5
29*9880d681SAndroid Build Coastguard Worker  store i32 %6, i32 addrspace(1)* %out
30*9880d681SAndroid Build Coastguard Worker  ret void
31*9880d681SAndroid Build Coastguard Worker}
32