[ARM] More MVE compare vector splat combines for ANDs
[llvm-complete.git] / test / Transforms / InferAddressSpaces / AMDGPU / basic.ll
blobe0a0043d76b7b19fe85fa9d4bdccd5a5acb74b95
1 ; RUN: opt -S -mtriple=amdgcn-amd-amdhsa -infer-address-spaces %s | FileCheck %s
3 ; Trivial optimization of generic addressing
5 ; CHECK-LABEL: @load_global_from_flat(
6 ; CHECK-NEXT: %tmp0 = addrspacecast float* %generic_scalar to float addrspace(1)*
7 ; CHECK-NEXT: %tmp1 = load float, float addrspace(1)* %tmp0
8 ; CHECK-NEXT: ret float %tmp1
9 define float @load_global_from_flat(float* %generic_scalar) #0 {
10   %tmp0 = addrspacecast float* %generic_scalar to float addrspace(1)*
11   %tmp1 = load float, float addrspace(1)* %tmp0
12   ret float %tmp1
15 ; CHECK-LABEL: @load_constant_from_flat(
16 ; CHECK-NEXT: %tmp0 = addrspacecast float* %generic_scalar to float addrspace(4)*
17 ; CHECK-NEXT: %tmp1 = load float, float addrspace(4)* %tmp0
18 ; CHECK-NEXT: ret float %tmp1
19 define float @load_constant_from_flat(float* %generic_scalar) #0 {
20   %tmp0 = addrspacecast float* %generic_scalar to float addrspace(4)*
21   %tmp1 = load float, float addrspace(4)* %tmp0
22   ret float %tmp1
25 ; CHECK-LABEL: @load_group_from_flat(
26 ; CHECK-NEXT: %tmp0 = addrspacecast float* %generic_scalar to float addrspace(3)*
27 ; CHECK-NEXT: %tmp1 = load float, float addrspace(3)* %tmp0
28 ; CHECK-NEXT: ret float %tmp1
29 define float @load_group_from_flat(float* %generic_scalar) #0 {
30   %tmp0 = addrspacecast float* %generic_scalar to float addrspace(3)*
31   %tmp1 = load float, float addrspace(3)* %tmp0
32   ret float %tmp1
35 ; CHECK-LABEL: @load_private_from_flat(
36 ; CHECK-NEXT: %tmp0 = addrspacecast float* %generic_scalar to float addrspace(5)*
37 ; CHECK-NEXT: %tmp1 = load float, float addrspace(5)* %tmp0
38 ; CHECK-NEXT: ret float %tmp1
39 define float @load_private_from_flat(float* %generic_scalar) #0 {
40   %tmp0 = addrspacecast float* %generic_scalar to float addrspace(5)*
41   %tmp1 = load float, float addrspace(5)* %tmp0
42   ret float %tmp1
45 ; CHECK-LABEL: @store_global_from_flat(
46 ; CHECK-NEXT: %tmp0 = addrspacecast float* %generic_scalar to float addrspace(1)*
47 ; CHECK-NEXT: store float 0.000000e+00, float addrspace(1)* %tmp0
48 define amdgpu_kernel void @store_global_from_flat(float* %generic_scalar) #0 {
49   %tmp0 = addrspacecast float* %generic_scalar to float addrspace(1)*
50   store float 0.0, float addrspace(1)* %tmp0
51   ret void
54 ; CHECK-LABEL: @store_group_from_flat(
55 ; CHECK-NEXT: %tmp0 = addrspacecast float* %generic_scalar to float addrspace(3)*
56 ; CHECK-NEXT: store float 0.000000e+00, float addrspace(3)* %tmp0
57 define amdgpu_kernel void @store_group_from_flat(float* %generic_scalar) #0 {
58   %tmp0 = addrspacecast float* %generic_scalar to float addrspace(3)*
59   store float 0.0, float addrspace(3)* %tmp0
60   ret void
63 ; CHECK-LABEL: @store_private_from_flat(
64 ; CHECK-NEXT: %tmp0 = addrspacecast float* %generic_scalar to float addrspace(5)*
65 ; CHECK-NEXT: store float 0.000000e+00, float addrspace(5)* %tmp0
66 define amdgpu_kernel void @store_private_from_flat(float* %generic_scalar) #0 {
67   %tmp0 = addrspacecast float* %generic_scalar to float addrspace(5)*
68   store float 0.0, float addrspace(5)* %tmp0
69   ret void
72 ; optimized to global load/store.
73 ; CHECK-LABEL: @load_store_global(
74 ; CHECK-NEXT: %val = load i32, i32 addrspace(1)* %input, align 4
75 ; CHECK-NEXT: store i32 %val, i32 addrspace(1)* %output, align 4
76 ; CHECK-NEXT: ret void
77 define amdgpu_kernel void @load_store_global(i32 addrspace(1)* nocapture %input, i32 addrspace(1)* nocapture %output) #0 {
78   %tmp0 = addrspacecast i32 addrspace(1)* %input to i32*
79   %tmp1 = addrspacecast i32 addrspace(1)* %output to i32*
80   %val = load i32, i32* %tmp0, align 4
81   store i32 %val, i32* %tmp1, align 4
82   ret void
85 ; Optimized to group load/store.
86 ; CHECK-LABEL: @load_store_group(
87 ; CHECK-NEXT: %val = load i32, i32 addrspace(3)* %input, align 4
88 ; CHECK-NEXT: store i32 %val, i32 addrspace(3)* %output, align 4
89 ; CHECK-NEXT: ret void
90 define amdgpu_kernel void @load_store_group(i32 addrspace(3)* nocapture %input, i32 addrspace(3)* nocapture %output) #0 {
91   %tmp0 = addrspacecast i32 addrspace(3)* %input to i32*
92   %tmp1 = addrspacecast i32 addrspace(3)* %output to i32*
93   %val = load i32, i32* %tmp0, align 4
94   store i32 %val, i32* %tmp1, align 4
95   ret void
98 ; Optimized to private load/store.
99 ; CHECK-LABEL: @load_store_private(
100 ; CHECK-NEXT: %val = load i32, i32 addrspace(5)* %input, align 4
101 ; CHECK-NEXT: store i32 %val, i32 addrspace(5)* %output, align 4
102 ; CHECK-NEXT: ret void
103 define amdgpu_kernel void @load_store_private(i32 addrspace(5)* nocapture %input, i32 addrspace(5)* nocapture %output) #0 {
104   %tmp0 = addrspacecast i32 addrspace(5)* %input to i32*
105   %tmp1 = addrspacecast i32 addrspace(5)* %output to i32*
106   %val = load i32, i32* %tmp0, align 4
107   store i32 %val, i32* %tmp1, align 4
108   ret void
111 ; No optimization. flat load/store.
112 ; CHECK-LABEL: @load_store_flat(
113 ; CHECK-NEXT: %val = load i32, i32* %input, align 4
114 ; CHECK-NEXT: store i32 %val, i32* %output, align 4
115 ; CHECK-NEXT: ret void
116 define amdgpu_kernel void @load_store_flat(i32* nocapture %input, i32* nocapture %output) #0 {
117   %val = load i32, i32* %input, align 4
118   store i32 %val, i32* %output, align 4
119   ret void
122 ; CHECK-LABEL: @store_addrspacecast_ptr_value(
123 ; CHECK: %cast = addrspacecast i32 addrspace(1)* %input to i32*
124 ; CHECK-NEXT: store i32* %cast, i32* addrspace(1)* %output, align 4
125 define amdgpu_kernel void @store_addrspacecast_ptr_value(i32 addrspace(1)* nocapture %input, i32* addrspace(1)* nocapture %output) #0 {
126   %cast = addrspacecast i32 addrspace(1)* %input to i32*
127   store i32* %cast, i32* addrspace(1)* %output, align 4
128   ret void
131 ; CHECK-LABEL: @atomicrmw_add_global_to_flat(
132 ; CHECK-NEXT: %ret = atomicrmw add i32 addrspace(1)* %global.ptr, i32 %y seq_cst
133 define i32 @atomicrmw_add_global_to_flat(i32 addrspace(1)* %global.ptr, i32 %y) #0 {
134   %cast = addrspacecast i32 addrspace(1)* %global.ptr to i32*
135   %ret = atomicrmw add i32* %cast, i32 %y seq_cst
136   ret i32 %ret
139 ; CHECK-LABEL: @atomicrmw_add_group_to_flat(
140 ; CHECK-NEXT: %ret = atomicrmw add i32 addrspace(3)* %group.ptr, i32 %y seq_cst
141 define i32 @atomicrmw_add_group_to_flat(i32 addrspace(3)* %group.ptr, i32 %y) #0 {
142   %cast = addrspacecast i32 addrspace(3)* %group.ptr to i32*
143   %ret = atomicrmw add i32* %cast, i32 %y seq_cst
144   ret i32 %ret
147 ; CHECK-LABEL: @cmpxchg_global_to_flat(
148 ; CHECK: %ret = cmpxchg i32 addrspace(1)* %global.ptr, i32 %cmp, i32 %val seq_cst monotonic
149 define { i32, i1 } @cmpxchg_global_to_flat(i32 addrspace(1)* %global.ptr, i32 %cmp, i32 %val) #0 {
150   %cast = addrspacecast i32 addrspace(1)* %global.ptr to i32*
151   %ret = cmpxchg i32* %cast, i32 %cmp, i32 %val seq_cst monotonic
152   ret { i32, i1 } %ret
155 ; CHECK-LABEL: @cmpxchg_group_to_flat(
156 ; CHECK: %ret = cmpxchg i32 addrspace(3)* %group.ptr, i32 %cmp, i32 %val seq_cst monotonic
157 define { i32, i1 } @cmpxchg_group_to_flat(i32 addrspace(3)* %group.ptr, i32 %cmp, i32 %val) #0 {
158   %cast = addrspacecast i32 addrspace(3)* %group.ptr to i32*
159   %ret = cmpxchg i32* %cast, i32 %cmp, i32 %val seq_cst monotonic
160   ret { i32, i1 } %ret
163 ; Not pointer operand
164 ; CHECK-LABEL: @cmpxchg_group_to_flat_wrong_operand(
165 ; CHECK: %cast.cmp = addrspacecast i32 addrspace(3)* %cmp.ptr to i32*
166 ; CHECK: %ret = cmpxchg i32* addrspace(3)* %cas.ptr, i32* %cast.cmp, i32* %val seq_cst monotonic
167 define { i32*, i1 } @cmpxchg_group_to_flat_wrong_operand(i32* addrspace(3)* %cas.ptr, i32 addrspace(3)* %cmp.ptr, i32* %val) #0 {
168   %cast.cmp = addrspacecast i32 addrspace(3)* %cmp.ptr to i32*
169   %ret = cmpxchg i32* addrspace(3)* %cas.ptr, i32* %cast.cmp, i32* %val seq_cst monotonic
170   ret { i32*, i1 } %ret
173 ; Null pointer in local addr space
174 ; CHECK-LABEL: @local_nullptr
175 ; CHECK: icmp ne i8 addrspace(3)* %a, addrspacecast (i8 addrspace(5)* null to i8 addrspace(3)*)
176 ; CHECK-NOT: i8 addrspace(3)* null
177 define void @local_nullptr(i32 addrspace(1)* nocapture %results, i8 addrspace(3)* %a) {
178 entry:
179   %tobool = icmp ne i8 addrspace(3)* %a, addrspacecast (i8 addrspace(5)* null to i8 addrspace(3)*)
180   %conv = zext i1 %tobool to i32
181   store i32 %conv, i32 addrspace(1)* %results, align 4
182   ret void
185 attributes #0 = { nounwind }