1; RUN: llc -verify-machineinstrs -march=amdgcn < %s | FileCheck -check-prefix=GCN %s
2; RUN: llc -verify-machineinstrs -march=amdgcn -mcpu=tonga < %s | FileCheck -check-prefix=GCN %s
3
4; When a frame index offset is more than 12-bits, make sure we don't store
5; it in mubuf's offset field.
6
7; Also, make sure we use the same register for storing the scratch buffer addresss
8; for both stores. This register is allocated by the register scavenger, so we
9; should be able to reuse the same regiser for each scratch buffer access.
10
11; GCN-LABEL: {{^}}legal_offset_fi:
12; GCN: buffer_store_dword v{{[0-9]+}}, off, s[{{[0-9]+}}:{{[0-9]+}}], s{{[0-9]+}} offset:4{{$}}
13; GCN: v_mov_b32_e32 [[OFFSET:v[0-9]+]], 0x8004
14; GCN: buffer_store_dword v{{[0-9]+}}, [[OFFSET]], s[{{[0-9]+}}:{{[0-9]+}}], s{{[0-9]+}} offen{{$}}
15
16define amdgpu_kernel void @legal_offset_fi(i32 addrspace(1)* %out, i32 %cond, i32 %if_offset, i32 %else_offset) {
17entry:
18  %scratch0 = alloca [8192 x i32]
19  %scratch1 = alloca [8192 x i32]
20
21  %scratchptr0 = getelementptr [8192 x i32], [8192 x i32]* %scratch0, i32 0, i32 0
22  store i32 1, i32* %scratchptr0
23
24  %scratchptr1 = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 0
25  store i32 2, i32* %scratchptr1
26
27  %cmp = icmp eq i32 %cond, 0
28  br i1 %cmp, label %if, label %else
29
30if:
31  %if_ptr = getelementptr [8192 x i32], [8192 x i32]* %scratch0, i32 0, i32 %if_offset
32  %if_value = load i32, i32* %if_ptr
33  br label %done
34
35else:
36  %else_ptr = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 %else_offset
37  %else_value = load i32, i32* %else_ptr
38  br label %done
39
40done:
41  %value = phi i32 [%if_value, %if], [%else_value, %else]
42  store i32 %value, i32 addrspace(1)* %out
43  ret void
44
45  ret void
46
47}
48
49; GCN-LABEL: {{^}}legal_offset_fi_offset:
50; GCN-DAG: buffer_store_dword v{{[0-9]+}}, v{{[0-9]+}}, s[{{[0-9]+}}:{{[0-9]+}}], s{{[0-9]+}} offen{{$}}
51; This constant isn't folded, because it has multiple uses.
52; GCN-DAG: v_mov_b32_e32 [[K8000:v[0-9]+]], 0x8004
53; GCN-DAG: v_add_i32_e32 [[OFFSET:v[0-9]+]], vcc, [[K8000]]
54; GCN: buffer_store_dword v{{[0-9]+}}, [[OFFSET]], s[{{[0-9]+}}:{{[0-9]+}}], s{{[0-9]+}} offen{{$}}
55
56define amdgpu_kernel void @legal_offset_fi_offset(i32 addrspace(1)* %out, i32 %cond, i32 addrspace(1)* %offsets, i32 %if_offset, i32 %else_offset) {
57entry:
58  %scratch0 = alloca [8192 x i32]
59  %scratch1 = alloca [8192 x i32]
60
61  %offset0 = load i32, i32 addrspace(1)* %offsets
62  %scratchptr0 = getelementptr [8192 x i32], [8192 x i32]* %scratch0, i32 0, i32 %offset0
63  store i32 %offset0, i32* %scratchptr0
64
65  %offsetptr1 = getelementptr i32, i32 addrspace(1)* %offsets, i32 1
66  %offset1 = load i32, i32 addrspace(1)* %offsetptr1
67  %scratchptr1 = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 %offset1
68  store i32 %offset1, i32* %scratchptr1
69
70  %cmp = icmp eq i32 %cond, 0
71  br i1 %cmp, label %if, label %else
72
73if:
74  %if_ptr = getelementptr [8192 x i32], [8192 x i32]* %scratch0, i32 0, i32 %if_offset
75  %if_value = load i32, i32* %if_ptr
76  br label %done
77
78else:
79  %else_ptr = getelementptr [8192 x i32], [8192 x i32]* %scratch1, i32 0, i32 %else_offset
80  %else_value = load i32, i32* %else_ptr
81  br label %done
82
83done:
84  %value = phi i32 [%if_value, %if], [%else_value, %else]
85  store i32 %value, i32 addrspace(1)* %out
86  ret void
87}
88
89; GCN-LABEL: {{^}}neg_vaddr_offset:
90; GCN: buffer_store_dword v{{[0-9]+}}, v{{[0-9]+}}, s[{{[0-9]+:[0-9]+}}], s{{[0-9]+}} offen offset:16{{$}}
91define amdgpu_kernel void @neg_vaddr_offset(i32 %offset) {
92entry:
93  %array = alloca [8192 x i32]
94  %ptr_offset = add i32 %offset, 4
95  %ptr = getelementptr [8192 x i32], [8192 x i32]* %array, i32 0, i32 %ptr_offset
96  store i32 0, i32* %ptr
97  ret void
98}
99
100; GCN-LABEL: {{^}}pos_vaddr_offset:
101; GCN: buffer_store_dword v{{[0-9]+}}, off, s[{{[0-9]+:[0-9]+}}], s{{[0-9]+}} offset:20
102define amdgpu_kernel void @pos_vaddr_offset(i32 addrspace(1)* %out, i32 %offset) {
103entry:
104  %array = alloca [8192 x i32]
105  %ptr = getelementptr [8192 x i32], [8192 x i32]* %array, i32 0, i32 4
106  store i32 0, i32* %ptr
107  %load_ptr = getelementptr [8192 x i32], [8192 x i32]* %array, i32 0, i32 %offset
108  %val = load i32, i32* %load_ptr
109  store i32 %val, i32 addrspace(1)* %out
110  ret void
111}
112