blob: dc698ac61b703c6992fda0e227546e06961686c4 [file] [log] [blame]
Matt Arsenault5660bb62019-11-18 16:48:07 +05301; RUN: llc -amdgpu-scalarize-global-loads=false -march=amdgcn -mcpu=fiji -amdgpu-sdwa-peephole=0 -verify-machineinstrs < %s | FileCheck -enable-var-scope -check-prefixes=NOSDWA,GCN %s
2; RUN: llc -amdgpu-scalarize-global-loads=false -march=amdgcn -mcpu=fiji -amdgpu-sdwa-peephole -verify-machineinstrs < %s | FileCheck -enable-var-scope -check-prefixes=VI,GFX89,SDWA,GCN %s
3; RUN: llc -amdgpu-scalarize-global-loads=false -march=amdgcn -mcpu=gfx900 -amdgpu-sdwa-peephole -verify-machineinstrs < %s | FileCheck -enable-var-scope -check-prefixes=GFX9,GFX9_10,SDWA,GCN %s
4; RUN: llc -amdgpu-scalarize-global-loads=false -march=amdgcn -mcpu=gfx1010 -amdgpu-sdwa-peephole -verify-machineinstrs < %s | FileCheck -enable-var-scope -check-prefixes=GFX10,GFX9_10,SDWA,GCN %s
Sam Koltonf60ad582017-03-21 12:51:34 +00005
6; GCN-LABEL: {{^}}add_shr_i32:
7; NOSDWA: v_lshrrev_b32_e32 v[[DST:[0-9]+]], 16, v{{[0-9]+}}
Dmitry Preobrazhenskya0342dc2017-11-20 18:24:21 +00008; NOSDWA: v_add_u32_e32 v{{[0-9]+}}, vcc, v{{[0-9]+}}, v[[DST]]
Matt Arsenault9a7e29a2017-11-29 02:25:14 +00009; NOSDWA-NOT: v_add_{{(_co)?}}_u32_sdwa
Sam Koltonf60ad582017-03-21 12:51:34 +000010
Matt Arsenault84445dd2017-11-30 22:51:26 +000011; VI: v_add_u32_sdwa v{{[0-9]+}}, vcc, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:WORD_1
12; GFX9: v_add_u32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:WORD_1
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +000013; GFX10: v_add_nc_u32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:WORD_1
Sam Koltonf60ad582017-03-21 12:51:34 +000014
Matt Arsenault5660bb62019-11-18 16:48:07 +053015define amdgpu_kernel void @add_shr_i32(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +000016 %a = load i32, i32 addrspace(1)* %in, align 4
17 %shr = lshr i32 %a, 16
18 %add = add i32 %a, %shr
19 store i32 %add, i32 addrspace(1)* %out, align 4
20 ret void
21}
22
23; GCN-LABEL: {{^}}sub_shr_i32:
24; NOSDWA: v_lshrrev_b32_e32 v[[DST:[0-9]+]], 16, v{{[0-9]+}}
Dmitry Preobrazhenskya0342dc2017-11-20 18:24:21 +000025; NOSDWA: v_subrev_u32_e32 v{{[0-9]+}}, vcc, v{{[0-9]+}}, v[[DST]]
Matt Arsenault9a7e29a2017-11-29 02:25:14 +000026; NOSDWA-NOT: v_subrev_{{(_co)?}}_u32_sdwa
Sam Koltonf60ad582017-03-21 12:51:34 +000027
Matt Arsenault84445dd2017-11-30 22:51:26 +000028; VI: v_subrev_u32_sdwa v{{[0-9]+}}, vcc, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:WORD_1
29; GFX9: v_sub_u32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +000030; GFX10: v_sub_nc_u32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
Matt Arsenault5660bb62019-11-18 16:48:07 +053031define amdgpu_kernel void @sub_shr_i32(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +000032 %a = load i32, i32 addrspace(1)* %in, align 4
33 %shr = lshr i32 %a, 16
34 %sub = sub i32 %shr, %a
35 store i32 %sub, i32 addrspace(1)* %out, align 4
36 ret void
37}
38
39; GCN-LABEL: {{^}}mul_shr_i32:
40; NOSDWA: v_lshrrev_b32_e32 v[[DST0:[0-9]+]], 16, v{{[0-9]+}}
41; NOSDWA: v_lshrrev_b32_e32 v[[DST1:[0-9]+]], 16, v{{[0-9]+}}
Matt Arsenault6c29c5a2017-07-10 19:53:57 +000042; NOSDWA: v_mul_u32_u24_e32 v{{[0-9]+}}, v[[DST0]], v[[DST1]]
Sam Koltonf60ad582017-03-21 12:51:34 +000043; NOSDWA-NOT: v_mul_u32_u24_sdwa
44
45; SDWA: v_mul_u32_u24_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
46
Matt Arsenault5660bb62019-11-18 16:48:07 +053047define amdgpu_kernel void @mul_shr_i32(i32 addrspace(1)* %out, i32 addrspace(1)* %in1, i32 addrspace(1)* %in2) #0 {
Jay Foadfdaa2d02021-02-19 15:04:03 +000048 %idx = call i32 @llvm.amdgcn.workitem.id.x()
49 %gep1 = getelementptr i32, i32 addrspace(1)* %in1, i32 %idx
50 %gep2 = getelementptr i32, i32 addrspace(1)* %in2, i32 %idx
51 %a = load i32, i32 addrspace(1)* %gep1, align 4
52 %b = load i32, i32 addrspace(1)* %gep2, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +000053 %shra = lshr i32 %a, 16
54 %shrb = lshr i32 %b, 16
55 %mul = mul i32 %shra, %shrb
56 store i32 %mul, i32 addrspace(1)* %out, align 4
57 ret void
58}
59
60; GCN-LABEL: {{^}}mul_i16:
Jay Foadfdaa2d02021-02-19 15:04:03 +000061; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +000062; NOSDWA-NOT: v_mul_u32_u24_sdwa
Jay Foadfdaa2d02021-02-19 15:04:03 +000063; GFX89: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Dmitry Preobrazhenskycd953432021-04-01 14:21:00 +030064; GFX10: v_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +000065; SDWA-NOT: v_mul_u32_u24_sdwa
66
Matt Arsenault5660bb62019-11-18 16:48:07 +053067define amdgpu_kernel void @mul_i16(i16 addrspace(1)* %out, i16 addrspace(1)* %ina, i16 addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +000068entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +000069 %idx = call i32 @llvm.amdgcn.workitem.id.x()
70 %gepa = getelementptr i16, i16 addrspace(1)* %ina, i32 %idx
71 %gepb = getelementptr i16, i16 addrspace(1)* %inb, i32 %idx
72 %a = load i16, i16 addrspace(1)* %gepa, align 4
73 %b = load i16, i16 addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +000074 %mul = mul i16 %a, %b
75 store i16 %mul, i16 addrspace(1)* %out, align 4
76 ret void
77}
78
79; GCN-LABEL: {{^}}mul_v2i16:
80; NOSDWA: v_lshrrev_b32_e32 v[[DST0:[0-9]+]], 16, v{{[0-9]+}}
81; NOSDWA: v_lshrrev_b32_e32 v[[DST1:[0-9]+]], 16, v{{[0-9]+}}
Jay Foadfdaa2d02021-02-19 15:04:03 +000082; NOSDWA: v_mul_lo_u16_e32 v[[DST_MUL:[0-9]+]], v[[DST1]], v[[DST0]]
Sam Koltonf60ad582017-03-21 12:51:34 +000083; NOSDWA: v_lshlrev_b32_e32 v[[DST_SHL:[0-9]+]], 16, v[[DST_MUL]]
Matt Arsenault6c29c5a2017-07-10 19:53:57 +000084; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v[[DST_SHL]]
Sam Koltonf60ad582017-03-21 12:51:34 +000085; NOSDWA-NOT: v_mul_u32_u24_sdwa
86
Jay Foadfdaa2d02021-02-19 15:04:03 +000087; VI-DAG: v_mul_lo_u16_e32 v[[DST_MUL_LO:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
88; VI-DAG: v_mul_lo_u16_sdwa v[[DST_MUL_HI:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
89; VI: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL_LO]], v[[DST_MUL_HI]]
Sam Kolton3c4933f2017-06-22 06:26:41 +000090
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +000091; GFX9_10: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +000092
Matt Arsenault5660bb62019-11-18 16:48:07 +053093define amdgpu_kernel void @mul_v2i16(<2 x i16> addrspace(1)* %out, <2 x i16> addrspace(1)* %ina, <2 x i16> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +000094entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +000095 %idx = call i32 @llvm.amdgcn.workitem.id.x()
96 %gepa = getelementptr <2 x i16>, <2 x i16> addrspace(1)* %ina, i32 %idx
97 %gepb = getelementptr <2 x i16>, <2 x i16> addrspace(1)* %inb, i32 %idx
98 %a = load <2 x i16>, <2 x i16> addrspace(1)* %gepa, align 4
99 %b = load <2 x i16>, <2 x i16> addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000100 %mul = mul <2 x i16> %a, %b
101 store <2 x i16> %mul, <2 x i16> addrspace(1)* %out, align 4
102 ret void
103}
104
105; GCN-LABEL: {{^}}mul_v4i16:
106; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
107; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
Jay Foadfdaa2d02021-02-19 15:04:03 +0000108; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000109; NOSDWA: v_lshlrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
110; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
111; NOSDWA-NOT: v_mul_u32_u24_sdwa
112
Jay Foadfdaa2d02021-02-19 15:04:03 +0000113; VI-DAG: v_mul_lo_u16_e32 v[[DST_MUL0:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
114; VI-DAG: v_mul_lo_u16_sdwa v[[DST_MUL1:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
115; VI-DAG: v_mul_lo_u16_e32 v[[DST_MUL2:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
116; VI-DAG: v_mul_lo_u16_sdwa v[[DST_MUL3:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
117; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL2]], v[[DST_MUL3]]
118; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL0]], v[[DST_MUL1]]
Sam Kolton3c4933f2017-06-22 06:26:41 +0000119
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000120; GFX9_10-DAG: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
121; GFX9_10-DAG: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000122
Matt Arsenault5660bb62019-11-18 16:48:07 +0530123define amdgpu_kernel void @mul_v4i16(<4 x i16> addrspace(1)* %out, <4 x i16> addrspace(1)* %ina, <4 x i16> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000124entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000125 %idx = call i32 @llvm.amdgcn.workitem.id.x()
126 %gepa = getelementptr <4 x i16>, <4 x i16> addrspace(1)* %ina, i32 %idx
127 %gepb = getelementptr <4 x i16>, <4 x i16> addrspace(1)* %inb, i32 %idx
128 %a = load <4 x i16>, <4 x i16> addrspace(1)* %gepa, align 4
129 %b = load <4 x i16>, <4 x i16> addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000130 %mul = mul <4 x i16> %a, %b
131 store <4 x i16> %mul, <4 x i16> addrspace(1)* %out, align 4
132 ret void
133}
134
135; GCN-LABEL: {{^}}mul_v8i16:
136; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
137; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
Jay Foadfdaa2d02021-02-19 15:04:03 +0000138; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000139; NOSDWA: v_lshlrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
140; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
141; NOSDWA-NOT: v_mul_u32_u24_sdwa
142
Jay Foadfdaa2d02021-02-19 15:04:03 +0000143; VI-DAG: v_mul_lo_u16_e32 v[[DST_MUL0:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
144; VI-DAG: v_mul_lo_u16_sdwa v[[DST_MUL1:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
145; VI-DAG: v_mul_lo_u16_e32 v[[DST_MUL2:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
146; VI-DAG: v_mul_lo_u16_sdwa v[[DST_MUL3:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
147; VI-DAG: v_mul_lo_u16_e32 v[[DST_MUL4:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
148; VI-DAG: v_mul_lo_u16_sdwa v[[DST_MUL5:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
149; VI-DAG: v_mul_lo_u16_e32 v[[DST_MUL6:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
150; VI-DAG: v_mul_lo_u16_sdwa v[[DST_MUL7:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
151; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL6]], v[[DST_MUL7]]
152; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL4]], v[[DST_MUL5]]
153; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL2]], v[[DST_MUL3]]
154; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL0]], v[[DST_MUL1]]
Sam Kolton3c4933f2017-06-22 06:26:41 +0000155
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000156; GFX9_10-DAG: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
157; GFX9_10-DAG: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
158; GFX9_10-DAG: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
159; GFX9_10-DAG: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000160
Matt Arsenault5660bb62019-11-18 16:48:07 +0530161define amdgpu_kernel void @mul_v8i16(<8 x i16> addrspace(1)* %out, <8 x i16> addrspace(1)* %ina, <8 x i16> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000162entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000163 %idx = call i32 @llvm.amdgcn.workitem.id.x()
164 %gepa = getelementptr <8 x i16>, <8 x i16> addrspace(1)* %ina, i32 %idx
165 %gepb = getelementptr <8 x i16>, <8 x i16> addrspace(1)* %inb, i32 %idx
166 %a = load <8 x i16>, <8 x i16> addrspace(1)* %gepa, align 4
167 %b = load <8 x i16>, <8 x i16> addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000168 %mul = mul <8 x i16> %a, %b
169 store <8 x i16> %mul, <8 x i16> addrspace(1)* %out, align 4
170 ret void
171}
172
173; GCN-LABEL: {{^}}mul_half:
174; NOSDWA: v_mul_f16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
175; NOSDWA-NOT: v_mul_f16_sdwa
176; SDWA: v_mul_f16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
177; SDWA-NOT: v_mul_f16_sdwa
178
Matt Arsenault5660bb62019-11-18 16:48:07 +0530179define amdgpu_kernel void @mul_half(half addrspace(1)* %out, half addrspace(1)* %ina, half addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000180entry:
181 %a = load half, half addrspace(1)* %ina, align 4
182 %b = load half, half addrspace(1)* %inb, align 4
183 %mul = fmul half %a, %b
184 store half %mul, half addrspace(1)* %out, align 4
185 ret void
186}
187
188; GCN-LABEL: {{^}}mul_v2half:
189; NOSDWA: v_lshrrev_b32_e32 v[[DST0:[0-9]+]], 16, v{{[0-9]+}}
190; NOSDWA: v_lshrrev_b32_e32 v[[DST1:[0-9]+]], 16, v{{[0-9]+}}
Matt Arsenault6c29c5a2017-07-10 19:53:57 +0000191; NOSDWA: v_mul_f16_e32 v[[DST_MUL:[0-9]+]], v[[DST0]], v[[DST1]]
Sam Koltonf60ad582017-03-21 12:51:34 +0000192; NOSDWA: v_lshlrev_b32_e32 v[[DST_SHL:[0-9]+]], 16, v[[DST_MUL]]
Matt Arsenault6c29c5a2017-07-10 19:53:57 +0000193; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v[[DST_SHL]]
Sam Koltonf60ad582017-03-21 12:51:34 +0000194; NOSDWA-NOT: v_mul_f16_sdwa
195
Sam Kolton3c4933f2017-06-22 06:26:41 +0000196; VI-DAG: v_mul_f16_sdwa v[[DST_MUL_HI:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
197; VI-DAG: v_mul_f16_e32 v[[DST_MUL_LO:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
Matt Arsenault6c29c5a2017-07-10 19:53:57 +0000198; VI: v_or_b32_e32 v{{[0-9]+}}, v[[DST_MUL_LO]], v[[DST_MUL_HI]]
Sam Kolton3c4933f2017-06-22 06:26:41 +0000199
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000200; GFX9_10: v_pk_mul_f16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Kolton3c4933f2017-06-22 06:26:41 +0000201
Matt Arsenault5660bb62019-11-18 16:48:07 +0530202define amdgpu_kernel void @mul_v2half(<2 x half> addrspace(1)* %out, <2 x half> addrspace(1)* %ina, <2 x half> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000203entry:
204 %a = load <2 x half>, <2 x half> addrspace(1)* %ina, align 4
205 %b = load <2 x half>, <2 x half> addrspace(1)* %inb, align 4
206 %mul = fmul <2 x half> %a, %b
207 store <2 x half> %mul, <2 x half> addrspace(1)* %out, align 4
208 ret void
209}
210
211; GCN-LABEL: {{^}}mul_v4half:
212; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
213; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
214; NOSDWA: v_mul_f16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
215; NOSDWA: v_lshlrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
216; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
217; NOSDWA-NOT: v_mul_f16_sdwa
218
Sam Kolton3c4933f2017-06-22 06:26:41 +0000219; VI-DAG: v_mul_f16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
220; VI-DAG: v_mul_f16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
221; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
222; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
223
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000224; GFX9_10-DAG: v_pk_mul_f16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
225; GFX9_10-DAG: v_pk_mul_f16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000226
Matt Arsenault5660bb62019-11-18 16:48:07 +0530227define amdgpu_kernel void @mul_v4half(<4 x half> addrspace(1)* %out, <4 x half> addrspace(1)* %ina, <4 x half> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000228entry:
229 %a = load <4 x half>, <4 x half> addrspace(1)* %ina, align 4
230 %b = load <4 x half>, <4 x half> addrspace(1)* %inb, align 4
231 %mul = fmul <4 x half> %a, %b
232 store <4 x half> %mul, <4 x half> addrspace(1)* %out, align 4
233 ret void
234}
235
236; GCN-LABEL: {{^}}mul_v8half:
237; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
238; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
239; NOSDWA: v_mul_f16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
240; NOSDWA: v_lshlrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
241; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
242; NOSDWA-NOT: v_mul_f16_sdwa
243
Sam Kolton3c4933f2017-06-22 06:26:41 +0000244; VI-DAG: v_mul_f16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
245; VI-DAG: v_mul_f16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
246; VI-DAG: v_mul_f16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
247; VI-DAG: v_mul_f16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
248; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
249; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
250; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
251; VI-DAG: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
252
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000253; GFX9_10-DAG: v_pk_mul_f16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
254; GFX9_10-DAG: v_pk_mul_f16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
255; GFX9_10-DAG: v_pk_mul_f16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
256; GFX9_10-DAG: v_pk_mul_f16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000257
Matt Arsenault5660bb62019-11-18 16:48:07 +0530258define amdgpu_kernel void @mul_v8half(<8 x half> addrspace(1)* %out, <8 x half> addrspace(1)* %ina, <8 x half> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000259entry:
260 %a = load <8 x half>, <8 x half> addrspace(1)* %ina, align 4
261 %b = load <8 x half>, <8 x half> addrspace(1)* %inb, align 4
262 %mul = fmul <8 x half> %a, %b
263 store <8 x half> %mul, <8 x half> addrspace(1)* %out, align 4
264 ret void
265}
266
267; GCN-LABEL: {{^}}mul_i8:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000268; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000269; NOSDWA-NOT: v_mul_u32_u24_sdwa
Jay Foadfdaa2d02021-02-19 15:04:03 +0000270; GFX89: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Dmitry Preobrazhenskycd953432021-04-01 14:21:00 +0300271; GFX10: v_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000272; SDWA-NOT: v_mul_u32_u24_sdwa
273
Matt Arsenault5660bb62019-11-18 16:48:07 +0530274define amdgpu_kernel void @mul_i8(i8 addrspace(1)* %out, i8 addrspace(1)* %ina, i8 addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000275entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000276 %idx = call i32 @llvm.amdgcn.workitem.id.x()
277 %gepa = getelementptr i8, i8 addrspace(1)* %ina, i32 %idx
278 %gepb = getelementptr i8, i8 addrspace(1)* %inb, i32 %idx
279 %a = load i8, i8 addrspace(1)* %gepa, align 4
280 %b = load i8, i8 addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000281 %mul = mul i8 %a, %b
282 store i8 %mul, i8 addrspace(1)* %out, align 4
283 ret void
284}
285
286; GCN-LABEL: {{^}}mul_v2i8:
287; NOSDWA: v_lshrrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
288; NOSDWA: v_lshrrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
Jay Foadfdaa2d02021-02-19 15:04:03 +0000289; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000290; NOSDWA: v_lshlrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
291; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
292; NOSDWA-NOT: v_mul_u32_u24_sdwa
293
Jay Foadfdaa2d02021-02-19 15:04:03 +0000294; VI: v_mul_lo_u16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:BYTE_1 dst_unused:UNUSED_PAD src0_sel:BYTE_1 src1_sel:BYTE_1
Sam Kolton3c4933f2017-06-22 06:26:41 +0000295
296; GFX9-DAG: v_mul_lo_u16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:BYTE_1 dst_unused:UNUSED_PAD src0_sel:BYTE_1 src1_sel:BYTE_1
297; GFX9-DAG: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000298
Dmitry Preobrazhenskycd953432021-04-01 14:21:00 +0300299; GFX10-DAG: v_mul_lo_u16
300; GFX10-DAG: v_mul_lo_u16
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000301
Sam Kolton3c4933f2017-06-22 06:26:41 +0000302; GFX9: v_or_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:BYTE_0 src1_sel:DWORD
Sam Koltonf60ad582017-03-21 12:51:34 +0000303
Dmitry Preobrazhenskycd953432021-04-01 14:21:00 +0300304; GFX10: v_lshlrev_b16 v{{[0-9]+}}, 8, v
Matt Arsenault190a17b2019-10-08 17:36:38 +0000305; GFX10: v_or_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:BYTE_0 src1_sel:DWORD
Matt Arsenault5660bb62019-11-18 16:48:07 +0530306define amdgpu_kernel void @mul_v2i8(<2 x i8> addrspace(1)* %out, <2 x i8> addrspace(1)* %ina, <2 x i8> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000307entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000308 %idx = call i32 @llvm.amdgcn.workitem.id.x()
309 %gepa = getelementptr <2 x i8>, <2 x i8> addrspace(1)* %ina, i32 %idx
310 %gepb = getelementptr <2 x i8>, <2 x i8> addrspace(1)* %inb, i32 %idx
311 %a = load <2 x i8>, <2 x i8> addrspace(1)* %gepa, align 4
312 %b = load <2 x i8>, <2 x i8> addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000313 %mul = mul <2 x i8> %a, %b
314 store <2 x i8> %mul, <2 x i8> addrspace(1)* %out, align 4
315 ret void
316}
317
318; GCN-LABEL: {{^}}mul_v4i8:
319; NOSDWA: v_lshrrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
320; NOSDWA: v_lshrrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
Jay Foadfdaa2d02021-02-19 15:04:03 +0000321; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000322; NOSDWA: v_lshlrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
323; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
324; NOSDWA-NOT: v_mul_u32_u24_sdwa
325
Jay Foadfdaa2d02021-02-19 15:04:03 +0000326; VI-DAG: v_mul_lo_u16_sdwa
327; VI-DAG: v_mul_lo_u16_sdwa
328; VI-DAG: v_mul_lo_u16_sdwa
Sam Kolton3c4933f2017-06-22 06:26:41 +0000329
330; GFX9-DAG: v_mul_lo_u16_sdwa
331; GFX9-DAG: v_mul_lo_u16_sdwa
332; GFX9-DAG: v_mul_lo_u16_sdwa
Sam Koltonf60ad582017-03-21 12:51:34 +0000333
Dmitry Preobrazhenskycd953432021-04-01 14:21:00 +0300334; GFX10-DAG: v_mul_lo_u16
335; GFX10-DAG: v_mul_lo_u16
336; GFX10-DAG: v_mul_lo_u16
337; GFX10-DAG: v_mul_lo_u16
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000338
Matt Arsenault5660bb62019-11-18 16:48:07 +0530339define amdgpu_kernel void @mul_v4i8(<4 x i8> addrspace(1)* %out, <4 x i8> addrspace(1)* %ina, <4 x i8> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000340entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000341 %idx = call i32 @llvm.amdgcn.workitem.id.x()
342 %gepa = getelementptr <4 x i8>, <4 x i8> addrspace(1)* %ina, i32 %idx
343 %gepb = getelementptr <4 x i8>, <4 x i8> addrspace(1)* %inb, i32 %idx
344 %a = load <4 x i8>, <4 x i8> addrspace(1)* %gepa, align 4
345 %b = load <4 x i8>, <4 x i8> addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000346 %mul = mul <4 x i8> %a, %b
347 store <4 x i8> %mul, <4 x i8> addrspace(1)* %out, align 4
348 ret void
349}
350
351; GCN-LABEL: {{^}}mul_v8i8:
352; NOSDWA: v_lshrrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
353; NOSDWA: v_lshrrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
Jay Foadfdaa2d02021-02-19 15:04:03 +0000354; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000355; NOSDWA: v_lshlrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
356; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
357; NOSDWA-NOT: v_mul_u32_u24_sdwa
358
Jay Foadfdaa2d02021-02-19 15:04:03 +0000359; VI-DAG: v_mul_lo_u16_sdwa
360; VI-DAG: v_mul_lo_u16_sdwa
361; VI-DAG: v_mul_lo_u16_sdwa
362; VI-DAG: v_mul_lo_u16_sdwa
363; VI-DAG: v_mul_lo_u16_sdwa
364; VI-DAG: v_mul_lo_u16_sdwa
Sam Kolton3c4933f2017-06-22 06:26:41 +0000365
366; GFX9-DAG: v_mul_lo_u16_sdwa
367; GFX9-DAG: v_mul_lo_u16_sdwa
368; GFX9-DAG: v_mul_lo_u16_sdwa
369; GFX9-DAG: v_mul_lo_u16_sdwa
370; GFX9-DAG: v_mul_lo_u16_sdwa
371; GFX9-DAG: v_mul_lo_u16_sdwa
Sam Koltonf60ad582017-03-21 12:51:34 +0000372
Dmitry Preobrazhenskycd953432021-04-01 14:21:00 +0300373; GFX10-DAG: v_mul_lo_u16
374; GFX10-DAG: v_mul_lo_u16
375; GFX10-DAG: v_mul_lo_u16
376; GFX10-DAG: v_mul_lo_u16
377; GFX10-DAG: v_mul_lo_u16
378; GFX10-DAG: v_mul_lo_u16
379; GFX10-DAG: v_mul_lo_u16
380; GFX10-DAG: v_mul_lo_u16
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000381
Matt Arsenault5660bb62019-11-18 16:48:07 +0530382define amdgpu_kernel void @mul_v8i8(<8 x i8> addrspace(1)* %out, <8 x i8> addrspace(1)* %ina, <8 x i8> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000383entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000384 %idx = call i32 @llvm.amdgcn.workitem.id.x()
385 %gepa = getelementptr <8 x i8>, <8 x i8> addrspace(1)* %ina, i32 %idx
386 %gepb = getelementptr <8 x i8>, <8 x i8> addrspace(1)* %inb, i32 %idx
387 %a = load <8 x i8>, <8 x i8> addrspace(1)* %gepa, align 4
388 %b = load <8 x i8>, <8 x i8> addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000389 %mul = mul <8 x i8> %a, %b
390 store <8 x i8> %mul, <8 x i8> addrspace(1)* %out, align 4
391 ret void
392}
393
Sam Kolton9fa16962017-04-06 15:03:28 +0000394; GCN-LABEL: {{^}}sitofp_v2i16_to_v2f16:
Matt Arsenault68e70fb2019-09-30 13:39:33 -0400395; NOSDWA-DAG: v_cvt_f16_i16_e32 v{{[0-9]+}}, v{{[0-9]+}}
396; NOSDWA-DAG: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
397; NOSDWA-DAG: v_cvt_f16_i16_e32 v{{[0-9]+}}, v{{[0-9]+}}
398; NOSDWA-NOT: v_cvt_f16_i16_sdwa
Sam Kolton9fa16962017-04-06 15:03:28 +0000399
Matt Arsenault68e70fb2019-09-30 13:39:33 -0400400; SDWA-DAG: v_cvt_f16_i16_e32 v{{[0-9]+}}, v{{[0-9]+}}
401; SDWA-DAG: v_cvt_f16_i16_sdwa v{{[0-9]+}}, v{{[0-9]+}} dst_sel:{{(WORD_1|DWORD)?}} dst_unused:UNUSED_PAD src0_sel:WORD_1
Sam Kolton9fa16962017-04-06 15:03:28 +0000402
Matt Arsenault68e70fb2019-09-30 13:39:33 -0400403; FIXME: Should be able to avoid or
Sam Kolton9fa16962017-04-06 15:03:28 +0000404define amdgpu_kernel void @sitofp_v2i16_to_v2f16(
405 <2 x half> addrspace(1)* %r,
Matt Arsenault5660bb62019-11-18 16:48:07 +0530406 <2 x i16> addrspace(1)* %a) #0 {
Sam Kolton9fa16962017-04-06 15:03:28 +0000407entry:
408 %a.val = load <2 x i16>, <2 x i16> addrspace(1)* %a
409 %r.val = sitofp <2 x i16> %a.val to <2 x half>
410 store <2 x half> %r.val, <2 x half> addrspace(1)* %r
411 ret void
412}
413
Sam Koltonf60ad582017-03-21 12:51:34 +0000414
415; GCN-LABEL: {{^}}mac_v2half:
416; NOSDWA: v_lshrrev_b32_e32 v[[DST0:[0-9]+]], 16, v{{[0-9]+}}
417; NOSDWA: v_lshrrev_b32_e32 v[[DST1:[0-9]+]], 16, v{{[0-9]+}}
Matt Arsenault6c29c5a2017-07-10 19:53:57 +0000418; NOSDWA: v_mac_f16_e32 v[[DST_MAC:[0-9]+]], v[[DST0]], v[[DST1]]
Sam Koltonf60ad582017-03-21 12:51:34 +0000419; NOSDWA: v_lshlrev_b32_e32 v[[DST_SHL:[0-9]+]], 16, v[[DST_MAC]]
Matt Arsenault6c29c5a2017-07-10 19:53:57 +0000420; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v[[DST_SHL]]
Sam Koltonf60ad582017-03-21 12:51:34 +0000421; NOSDWA-NOT: v_mac_f16_sdwa
422
Sam Kolton3c4933f2017-06-22 06:26:41 +0000423; VI: v_mac_f16_sdwa v[[DST_MAC:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
424; VI: v_lshlrev_b32_e32 v[[DST_SHL:[0-9]+]], 16, v[[DST_MAC]]
425
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000426; GFX9_10: v_pk_mul_f16 v[[DST_MUL:[0-9]+]], v{{[0-9]+}}, v[[SRC:[0-9]+]]
427; GFX9_10: v_pk_add_f16 v{{[0-9]+}}, v[[DST_MUL]], v[[SRC]]
Sam Koltonf60ad582017-03-21 12:51:34 +0000428
Matt Arsenault5660bb62019-11-18 16:48:07 +0530429define amdgpu_kernel void @mac_v2half(<2 x half> addrspace(1)* %out, <2 x half> addrspace(1)* %ina, <2 x half> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000430entry:
431 %a = load <2 x half>, <2 x half> addrspace(1)* %ina, align 4
432 %b = load <2 x half>, <2 x half> addrspace(1)* %inb, align 4
433 %mul = fmul <2 x half> %a, %b
434 %mac = fadd <2 x half> %mul, %b
435 store <2 x half> %mac, <2 x half> addrspace(1)* %out, align 4
436 ret void
437}
438
439; GCN-LABEL: {{^}}immediate_mul_v2i16:
440; NOSDWA-NOT: v_mul_u32_u24_sdwa
Sam Kolton3c4933f2017-06-22 06:26:41 +0000441; VI-DAG: v_mov_b32_e32 v[[M321:[0-9]+]], 0x141
Jay Foadfdaa2d02021-02-19 15:04:03 +0000442; VI-DAG: v_mul_lo_u16_e32 v{{[0-9]+}}, 0x7b, v{{[0-9]+}}
443; VI-DAG: v_mul_lo_u16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v[[M321]] dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
Sam Kolton3c4933f2017-06-22 06:26:41 +0000444
445; GFX9: s_mov_b32 s[[IMM:[0-9]+]], 0x141007b
446; GFX9: v_pk_mul_lo_u16 v{{[0-9]+}}, v{{[0-9]+}}, s[[IMM]]
Sam Koltonf60ad582017-03-21 12:51:34 +0000447
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000448; GFX10: v_pk_mul_lo_u16 v{{[0-9]+}}, 0x141007b, v{{[0-9]+}}
449
Matt Arsenault5660bb62019-11-18 16:48:07 +0530450define amdgpu_kernel void @immediate_mul_v2i16(<2 x i16> addrspace(1)* %out, <2 x i16> addrspace(1)* %in) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000451entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000452 %idx = call i32 @llvm.amdgcn.workitem.id.x()
453 %gep = getelementptr <2 x i16>, <2 x i16> addrspace(1)* %in, i32 %idx
454 %a = load <2 x i16>, <2 x i16> addrspace(1)* %gep, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000455 %mul = mul <2 x i16> %a, <i16 123, i16 321>
456 store <2 x i16> %mul, <2 x i16> addrspace(1)* %out, align 4
457 ret void
458}
459
460; Double use of same src - should not convert it
461; GCN-LABEL: {{^}}mulmul_v2i16:
462; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
463; NOSDWA: v_lshrrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
Jay Foadfdaa2d02021-02-19 15:04:03 +0000464; NOSDWA: v_mul_lo_u16_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000465; NOSDWA: v_lshlrev_b32_e32 v{{[0-9]+}}, 16, v{{[0-9]+}}
466; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
467; NOSDWA-NOT: v_mul_u32_u24_sdwa
468
Jay Foadfdaa2d02021-02-19 15:04:03 +0000469; VI: v_mul_lo_u16_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
Sam Kolton3c4933f2017-06-22 06:26:41 +0000470
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000471; GFX9_10: v_pk_mul_lo_u16 v[[DST1:[0-9]+]], v{{[0-9]+}}, v{{[0-9]+}}
472; GFX9_10: v_pk_mul_lo_u16 v{{[0-9]+}}, v[[DST1]], v{{[0-9]+}}
Sam Koltonf60ad582017-03-21 12:51:34 +0000473
Matt Arsenault5660bb62019-11-18 16:48:07 +0530474define amdgpu_kernel void @mulmul_v2i16(<2 x i16> addrspace(1)* %out, <2 x i16> addrspace(1)* %ina, <2 x i16> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000475entry:
Jay Foadfdaa2d02021-02-19 15:04:03 +0000476 %idx = call i32 @llvm.amdgcn.workitem.id.x()
477 %gepa = getelementptr <2 x i16>, <2 x i16> addrspace(1)* %ina, i32 %idx
478 %gepb = getelementptr <2 x i16>, <2 x i16> addrspace(1)* %inb, i32 %idx
479 %a = load <2 x i16>, <2 x i16> addrspace(1)* %gepa, align 4
480 %b = load <2 x i16>, <2 x i16> addrspace(1)* %gepb, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000481 %mul = mul <2 x i16> %a, %b
482 %mul2 = mul <2 x i16> %mul, %b
483 store <2 x i16> %mul2, <2 x i16> addrspace(1)* %out, align 4
484 ret void
485}
486
Sam Koltonaff83412017-04-12 09:36:05 +0000487; GCN-LABEL: {{^}}add_bb_v2i16:
Matt Arsenault9a7e29a2017-11-29 02:25:14 +0000488; NOSDWA-NOT: v_add_{{(_co)?}}_u32_sdwa
Sam Koltonf60ad582017-03-21 12:51:34 +0000489
Dmitry Preobrazhenskya0342dc2017-11-20 18:24:21 +0000490; VI: v_add_u32_sdwa v{{[0-9]+}}, vcc, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:WORD_1
Sam Kolton3c4933f2017-06-22 06:26:41 +0000491
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000492; GFX9_10: v_pk_add_u16 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonaff83412017-04-12 09:36:05 +0000493
Matt Arsenault5660bb62019-11-18 16:48:07 +0530494define amdgpu_kernel void @add_bb_v2i16(<2 x i16> addrspace(1)* %out, <2 x i16> addrspace(1)* %ina, <2 x i16> addrspace(1)* %inb) #0 {
Sam Koltonf60ad582017-03-21 12:51:34 +0000495entry:
Sam Koltonaff83412017-04-12 09:36:05 +0000496 %a = load <2 x i16>, <2 x i16> addrspace(1)* %ina, align 4
497 %b = load <2 x i16>, <2 x i16> addrspace(1)* %inb, align 4
498 br label %add_label
Sam Koltonf60ad582017-03-21 12:51:34 +0000499add_label:
Sam Koltonaff83412017-04-12 09:36:05 +0000500 %add = add <2 x i16> %a, %b
Sam Koltonf60ad582017-03-21 12:51:34 +0000501 br label %store_label
502store_label:
Sam Koltonaff83412017-04-12 09:36:05 +0000503 store <2 x i16> %add, <2 x i16> addrspace(1)* %out, align 4
Sam Koltonf60ad582017-03-21 12:51:34 +0000504 ret void
Sam Koltonaff83412017-04-12 09:36:05 +0000505}
Sam Koltonebfdaf72017-05-18 12:12:03 +0000506
507
508; Check that "pulling out" SDWA operands works correctly.
509; GCN-LABEL: {{^}}pulled_out_test:
Stanislav Mekhanoshin465a1ff2017-06-20 18:32:42 +0000510; NOSDWA-DAG: v_and_b32_e32 v{{[0-9]+}}, s{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonebfdaf72017-05-18 12:12:03 +0000511; NOSDWA-DAG: v_lshlrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
Stanislav Mekhanoshin465a1ff2017-06-20 18:32:42 +0000512; NOSDWA-DAG: v_and_b32_e32 v{{[0-9]+}}, s{{[0-9]+}}, v{{[0-9]+}}
Sam Koltonebfdaf72017-05-18 12:12:03 +0000513; NOSDWA-DAG: v_lshlrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
514; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}}
515; NOSDWA-NOT: v_and_b32_sdwa
516; NOSDWA-NOT: v_or_b32_sdwa
517
Sam Kolton3c4933f2017-06-22 06:26:41 +0000518; VI-DAG: v_and_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000519; GFX9_10-DAG: v_and_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, s{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
520; GFX89-DAG: v_lshlrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
521;
522; GFX10-DAG: v_lshrrev_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:BYTE_1 dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:DWORD
523;
Sam Kolton3c4933f2017-06-22 06:26:41 +0000524; VI-DAG: v_and_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
Stanislav Mekhanoshin971cb8b2019-05-06 22:27:05 +0000525; GFX9_10-DAG: v_and_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, s{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:WORD_1 src1_sel:DWORD
526; GFX89-DAG: v_lshlrev_b16_e32 v{{[0-9]+}}, 8, v{{[0-9]+}}
527;
528; GFX10-DAG: v_lshrrev_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:BYTE_1 dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:DWORD
529;
530; GFX89: v_or_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:DWORD
531;
Matt Arsenault190a17b2019-10-08 17:36:38 +0000532; GFX10: v_or_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:BYTE_0 src1_sel:DWORD
533; GFX10: v_or_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:DWORD dst_unused:UNUSED_PAD src0_sel:BYTE_0 src1_sel:DWORD
534; GFX10: v_or_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:DWORD
535; GFX10: v_or_b32_sdwa v{{[0-9]+}}, v{{[0-9]+}}, v{{[0-9]+}} dst_sel:WORD_1 dst_unused:UNUSED_PAD src0_sel:DWORD src1_sel:DWORD
Sam Koltonebfdaf72017-05-18 12:12:03 +0000536
Matt Arsenault5660bb62019-11-18 16:48:07 +0530537define amdgpu_kernel void @pulled_out_test(<8 x i8> addrspace(1)* %sourceA, <8 x i8> addrspace(1)* %destValues) #0 {
Sam Koltonebfdaf72017-05-18 12:12:03 +0000538entry:
539 %idxprom = ashr exact i64 15, 32
540 %arrayidx = getelementptr inbounds <8 x i8>, <8 x i8> addrspace(1)* %sourceA, i64 %idxprom
541 %tmp = load <8 x i8>, <8 x i8> addrspace(1)* %arrayidx, align 8
542
543 %tmp1 = extractelement <8 x i8> %tmp, i32 0
544 %tmp2 = extractelement <8 x i8> %tmp, i32 1
545 %tmp3 = extractelement <8 x i8> %tmp, i32 2
546 %tmp4 = extractelement <8 x i8> %tmp, i32 3
547 %tmp5 = extractelement <8 x i8> %tmp, i32 4
548 %tmp6 = extractelement <8 x i8> %tmp, i32 5
549 %tmp7 = extractelement <8 x i8> %tmp, i32 6
550 %tmp8 = extractelement <8 x i8> %tmp, i32 7
551
552 %tmp9 = insertelement <2 x i8> undef, i8 %tmp1, i32 0
553 %tmp10 = insertelement <2 x i8> %tmp9, i8 %tmp2, i32 1
554 %tmp11 = insertelement <2 x i8> undef, i8 %tmp3, i32 0
555 %tmp12 = insertelement <2 x i8> %tmp11, i8 %tmp4, i32 1
556 %tmp13 = insertelement <2 x i8> undef, i8 %tmp5, i32 0
557 %tmp14 = insertelement <2 x i8> %tmp13, i8 %tmp6, i32 1
558 %tmp15 = insertelement <2 x i8> undef, i8 %tmp7, i32 0
559 %tmp16 = insertelement <2 x i8> %tmp15, i8 %tmp8, i32 1
560
561 %tmp17 = shufflevector <2 x i8> %tmp10, <2 x i8> %tmp12, <4 x i32> <i32 0, i32 1, i32 2, i32 3>
562 %tmp18 = shufflevector <2 x i8> %tmp14, <2 x i8> %tmp16, <4 x i32> <i32 0, i32 1, i32 2, i32 3>
563 %tmp19 = shufflevector <4 x i8> %tmp17, <4 x i8> %tmp18, <8 x i32> <i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>
Matt Arsenault6c29c5a2017-07-10 19:53:57 +0000564
Sam Koltonebfdaf72017-05-18 12:12:03 +0000565 %arrayidx5 = getelementptr inbounds <8 x i8>, <8 x i8> addrspace(1)* %destValues, i64 %idxprom
566 store <8 x i8> %tmp19, <8 x i8> addrspace(1)* %arrayidx5, align 8
567 ret void
568}
Matt Arsenault8ae38bc2017-12-05 20:32:01 +0000569
Matt Arsenaultc24d5e22018-02-08 22:46:38 +0000570; GCN-LABEL: {{^}}sdwa_crash_inlineasm_def:
Matt Arsenault8ae38bc2017-12-05 20:32:01 +0000571; GCN: s_mov_b32 s{{[0-9]+}}, 0xffff
572; GCN: v_and_b32_e32 v{{[0-9]+}}, s{{[0-9]+}}, v{{[0-9]+}}
Nicolai Haehnlea9cc92c2018-11-30 22:55:29 +0000573;
574; TODO: Why is the constant not peepholed into the v_or_b32_e32?
575;
Jay Foad8a52bd82021-11-19 16:40:29 +0000576; NOSDWA: v_or_b32_e32 v{{[0-9]+}}, 0x10000,
Nicolai Haehnlea9cc92c2018-11-30 22:55:29 +0000577; SDWA: v_or_b32_e32 v{{[0-9]+}}, 0x10000,
Matt Arsenault8ae38bc2017-12-05 20:32:01 +0000578define amdgpu_kernel void @sdwa_crash_inlineasm_def() #0 {
579bb:
580 br label %bb1
581
582bb1: ; preds = %bb11, %bb
583 %tmp = phi <2 x i32> [ %tmp12, %bb11 ], [ undef, %bb ]
584 br i1 true, label %bb2, label %bb11
585
586bb2: ; preds = %bb1
587 %tmp3 = call i32 asm "v_and_b32_e32 $0, $1, $2", "=v,s,v"(i32 65535, i32 undef) #1
588 %tmp5 = or i32 %tmp3, 65536
589 %tmp6 = insertelement <2 x i32> %tmp, i32 %tmp5, i64 0
590 br label %bb11
591
592bb11: ; preds = %bb10, %bb2
593 %tmp12 = phi <2 x i32> [ %tmp6, %bb2 ], [ %tmp, %bb1 ]
Stanislav Mekhanoshinc8f78f82019-04-05 20:11:32 +0000594 store volatile <2 x i32> %tmp12, <2 x i32> addrspace(1)* undef
Matt Arsenault8ae38bc2017-12-05 20:32:01 +0000595 br label %bb1
596}
Matt Arsenault5660bb62019-11-18 16:48:07 +0530597
Jay Foadfdaa2d02021-02-19 15:04:03 +0000598declare i32 @llvm.amdgcn.workitem.id.x()
599
Matt Arsenault5660bb62019-11-18 16:48:07 +0530600attributes #0 = { "denormal-fp-math"="preserve-sign,preserve-sign" }