R600/SI: Use bcnt instruction for ctpop
[oota-llvm.git] / test / CodeGen / R600 / ctpop.ll
1 ; RUN: llc -march=r600 -mcpu=SI -verify-machineinstrs < %s | FileCheck -check-prefix=SI -check-prefix=FUNC %s
2 ; RUN: llc -march=r600 -mcpu=cypress -verify-machineinstrs < %s | FileCheck -check-prefix=EG -check-prefix=FUNC %s
3
4 declare i32 @llvm.ctpop.i32(i32) nounwind readnone
5 declare <2 x i32> @llvm.ctpop.v2i32(<2 x i32>) nounwind readnone
6 declare <4 x i32> @llvm.ctpop.v4i32(<4 x i32>) nounwind readnone
7 declare <8 x i32> @llvm.ctpop.v8i32(<8 x i32>) nounwind readnone
8 declare <16 x i32> @llvm.ctpop.v16i32(<16 x i32>) nounwind readnone
9
10 ; FUNC-LABEL: @s_ctpop_i32:
11 ; SI: S_LOAD_DWORD [[SVAL:s[0-9]+]],
12 ; SI: S_BCNT1_I32_B32 [[SRESULT:s[0-9]+]], [[SVAL]]
13 ; SI: V_MOV_B32_e32 [[VRESULT:v[0-9]+]], [[SRESULT]]
14 ; SI: BUFFER_STORE_DWORD [[VRESULT]],
15 ; SI: S_ENDPGM
16 define void @s_ctpop_i32(i32 addrspace(1)* noalias %out, i32 %val) nounwind {
17   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
18   store i32 %ctpop, i32 addrspace(1)* %out, align 4
19   ret void
20 }
21
22 ; XXX - Why 0 in register?
23 ; FUNC-LABEL: @v_ctpop_i32:
24 ; SI: BUFFER_LOAD_DWORD [[VAL:v[0-9]+]],
25 ; SI: V_MOV_B32_e32 [[VZERO:v[0-9]+]], 0
26 ; SI: V_BCNT_U32_B32_e32 [[RESULT:v[0-9]+]], [[VAL]], [[VZERO]]
27 ; SI: BUFFER_STORE_DWORD [[RESULT]],
28 ; SI: S_ENDPGM
29 define void @v_ctpop_i32(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in) nounwind {
30   %val = load i32 addrspace(1)* %in, align 4
31   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
32   store i32 %ctpop, i32 addrspace(1)* %out, align 4
33   ret void
34 }
35
36 ; FUNC-LABEL: @v_ctpop_add_chain_i32
37 ; SI: BUFFER_LOAD_DWORD [[VAL0:v[0-9]+]],
38 ; SI: BUFFER_LOAD_DWORD [[VAL1:v[0-9]+]],
39 ; SI: V_MOV_B32_e32 [[VZERO:v[0-9]+]], 0
40 ; SI: V_BCNT_U32_B32_e32 [[MIDRESULT:v[0-9]+]], [[VAL1]], [[VZERO]]
41 ; SI-NOT: ADD
42 ; SI: V_BCNT_U32_B32_e64 [[RESULT:v[0-9]+]], [[VAL0]], [[MIDRESULT]]
43 ; SI: BUFFER_STORE_DWORD [[RESULT]],
44 ; SI: S_ENDPGM
45 define void @v_ctpop_add_chain_i32(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in0, i32 addrspace(1)* noalias %in1) nounwind {
46   %val0 = load i32 addrspace(1)* %in0, align 4
47   %val1 = load i32 addrspace(1)* %in1, align 4
48   %ctpop0 = call i32 @llvm.ctpop.i32(i32 %val0) nounwind readnone
49   %ctpop1 = call i32 @llvm.ctpop.i32(i32 %val1) nounwind readnone
50   %add = add i32 %ctpop0, %ctpop1
51   store i32 %add, i32 addrspace(1)* %out, align 4
52   ret void
53 }
54
55 ; FUNC-LABEL: @v_ctpop_v2i32:
56 ; SI: V_BCNT_U32_B32_e32
57 ; SI: V_BCNT_U32_B32_e32
58 ; SI: S_ENDPGM
59 define void @v_ctpop_v2i32(<2 x i32> addrspace(1)* noalias %out, <2 x i32> addrspace(1)* noalias %in) nounwind {
60   %val = load <2 x i32> addrspace(1)* %in, align 8
61   %ctpop = call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %val) nounwind readnone
62   store <2 x i32> %ctpop, <2 x i32> addrspace(1)* %out, align 8
63   ret void
64 }
65
66 ; FUNC-LABEL: @v_ctpop_v4i32:
67 ; SI: V_BCNT_U32_B32_e32
68 ; SI: V_BCNT_U32_B32_e32
69 ; SI: V_BCNT_U32_B32_e32
70 ; SI: V_BCNT_U32_B32_e32
71 ; SI: S_ENDPGM
72 define void @v_ctpop_v4i32(<4 x i32> addrspace(1)* noalias %out, <4 x i32> addrspace(1)* noalias %in) nounwind {
73   %val = load <4 x i32> addrspace(1)* %in, align 16
74   %ctpop = call <4 x i32> @llvm.ctpop.v4i32(<4 x i32> %val) nounwind readnone
75   store <4 x i32> %ctpop, <4 x i32> addrspace(1)* %out, align 16
76   ret void
77 }
78
79 ; FUNC-LABEL: @v_ctpop_v8i32:
80 ; SI: V_BCNT_U32_B32_e32
81 ; SI: V_BCNT_U32_B32_e32
82 ; SI: V_BCNT_U32_B32_e32
83 ; SI: V_BCNT_U32_B32_e32
84 ; SI: V_BCNT_U32_B32_e32
85 ; SI: V_BCNT_U32_B32_e32
86 ; SI: V_BCNT_U32_B32_e32
87 ; SI: V_BCNT_U32_B32_e32
88 ; SI: S_ENDPGM
89 define void @v_ctpop_v8i32(<8 x i32> addrspace(1)* noalias %out, <8 x i32> addrspace(1)* noalias %in) nounwind {
90   %val = load <8 x i32> addrspace(1)* %in, align 32
91   %ctpop = call <8 x i32> @llvm.ctpop.v8i32(<8 x i32> %val) nounwind readnone
92   store <8 x i32> %ctpop, <8 x i32> addrspace(1)* %out, align 32
93   ret void
94 }
95
96 ; FUNC-LABEL: @v_ctpop_v16i32:
97 ; SI: V_BCNT_U32_B32_e32
98 ; SI: V_BCNT_U32_B32_e32
99 ; SI: V_BCNT_U32_B32_e32
100 ; SI: V_BCNT_U32_B32_e32
101 ; SI: V_BCNT_U32_B32_e32
102 ; SI: V_BCNT_U32_B32_e32
103 ; SI: V_BCNT_U32_B32_e32
104 ; SI: V_BCNT_U32_B32_e32
105 ; SI: V_BCNT_U32_B32_e32
106 ; SI: V_BCNT_U32_B32_e32
107 ; SI: V_BCNT_U32_B32_e32
108 ; SI: V_BCNT_U32_B32_e32
109 ; SI: V_BCNT_U32_B32_e32
110 ; SI: V_BCNT_U32_B32_e32
111 ; SI: V_BCNT_U32_B32_e32
112 ; SI: V_BCNT_U32_B32_e32
113 ; SI: S_ENDPGM
114 define void @v_ctpop_v16i32(<16 x i32> addrspace(1)* noalias %out, <16 x i32> addrspace(1)* noalias %in) nounwind {
115   %val = load <16 x i32> addrspace(1)* %in, align 32
116   %ctpop = call <16 x i32> @llvm.ctpop.v16i32(<16 x i32> %val) nounwind readnone
117   store <16 x i32> %ctpop, <16 x i32> addrspace(1)* %out, align 32
118   ret void
119 }
120
121 ; FUNC-LABEL: @v_ctpop_i32_add_inline_constant:
122 ; SI: BUFFER_LOAD_DWORD [[VAL:v[0-9]+]],
123 ; SI: V_BCNT_U32_B32_e64 [[RESULT:v[0-9]+]], [[VAL]], 4
124 ; SI: BUFFER_STORE_DWORD [[RESULT]],
125 ; SI: S_ENDPGM
126 define void @v_ctpop_i32_add_inline_constant(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in) nounwind {
127   %val = load i32 addrspace(1)* %in, align 4
128   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
129   %add = add i32 %ctpop, 4
130   store i32 %add, i32 addrspace(1)* %out, align 4
131   ret void
132 }
133
134 ; FUNC-LABEL: @v_ctpop_i32_add_inline_constant_inv:
135 ; SI: BUFFER_LOAD_DWORD [[VAL:v[0-9]+]],
136 ; SI: V_BCNT_U32_B32_e64 [[RESULT:v[0-9]+]], [[VAL]], 4
137 ; SI: BUFFER_STORE_DWORD [[RESULT]],
138 ; SI: S_ENDPGM
139 define void @v_ctpop_i32_add_inline_constant_inv(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in) nounwind {
140   %val = load i32 addrspace(1)* %in, align 4
141   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
142   %add = add i32 4, %ctpop
143   store i32 %add, i32 addrspace(1)* %out, align 4
144   ret void
145 }
146
147 ; FUNC-LABEL: @v_ctpop_i32_add_literal:
148 ; SI: BUFFER_LOAD_DWORD [[VAL:v[0-9]+]],
149 ; SI: V_MOV_B32_e32 [[LIT:v[0-9]+]], 0x1869f
150 ; SI: V_BCNT_U32_B32_e32 [[RESULT:v[0-9]+]], [[VAL]], [[LIT]]
151 ; SI: BUFFER_STORE_DWORD [[RESULT]],
152 ; SI: S_ENDPGM
153 define void @v_ctpop_i32_add_literal(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in) nounwind {
154   %val = load i32 addrspace(1)* %in, align 4
155   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
156   %add = add i32 %ctpop, 99999
157   store i32 %add, i32 addrspace(1)* %out, align 4
158   ret void
159 }
160
161 ; FUNC-LABEL: @v_ctpop_i32_add_var:
162 ; SI-DAG: BUFFER_LOAD_DWORD [[VAL:v[0-9]+]],
163 ; SI-DAG: S_LOAD_DWORD [[VAR:s[0-9]+]],
164 ; SI: V_BCNT_U32_B32_e64 [[RESULT:v[0-9]+]], [[VAL]], [[VAR]]
165 ; SI: BUFFER_STORE_DWORD [[RESULT]],
166 ; SI: S_ENDPGM
167 define void @v_ctpop_i32_add_var(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in, i32 %const) nounwind {
168   %val = load i32 addrspace(1)* %in, align 4
169   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
170   %add = add i32 %ctpop, %const
171   store i32 %add, i32 addrspace(1)* %out, align 4
172   ret void
173 }
174
175 ; FUNC-LABEL: @v_ctpop_i32_add_var_inv:
176 ; SI-DAG: BUFFER_LOAD_DWORD [[VAL:v[0-9]+]],
177 ; SI-DAG: S_LOAD_DWORD [[VAR:s[0-9]+]],
178 ; SI: V_BCNT_U32_B32_e64 [[RESULT:v[0-9]+]], [[VAL]], [[VAR]]
179 ; SI: BUFFER_STORE_DWORD [[RESULT]],
180 ; SI: S_ENDPGM
181 define void @v_ctpop_i32_add_var_inv(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in, i32 %const) nounwind {
182   %val = load i32 addrspace(1)* %in, align 4
183   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
184   %add = add i32 %const, %ctpop
185   store i32 %add, i32 addrspace(1)* %out, align 4
186   ret void
187 }
188
189 ; FUNC-LABEL: @v_ctpop_i32_add_vvar_inv
190 ; SI-DAG: BUFFER_LOAD_DWORD [[VAL:v[0-9]+]], {{.*}} + 0x0
191 ; SI-DAG: BUFFER_LOAD_DWORD [[VAR:v[0-9]+]], {{.*}} + 0x10
192 ; SI: V_BCNT_U32_B32_e32 [[RESULT:v[0-9]+]], [[VAL]], [[VAR]]
193 ; SI: BUFFER_STORE_DWORD [[RESULT]],
194 ; SI: S_ENDPGM
195 define void @v_ctpop_i32_add_vvar_inv(i32 addrspace(1)* noalias %out, i32 addrspace(1)* noalias %in, i32 addrspace(1)* noalias %constptr) nounwind {
196   %val = load i32 addrspace(1)* %in, align 4
197   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
198   %gep = getelementptr i32 addrspace(1)* %constptr, i32 4
199   %const = load i32 addrspace(1)* %gep, align 4
200   %add = add i32 %const, %ctpop
201   store i32 %add, i32 addrspace(1)* %out, align 4
202   ret void
203 }