1; RUN: llc -march=amdgcn -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=SI %s
2; RUN: llc -march=amdgcn -mcpu=fiji -mattr=-flat-for-global -verify-machineinstrs < %s | FileCheck -check-prefix=GCN -check-prefix=VI %s
3
4; GCN-LABEL: {{^}}select_f16:
5; GCN: buffer_load_ushort v[[A_F16:[0-9]+]]
6; GCN: buffer_load_ushort v[[B_F16:[0-9]+]]
7; GCN: buffer_load_ushort v[[C_F16:[0-9]+]]
8; GCN: buffer_load_ushort v[[D_F16:[0-9]+]]
9; SI:  v_cvt_f32_f16_e32 v[[A_F32:[0-9]+]], v[[A_F16]]
10; SI:  v_cvt_f32_f16_e32 v[[B_F32:[0-9]+]], v[[B_F16]]
11; SI:  v_cmp_lt_f32_e32 vcc, v[[A_F32]], v[[B_F32]]
12; SI:  v_cvt_f32_f16_e32 v[[C_F32:[0-9]+]], v[[C_F16]]
13; SI:  v_cvt_f32_f16_e32 v[[D_F32:[0-9]+]], v[[D_F16]]
14; SI:  v_cndmask_b32_e32 v[[R_F32:[0-9]+]], v[[D_F32]], v[[C_F32]]
15; SI:  v_cvt_f16_f32_e32 v[[R_F16:[0-9]+]], v[[R_F32]]
16; VI:  v_cmp_lt_f16_e32 vcc, v[[A_F16]], v[[B_F16]]
17; VI:  v_cndmask_b32_e32 v[[R_F16:[0-9]+]], v[[D_F16]], v[[C_F16]], vcc
18; GCN: buffer_store_short v[[R_F16]]
19; GCN: s_endpgm
20define void @select_f16(
21    half addrspace(1)* %r,
22    half addrspace(1)* %a,
23    half addrspace(1)* %b,
24    half addrspace(1)* %c,
25    half addrspace(1)* %d) {
26entry:
27  %a.val = load half, half addrspace(1)* %a
28  %b.val = load half, half addrspace(1)* %b
29  %c.val = load half, half addrspace(1)* %c
30  %d.val = load half, half addrspace(1)* %d
31  %fcmp = fcmp olt half %a.val, %b.val
32  %r.val = select i1 %fcmp, half %c.val, half %d.val
33  store half %r.val, half addrspace(1)* %r
34  ret void
35}
36
37; GCN-LABEL: {{^}}select_f16_imm_a:
38; GCN: buffer_load_ushort v[[B_F16:[0-9]+]]
39; GCN: buffer_load_ushort v[[C_F16:[0-9]+]]
40; GCN: buffer_load_ushort v[[D_F16:[0-9]+]]
41; SI:  v_cvt_f32_f16_e32 v[[B_F32:[0-9]+]], v[[B_F16]]
42; SI:  v_cmp_lt_f32_e32 vcc, 0.5, v[[B_F32]]
43; SI:  v_cvt_f32_f16_e32 v[[C_F32:[0-9]+]], v[[C_F16]]
44; SI:  v_cvt_f32_f16_e32 v[[D_F32:[0-9]+]], v[[D_F16]]
45; SI:  v_cndmask_b32_e32 v[[R_F32:[0-9]+]], v[[D_F32]], v[[C_F32]]
46; SI:  v_cvt_f16_f32_e32 v[[R_F16:[0-9]+]], v[[R_F32]]
47; VI:  v_cmp_lt_f16_e32 vcc, 0.5, v[[B_F16]]
48; VI:  v_cndmask_b32_e32 v[[R_F16:[0-9]+]], v[[D_F16]], v[[C_F16]], vcc
49; GCN: buffer_store_short v[[R_F16]]
50; GCN: s_endpgm
51define void @select_f16_imm_a(
52    half addrspace(1)* %r,
53    half addrspace(1)* %b,
54    half addrspace(1)* %c,
55    half addrspace(1)* %d) {
56entry:
57  %b.val = load half, half addrspace(1)* %b
58  %c.val = load half, half addrspace(1)* %c
59  %d.val = load half, half addrspace(1)* %d
60  %fcmp = fcmp olt half 0xH3800, %b.val
61  %r.val = select i1 %fcmp, half %c.val, half %d.val
62  store half %r.val, half addrspace(1)* %r
63  ret void
64}
65
66; GCN-LABEL: {{^}}select_f16_imm_b:
67; GCN: buffer_load_ushort v[[A_F16:[0-9]+]]
68; GCN: buffer_load_ushort v[[C_F16:[0-9]+]]
69; GCN: buffer_load_ushort v[[D_F16:[0-9]+]]
70; SI:  v_cvt_f32_f16_e32 v[[A_F32:[0-9]+]], v[[A_F16]]
71; SI:  v_cmp_gt_f32_e32 vcc, 0.5, v[[A_F32]]
72; SI:  v_cvt_f32_f16_e32 v[[C_F32:[0-9]+]], v[[C_F16]]
73; SI:  v_cvt_f32_f16_e32 v[[D_F32:[0-9]+]], v[[D_F16]]
74; SI:  v_cndmask_b32_e32 v[[R_F32:[0-9]+]], v[[D_F32]], v[[C_F32]]
75; SI:  v_cvt_f16_f32_e32 v[[R_F16:[0-9]+]], v[[R_F32]]
76
77; VI:  v_cmp_gt_f16_e32 vcc, 0.5, v[[A_F16]]
78; VI:  v_cndmask_b32_e32 v[[R_F16:[0-9]+]], v[[D_F16]], v[[C_F16]], vcc
79; GCN: buffer_store_short v[[R_F16]]
80; GCN: s_endpgm
81define void @select_f16_imm_b(
82    half addrspace(1)* %r,
83    half addrspace(1)* %a,
84    half addrspace(1)* %c,
85    half addrspace(1)* %d) {
86entry:
87  %a.val = load half, half addrspace(1)* %a
88  %c.val = load half, half addrspace(1)* %c
89  %d.val = load half, half addrspace(1)* %d
90  %fcmp = fcmp olt half %a.val, 0xH3800
91  %r.val = select i1 %fcmp, half %c.val, half %d.val
92  store half %r.val, half addrspace(1)* %r
93  ret void
94}
95
96; GCN-LABEL: {{^}}select_f16_imm_c:
97; GCN: buffer_load_ushort v[[A_F16:[0-9]+]]
98; GCN: buffer_load_ushort v[[B_F16:[0-9]+]]
99; GCN: buffer_load_ushort v[[D_F16:[0-9]+]]
100; SI:  v_cvt_f32_f16_e32 v[[A_F32:[0-9]+]], v[[A_F16]]
101; SI:  v_cvt_f32_f16_e32 v[[B_F32:[0-9]+]], v[[B_F16]]
102; SI:  v_cvt_f32_f16_e32 v[[D_F32:[0-9]+]], v[[D_F16]]
103; SI:  v_cmp_nlt_f32_e32 vcc, v[[A_F32]], v[[B_F32]]
104; SI:  v_cndmask_b32_e32 v[[R_F32:[0-9]+]], 0.5, v[[D_F32]], vcc
105; SI:  v_cvt_f16_f32_e32 v[[R_F16:[0-9]+]], v[[R_F32]]
106
107; VI:  v_cmp_nlt_f16_e32 vcc, v[[A_F16]], v[[B_F16]]
108; VI:  v_mov_b32_e32 v[[C_F16:[0-9]+]], 0x3800{{$}}
109; VI:  v_cndmask_b32_e32 v[[R_F16:[0-9]+]], v[[C_F16]], v[[D_F16]], vcc
110; GCN: buffer_store_short v[[R_F16]]
111; GCN: s_endpgm
112define void @select_f16_imm_c(
113    half addrspace(1)* %r,
114    half addrspace(1)* %a,
115    half addrspace(1)* %b,
116    half addrspace(1)* %d) {
117entry:
118  %a.val = load half, half addrspace(1)* %a
119  %b.val = load half, half addrspace(1)* %b
120  %d.val = load half, half addrspace(1)* %d
121  %fcmp = fcmp olt half %a.val, %b.val
122  %r.val = select i1 %fcmp, half 0xH3800, half %d.val
123  store half %r.val, half addrspace(1)* %r
124  ret void
125}
126
127; GCN-LABEL: {{^}}select_f16_imm_d:
128; GCN: buffer_load_ushort v[[A_F16:[0-9]+]]
129; GCN: buffer_load_ushort v[[B_F16:[0-9]+]]
130; GCN: buffer_load_ushort v[[C_F16:[0-9]+]]
131; SI:  v_cvt_f32_f16_e32 v[[A_F32:[0-9]+]], v[[A_F16]]
132; SI:  v_cvt_f32_f16_e32 v[[B_F32:[0-9]+]], v[[B_F16]]
133; SI:  v_cvt_f32_f16_e32 v[[C_F32:[0-9]+]], v[[C_F16]]
134; SI:  v_cmp_lt_f32_e32 vcc, v[[A_F32]], v[[B_F32]]
135; SI:  v_cndmask_b32_e32 v[[R_F32:[0-9]+]], 0.5, v[[C_F32]]
136; SI:  v_cvt_f16_f32_e32 v[[R_F16:[0-9]+]], v[[R_F32]]
137; VI:  v_cmp_lt_f16_e32 vcc, v[[A_F16]], v[[B_F16]]
138; VI:  v_mov_b32_e32 v[[D_F16:[0-9]+]], 0x3800{{$}}
139; VI:  v_cndmask_b32_e32 v[[R_F16:[0-9]+]], v[[D_F16]], v[[C_F16]], vcc
140; GCN: buffer_store_short v[[R_F16]]
141; GCN: s_endpgm
142define void @select_f16_imm_d(
143    half addrspace(1)* %r,
144    half addrspace(1)* %a,
145    half addrspace(1)* %b,
146    half addrspace(1)* %c) {
147entry:
148  %a.val = load half, half addrspace(1)* %a
149  %b.val = load half, half addrspace(1)* %b
150  %c.val = load half, half addrspace(1)* %c
151  %fcmp = fcmp olt half %a.val, %b.val
152  %r.val = select i1 %fcmp, half %c.val, half 0xH3800
153  store half %r.val, half addrspace(1)* %r
154  ret void
155}
156
157; GCN-LABEL: {{^}}select_v2f16:
158; SI:  v_cvt_f32_f16_e32
159; SI:  v_cvt_f32_f16_e32
160; SI:  v_cvt_f32_f16_e32
161; SI:  v_cvt_f32_f16_e32
162; SI:  v_cmp_lt_f32_e64
163; SI:  v_cmp_lt_f32_e32
164; VI:  v_cmp_lt_f16_e32
165; VI:  v_cmp_lt_f16_e64
166; GCN: v_cndmask_b32_e32
167; GCN: v_cndmask_b32_e64
168; SI:  v_cvt_f16_f32_e32
169; SI:  v_cvt_f16_f32_e32
170; GCN: s_endpgm
171define void @select_v2f16(
172    <2 x half> addrspace(1)* %r,
173    <2 x half> addrspace(1)* %a,
174    <2 x half> addrspace(1)* %b,
175    <2 x half> addrspace(1)* %c,
176    <2 x half> addrspace(1)* %d) {
177entry:
178  %a.val = load <2 x half>, <2 x half> addrspace(1)* %a
179  %b.val = load <2 x half>, <2 x half> addrspace(1)* %b
180  %c.val = load <2 x half>, <2 x half> addrspace(1)* %c
181  %d.val = load <2 x half>, <2 x half> addrspace(1)* %d
182  %fcmp = fcmp olt <2 x half> %a.val, %b.val
183  %r.val = select <2 x i1> %fcmp, <2 x half> %c.val, <2 x half> %d.val
184  store <2 x half> %r.val, <2 x half> addrspace(1)* %r
185  ret void
186}
187
188; GCN-LABEL: {{^}}select_v2f16_imm_a:
189; SI:  v_cvt_f32_f16_e32
190; SI:  v_cvt_f32_f16_e32
191; SI:  v_cmp_lt_f32_e32 vcc, 0.5
192; SI:  v_cvt_f32_f16_e32
193; SI:  v_cvt_f32_f16_e32
194; SI:  v_cvt_f32_f16_e32
195; SI:  v_cvt_f32_f16_e32
196; SI:  v_cmp_lt_f32_e64
197
198; VI:  v_cmp_lt_f16_e32
199; VI:  v_cmp_lt_f16_e64
200; GCN: v_cndmask_b32_e32
201; SI:  v_cvt_f16_f32_e32
202; GCN: v_cndmask_b32_e64
203; SI:  v_cvt_f16_f32_e32
204; GCN: s_endpgm
205define void @select_v2f16_imm_a(
206    <2 x half> addrspace(1)* %r,
207    <2 x half> addrspace(1)* %b,
208    <2 x half> addrspace(1)* %c,
209    <2 x half> addrspace(1)* %d) {
210entry:
211  %b.val = load <2 x half>, <2 x half> addrspace(1)* %b
212  %c.val = load <2 x half>, <2 x half> addrspace(1)* %c
213  %d.val = load <2 x half>, <2 x half> addrspace(1)* %d
214  %fcmp = fcmp olt <2 x half> <half 0xH3800, half 0xH3900>, %b.val
215  %r.val = select <2 x i1> %fcmp, <2 x half> %c.val, <2 x half> %d.val
216  store <2 x half> %r.val, <2 x half> addrspace(1)* %r
217  ret void
218}
219
220; GCN-LABEL: {{^}}select_v2f16_imm_b:
221; SI:  v_cvt_f32_f16_e32
222; SI:  v_cvt_f32_f16_e32
223; SI:  v_cmp_gt_f32_e32 vcc, 0.5
224; SI:  v_cvt_f32_f16_e32
225; SI:  v_cvt_f32_f16_e32
226; SI:  v_cvt_f32_f16_e32
227; SI:  v_cvt_f32_f16_e32
228; SI:  v_cmp_gt_f32_e64
229
230; VI:  v_cmp_gt_f16_e32
231; VI:  v_cmp_gt_f16_e64
232; GCN: v_cndmask_b32_e32
233; SI:  v_cvt_f16_f32_e32
234; GCN: v_cndmask_b32_e64
235; SI:  v_cvt_f16_f32_e32
236; GCN: s_endpgm
237define void @select_v2f16_imm_b(
238    <2 x half> addrspace(1)* %r,
239    <2 x half> addrspace(1)* %a,
240    <2 x half> addrspace(1)* %c,
241    <2 x half> addrspace(1)* %d) {
242entry:
243  %a.val = load <2 x half>, <2 x half> addrspace(1)* %a
244  %c.val = load <2 x half>, <2 x half> addrspace(1)* %c
245  %d.val = load <2 x half>, <2 x half> addrspace(1)* %d
246  %fcmp = fcmp olt <2 x half> %a.val, <half 0xH3800, half 0xH3900>
247  %r.val = select <2 x i1> %fcmp, <2 x half> %c.val, <2 x half> %d.val
248  store <2 x half> %r.val, <2 x half> addrspace(1)* %r
249  ret void
250}
251
252; GCN-LABEL: {{^}}select_v2f16_imm_c:
253; SI:  v_cvt_f32_f16_e32
254; SI:  v_cvt_f32_f16_e32
255; SI:  v_cvt_f32_f16_e32
256; SI:  v_cvt_f32_f16_e32
257; SI:  v_cvt_f32_f16_e32
258; SI:  v_cvt_f32_f16_e32
259
260; SI: v_cmp_nlt_f32_e32
261; SI: v_cndmask_b32_e32
262; SI: v_cmp_nlt_f32_e32
263; SI: v_cndmask_b32_e32
264
265; VI: v_cmp_nlt_f16_e32
266; VI: v_cndmask_b32_e32
267
268; VI: v_cmp_nlt_f16_e32
269; VI: v_cndmask_b32_e32
270
271; SI:  v_cvt_f16_f32_e32
272; SI:  v_cvt_f16_f32_e32
273; GCN: s_endpgm
274define void @select_v2f16_imm_c(
275    <2 x half> addrspace(1)* %r,
276    <2 x half> addrspace(1)* %a,
277    <2 x half> addrspace(1)* %b,
278    <2 x half> addrspace(1)* %d) {
279entry:
280  %a.val = load <2 x half>, <2 x half> addrspace(1)* %a
281  %b.val = load <2 x half>, <2 x half> addrspace(1)* %b
282  %d.val = load <2 x half>, <2 x half> addrspace(1)* %d
283  %fcmp = fcmp olt <2 x half> %a.val, %b.val
284  %r.val = select <2 x i1> %fcmp, <2 x half> <half 0xH3800, half 0xH3900>, <2 x half> %d.val
285  store <2 x half> %r.val, <2 x half> addrspace(1)* %r
286  ret void
287}
288
289; GCN-LABEL: {{^}}select_v2f16_imm_d:
290; SI:  v_cvt_f32_f16_e32
291; SI:  v_cvt_f32_f16_e32
292; SI:  v_cvt_f32_f16_e32
293; SI:  v_cvt_f32_f16_e32
294; SI:  v_cvt_f32_f16_e32
295; SI:  v_cvt_f32_f16_e32
296; SI:  v_cmp_lt_f32_e32
297; SI:  v_cmp_lt_f32_e64
298; VI:  v_cmp_lt_f16_e32
299; VI:  v_cmp_lt_f16_e64
300; GCN: v_cndmask_b32_e32
301; GCN: v_cndmask_b32_e64
302; SI:  v_cvt_f16_f32_e32
303; SI:  v_cvt_f16_f32_e32
304; GCN: s_endpgm
305define void @select_v2f16_imm_d(
306    <2 x half> addrspace(1)* %r,
307    <2 x half> addrspace(1)* %a,
308    <2 x half> addrspace(1)* %b,
309    <2 x half> addrspace(1)* %c) {
310entry:
311  %a.val = load <2 x half>, <2 x half> addrspace(1)* %a
312  %b.val = load <2 x half>, <2 x half> addrspace(1)* %b
313  %c.val = load <2 x half>, <2 x half> addrspace(1)* %c
314  %fcmp = fcmp olt <2 x half> %a.val, %b.val
315  %r.val = select <2 x i1> %fcmp, <2 x half> %c.val, <2 x half> <half 0xH3800, half 0xH3900>
316  store <2 x half> %r.val, <2 x half> addrspace(1)* %r
317  ret void
318}
319