1; RUN: llc -march=amdgcn -mcpu=tonga -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefix=VI -check-prefix=GCN %s
2
3; FIXME: Need to handle non-uniform case for function below (load without gep).
4; GCN-LABEL: {{^}}v_test_add_i16:
5; VI: flat_load_ushort [[A:v[0-9]+]]
6; VI: flat_load_ushort [[B:v[0-9]+]]
7; VI: v_add_u16_e32 [[ADD:v[0-9]+]], [[A]], [[B]]
8; VI-NEXT: buffer_store_short [[ADD]]
9define amdgpu_kernel void @v_test_add_i16(i16 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 {
10  %tid = call i32 @llvm.amdgcn.workitem.id.x()
11  %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid
12  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
13  %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid
14  %a = load volatile i16, i16 addrspace(1)* %gep.in0
15  %b = load volatile i16, i16 addrspace(1)* %gep.in1
16  %add = add i16 %a, %b
17  store i16 %add, i16 addrspace(1)* %out
18  ret void
19}
20
21; FIXME: Need to handle non-uniform case for function below (load without gep).
22; GCN-LABEL: {{^}}v_test_add_i16_constant:
23; VI: flat_load_ushort [[A:v[0-9]+]]
24; VI: v_add_u16_e32 [[ADD:v[0-9]+]], 0x7b, [[A]]
25; VI-NEXT: buffer_store_short [[ADD]]
26define amdgpu_kernel void @v_test_add_i16_constant(i16 addrspace(1)* %out, i16 addrspace(1)* %in0) #1 {
27  %tid = call i32 @llvm.amdgcn.workitem.id.x()
28  %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid
29  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
30  %a = load volatile i16, i16 addrspace(1)* %gep.in0
31  %add = add i16 %a, 123
32  store i16 %add, i16 addrspace(1)* %out
33  ret void
34}
35
36; FIXME: Need to handle non-uniform case for function below (load without gep).
37; GCN-LABEL: {{^}}v_test_add_i16_neg_constant:
38; VI: flat_load_ushort [[A:v[0-9]+]]
39; VI: v_add_u16_e32 [[ADD:v[0-9]+]], 0xfcb3, [[A]]
40; VI-NEXT: buffer_store_short [[ADD]]
41define amdgpu_kernel void @v_test_add_i16_neg_constant(i16 addrspace(1)* %out, i16 addrspace(1)* %in0) #1 {
42  %tid = call i32 @llvm.amdgcn.workitem.id.x()
43  %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid
44  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
45  %a = load volatile i16, i16 addrspace(1)* %gep.in0
46  %add = add i16 %a, -845
47  store i16 %add, i16 addrspace(1)* %out
48  ret void
49}
50
51; FIXME: Need to handle non-uniform case for function below (load without gep).
52; GCN-LABEL: {{^}}v_test_add_i16_inline_neg1:
53; VI: flat_load_ushort [[A:v[0-9]+]]
54; VI: v_add_u16_e32 [[ADD:v[0-9]+]], -1, [[A]]
55; VI-NEXT: buffer_store_short [[ADD]]
56define amdgpu_kernel void @v_test_add_i16_inline_neg1(i16 addrspace(1)* %out, i16 addrspace(1)* %in0) #1 {
57  %tid = call i32 @llvm.amdgcn.workitem.id.x()
58  %gep.out = getelementptr inbounds i16, i16 addrspace(1)* %out, i32 %tid
59  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
60  %a = load volatile i16, i16 addrspace(1)* %gep.in0
61  %add = add i16 %a, -1
62  store i16 %add, i16 addrspace(1)* %out
63  ret void
64}
65
66; FIXME: Need to handle non-uniform case for function below (load without gep).
67; GCN-LABEL: {{^}}v_test_add_i16_zext_to_i32:
68; VI: flat_load_ushort [[A:v[0-9]+]]
69; VI: flat_load_ushort [[B:v[0-9]+]]
70; VI: v_add_u16_e32 [[ADD:v[0-9]+]], [[A]], [[B]]
71; VI-NEXT: buffer_store_dword [[ADD]]
72define amdgpu_kernel void @v_test_add_i16_zext_to_i32(i32 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 {
73  %tid = call i32 @llvm.amdgcn.workitem.id.x()
74  %gep.out = getelementptr inbounds i32, i32 addrspace(1)* %out, i32 %tid
75  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
76  %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid
77  %a = load volatile i16, i16 addrspace(1)* %gep.in0
78  %b = load volatile i16, i16 addrspace(1)* %gep.in1
79  %add = add i16 %a, %b
80  %ext = zext i16 %add to i32
81  store i32 %ext, i32 addrspace(1)* %out
82  ret void
83}
84
85; FIXME: Need to handle non-uniform case for function below (load without gep).
86; GCN-LABEL: {{^}}v_test_add_i16_zext_to_i64:
87; VI: flat_load_ushort [[A:v[0-9]+]]
88; VI: flat_load_ushort [[B:v[0-9]+]]
89; VI-DAG: v_add_u16_e32 v[[ADD:[0-9]+]], [[A]], [[B]]
90; VI: buffer_store_dwordx2 v{{\[}}[[ADD]]:{{[0-9]+\]}}, off, {{s\[[0-9]+:[0-9]+\]}}, 0{{$}}
91define amdgpu_kernel void @v_test_add_i16_zext_to_i64(i64 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 {
92  %tid = call i32 @llvm.amdgcn.workitem.id.x()
93  %gep.out = getelementptr inbounds i64, i64 addrspace(1)* %out, i32 %tid
94  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
95  %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid
96  %a = load volatile i16, i16 addrspace(1)* %gep.in0
97  %b = load volatile i16, i16 addrspace(1)* %gep.in1
98  %add = add i16 %a, %b
99  %ext = zext i16 %add to i64
100  store i64 %ext, i64 addrspace(1)* %out
101  ret void
102}
103
104; FIXME: Need to handle non-uniform case for function below (load without gep).
105; GCN-LABEL: {{^}}v_test_add_i16_sext_to_i32:
106; VI: flat_load_ushort [[A:v[0-9]+]]
107; VI: flat_load_ushort [[B:v[0-9]+]]
108; VI: v_add_u16_e32 [[ADD:v[0-9]+]],  [[A]], [[B]]
109; VI-NEXT: v_bfe_i32 [[SEXT:v[0-9]+]], [[ADD]], 0, 16
110; VI-NEXT: buffer_store_dword [[SEXT]]
111define amdgpu_kernel void @v_test_add_i16_sext_to_i32(i32 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 {
112  %tid = call i32 @llvm.amdgcn.workitem.id.x()
113  %gep.out = getelementptr inbounds i32, i32 addrspace(1)* %out, i32 %tid
114  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
115  %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid
116  %a = load i16, i16 addrspace(1)* %gep.in0
117  %b = load i16, i16 addrspace(1)* %gep.in1
118  %add = add i16 %a, %b
119  %ext = sext i16 %add to i32
120  store i32 %ext, i32 addrspace(1)* %out
121  ret void
122}
123
124; FIXME: Need to handle non-uniform case for function below (load without gep).
125; GCN-LABEL: {{^}}v_test_add_i16_sext_to_i64:
126; VI: flat_load_ushort [[A:v[0-9]+]]
127; VI: flat_load_ushort [[B:v[0-9]+]]
128; VI: v_add_u16_e32 [[ADD:v[0-9]+]], [[A]], [[B]]
129; VI-NEXT: v_bfe_i32 v[[LO:[0-9]+]], [[ADD]], 0, 16
130; VI-NEXT: v_ashrrev_i32_e32 v[[HI:[0-9]+]], 31, v[[LO]]
131; VI-NEXT: buffer_store_dwordx2 v{{\[}}[[LO]]:[[HI]]{{\]}}
132define amdgpu_kernel void @v_test_add_i16_sext_to_i64(i64 addrspace(1)* %out, i16 addrspace(1)* %in0, i16 addrspace(1)* %in1) #1 {
133  %tid = call i32 @llvm.amdgcn.workitem.id.x()
134  %gep.out = getelementptr inbounds i64, i64 addrspace(1)* %out, i32 %tid
135  %gep.in0 = getelementptr inbounds i16, i16 addrspace(1)* %in0, i32 %tid
136  %gep.in1 = getelementptr inbounds i16, i16 addrspace(1)* %in1, i32 %tid
137  %a = load i16, i16 addrspace(1)* %gep.in0
138  %b = load i16, i16 addrspace(1)* %gep.in1
139  %add = add i16 %a, %b
140  %ext = sext i16 %add to i64
141  store i64 %ext, i64 addrspace(1)* %out
142  ret void
143}
144
145declare i32 @llvm.amdgcn.workitem.id.x() #0
146
147attributes #0 = { nounwind readnone }
148attributes #1 = { nounwind }
149