xref: /llvm-project/llvm/test/Transforms/InferAddressSpaces/AMDGPU/assumed-addrspace.ll (revision 5651af896c3df30da9edd101b1fb17c00de6636d)
1; RUN: opt -S -mtriple=amdgcn-amd-amdhsa -passes=infer-address-spaces -o - %s | FileCheck %s
2
3@c0 = addrspace(4) global ptr undef
4
5; CHECK-LABEL: @generic_ptr_from_constant
6; CHECK: addrspacecast ptr %p to ptr addrspace(1)
7; CHECK-NEXT: load float, ptr addrspace(1)
8define float @generic_ptr_from_constant() {
9  %p = load ptr, ptr addrspace(4) @c0
10  %v = load float, ptr %p
11  ret float %v
12}
13
14%struct.S = type { ptr, ptr }
15
16; CHECK-LABEL: @generic_ptr_from_aggregate_argument
17; CHECK: addrspacecast ptr %p0 to ptr addrspace(1)
18; CHECK: addrspacecast ptr %p1 to ptr addrspace(1)
19; CHECK: load i32, ptr addrspace(1)
20; CHECK: store float %v1, ptr addrspace(1)
21; CHECK: ret
22define amdgpu_kernel void @generic_ptr_from_aggregate_argument(ptr addrspace(4) byref(%struct.S) align 8 %0) {
23  %p0 = load ptr, ptr addrspace(4) %0
24  %f1 = getelementptr inbounds %struct.S, ptr addrspace(4) %0, i64 0, i32 1
25  %p1 = load ptr, ptr addrspace(4) %f1
26  %v0 = load i32, ptr %p0
27  %v1 = sitofp i32 %v0 to float
28  store float %v1, ptr %p1
29  ret void
30}
31