blob: aeb314c0a249f50d792af7e8af6d22f645545417 [file] [log] [blame]
Changpeng Fangb28fe032016-09-01 17:54:54 +00001//===-- MIMGInstructions.td - MIMG Instruction Defintions -----------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10class MIMG_Mask <string op, int channels> {
11 string Op = op;
12 int Channels = channels;
13}
14
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +000015class MIMG_Atomic_Size <string op, bit is32Bit> {
16 string Op = op;
17 int AtomicSize = !if(is32Bit, 1, 2);
18}
19
Nicolai Haehnlef2674312018-06-21 13:36:01 +000020class MIMG_Gather_Size <string op, int channels> {
21 string Op = op;
22 int Channels = channels;
23}
24
Changpeng Fangb28fe032016-09-01 17:54:54 +000025class mimg <bits<7> si, bits<7> vi = si> {
26 field bits<7> SI = si;
27 field bits<7> VI = vi;
28}
29
30class MIMG_Helper <dag outs, dag ins, string asm,
31 string dns=""> : MIMG<outs, ins, asm,[]> {
32 let mayLoad = 1;
33 let mayStore = 0;
34 let hasPostISelHook = 1;
35 let DecoderNamespace = dns;
36 let isAsmParserOnly = !if(!eq(dns,""), 1, 0);
37 let AsmMatchConverter = "cvtMIMG";
Tom Stellard244891d2016-12-20 15:52:17 +000038 let usesCustomInserter = 1;
Marek Olsakb83f5c92017-07-04 14:43:38 +000039 let SchedRW = [WriteVMEM];
Changpeng Fangb28fe032016-09-01 17:54:54 +000040}
41
42class MIMG_NoSampler_Helper <bits<7> op, string asm,
43 RegisterClass dst_rc,
44 RegisterClass addr_rc,
Nicolai Haehnlef2674312018-06-21 13:36:01 +000045 bit has_d16,
46 string dns="">
47 : MIMG_Helper <(outs dst_rc:$vdata),
48 !con((ins addr_rc:$vaddr, SReg_256:$srsrc,
49 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
50 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
51 !if(has_d16, (ins D16:$d16), (ins))),
52 asm#" $vdata, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da"
53 #!if(has_d16, "$d16", ""),
54 dns>,
55 MIMGe<op> {
Changpeng Fangb28fe032016-09-01 17:54:54 +000056 let ssamp = 0;
Changpeng Fang4737e892018-01-18 22:08:53 +000057
Nicolai Haehnlef2674312018-06-21 13:36:01 +000058 let HasD16 = has_d16;
59 let d16 = !if(HasD16, ?, 0);
Changpeng Fangb28fe032016-09-01 17:54:54 +000060}
61
62multiclass MIMG_NoSampler_Src_Helper <bits<7> op, string asm,
Nicolai Haehnlef2674312018-06-21 13:36:01 +000063 RegisterClass dst_rc,
64 int channels, bit has_d16> {
65 def NAME # _V1 : MIMG_NoSampler_Helper <op, asm, dst_rc, VGPR_32, has_d16,
66 !if(!eq(channels, 1), "AMDGPU", "")>,
67 MIMG_Mask<asm#"_V1", channels>;
68 def NAME # _V2 : MIMG_NoSampler_Helper <op, asm, dst_rc, VReg_64, has_d16>,
69 MIMG_Mask<asm#"_V2", channels>;
70 def NAME # _V3 : MIMG_NoSampler_Helper <op, asm, dst_rc, VReg_96, has_d16>,
71 MIMG_Mask<asm#"_V3", channels>;
72 def NAME # _V4 : MIMG_NoSampler_Helper <op, asm, dst_rc, VReg_128, has_d16>,
73 MIMG_Mask<asm#"_V4", channels>;
Changpeng Fangb28fe032016-09-01 17:54:54 +000074}
75
Nicolai Haehnlef2674312018-06-21 13:36:01 +000076multiclass MIMG_NoSampler <bits<7> op, string asm, bit has_d16> {
77 defm _V1 : MIMG_NoSampler_Src_Helper <op, asm, VGPR_32, 1, has_d16>;
78 defm _V2 : MIMG_NoSampler_Src_Helper <op, asm, VReg_64, 2, has_d16>;
79 defm _V3 : MIMG_NoSampler_Src_Helper <op, asm, VReg_96, 3, has_d16>;
80 defm _V4 : MIMG_NoSampler_Src_Helper <op, asm, VReg_128, 4, has_d16>;
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +000081}
82
Changpeng Fangb28fe032016-09-01 17:54:54 +000083class MIMG_Store_Helper <bits<7> op, string asm,
84 RegisterClass data_rc,
Matt Arsenaultcad7fa82017-12-13 21:07:51 +000085 RegisterClass addr_rc,
Nicolai Haehnlef2674312018-06-21 13:36:01 +000086 bit has_d16,
87 string dns = "">
88 : MIMG_Helper <(outs),
89 !con((ins data_rc:$vdata, addr_rc:$vaddr, SReg_256:$srsrc,
90 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
91 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
92 !if(has_d16, (ins D16:$d16), (ins))),
93 asm#" $vdata, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da"
94 #!if(has_d16, "$d16", ""),
95 dns>,
96 MIMGe<op> {
Changpeng Fangb28fe032016-09-01 17:54:54 +000097 let ssamp = 0;
Matt Arsenaultd94b63d2017-12-29 17:18:18 +000098 let mayLoad = 0;
Changpeng Fangb28fe032016-09-01 17:54:54 +000099 let mayStore = 1;
Matt Arsenaultd94b63d2017-12-29 17:18:18 +0000100 let hasSideEffects = 0;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000101 let hasPostISelHook = 0;
102 let DisableWQM = 1;
Changpeng Fang4737e892018-01-18 22:08:53 +0000103
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000104 let HasD16 = has_d16;
105 let d16 = !if(HasD16, ?, 0);
Changpeng Fangb28fe032016-09-01 17:54:54 +0000106}
107
108multiclass MIMG_Store_Addr_Helper <bits<7> op, string asm,
109 RegisterClass data_rc,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000110 int channels, bit has_d16> {
111 def NAME # _V1 : MIMG_Store_Helper <op, asm, data_rc, VGPR_32, has_d16,
112 !if(!eq(channels, 1), "AMDGPU", "")>,
113 MIMG_Mask<asm#"_V1", channels>;
114 def NAME # _V2 : MIMG_Store_Helper <op, asm, data_rc, VReg_64, has_d16>,
115 MIMG_Mask<asm#"_V2", channels>;
116 def NAME # _V3 : MIMG_Store_Helper <op, asm, data_rc, VReg_96, has_d16>,
117 MIMG_Mask<asm#"_V3", channels>;
118 def NAME # _V4 : MIMG_Store_Helper <op, asm, data_rc, VReg_128, has_d16>,
119 MIMG_Mask<asm#"_V4", channels>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000120}
121
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000122multiclass MIMG_Store <bits<7> op, string asm, bit has_d16> {
123 defm _V1 : MIMG_Store_Addr_Helper <op, asm, VGPR_32, 1, has_d16>;
124 defm _V2 : MIMG_Store_Addr_Helper <op, asm, VReg_64, 2, has_d16>;
125 defm _V3 : MIMG_Store_Addr_Helper <op, asm, VReg_96, 3, has_d16>;
126 defm _V4 : MIMG_Store_Addr_Helper <op, asm, VReg_128, 4, has_d16>;
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000127}
128
Changpeng Fangb28fe032016-09-01 17:54:54 +0000129class MIMG_Atomic_Helper <string asm, RegisterClass data_rc,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000130 RegisterClass addr_rc, string dns="",
131 bit enableDasm = 0> : MIMG_Helper <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000132 (outs data_rc:$vdst),
133 (ins data_rc:$vdata, addr_rc:$vaddr, SReg_256:$srsrc,
Nicolai Haehnle59198ed2018-06-04 14:45:20 +0000134 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
135 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000136 asm#" $vdst, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da",
137 !if(enableDasm, dns, "")> {
Matt Arsenaultd94b63d2017-12-29 17:18:18 +0000138 let mayLoad = 1;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000139 let mayStore = 1;
Matt Arsenaultd94b63d2017-12-29 17:18:18 +0000140 let hasSideEffects = 1; // FIXME: Remove this
Changpeng Fangb28fe032016-09-01 17:54:54 +0000141 let hasPostISelHook = 0;
142 let DisableWQM = 1;
143 let Constraints = "$vdst = $vdata";
144 let AsmMatchConverter = "cvtMIMGAtomic";
145}
146
147class MIMG_Atomic_Real_si<mimg op, string name, string asm,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000148 RegisterClass data_rc, RegisterClass addr_rc,
149 bit enableDasm>
150 : MIMG_Atomic_Helper<asm, data_rc, addr_rc, "SICI", enableDasm>,
151 SIMCInstr<name, SIEncodingFamily.SI>,
152 MIMGe<op.SI> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000153 let isCodeGenOnly = 0;
154 let AssemblerPredicates = [isSICI];
Changpeng Fangb28fe032016-09-01 17:54:54 +0000155 let DisableDecoder = DisableSIDecoder;
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000156 let d16 = 0;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000157}
158
159class MIMG_Atomic_Real_vi<mimg op, string name, string asm,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000160 RegisterClass data_rc, RegisterClass addr_rc,
161 bit enableDasm>
162 : MIMG_Atomic_Helper<asm, data_rc, addr_rc, "VI", enableDasm>,
163 SIMCInstr<name, SIEncodingFamily.VI>,
164 MIMGe<op.VI> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000165 let isCodeGenOnly = 0;
166 let AssemblerPredicates = [isVI];
Changpeng Fangb28fe032016-09-01 17:54:54 +0000167 let DisableDecoder = DisableVIDecoder;
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000168 let d16 = 0;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000169}
170
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000171multiclass MIMG_Atomic_Helper_m <mimg op,
172 string name,
173 string asm,
174 string key,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000175 RegisterClass data_rc,
176 RegisterClass addr_rc,
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000177 bit is32Bit,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000178 bit enableDasm = 0> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000179 let isPseudo = 1, isCodeGenOnly = 1 in {
180 def "" : MIMG_Atomic_Helper<asm, data_rc, addr_rc>,
181 SIMCInstr<name, SIEncodingFamily.NONE>;
182 }
183
184 let ssamp = 0 in {
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000185 def _si : MIMG_Atomic_Real_si<op, name, asm, data_rc, addr_rc, enableDasm>,
186 MIMG_Atomic_Size<key # "_si", is32Bit>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000187
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000188 def _vi : MIMG_Atomic_Real_vi<op, name, asm, data_rc, addr_rc, enableDasm>,
189 MIMG_Atomic_Size<key # "_vi", is32Bit>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000190 }
191}
192
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000193multiclass MIMG_Atomic_Addr_Helper_m <mimg op,
194 string name,
195 string asm,
196 RegisterClass data_rc,
197 bit is32Bit,
198 bit enableDasm = 0> {
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000199 // _V* variants have different address size, but the size is not encoded.
200 // So only one variant can be disassembled. V1 looks the safest to decode.
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000201 defm _V1 : MIMG_Atomic_Helper_m <op, name # "_V1", asm, asm # "_V1", data_rc, VGPR_32, is32Bit, enableDasm>;
202 defm _V2 : MIMG_Atomic_Helper_m <op, name # "_V2", asm, asm # "_V2", data_rc, VReg_64, is32Bit>;
Dmitry Preobrazhenskya0b8cd02018-04-04 13:01:17 +0000203 defm _V3 : MIMG_Atomic_Helper_m <op, name # "_V3", asm, asm # "_V3", data_rc, VReg_96, is32Bit>;
204 defm _V4 : MIMG_Atomic_Helper_m <op, name # "_V4", asm, asm # "_V4", data_rc, VReg_128, is32Bit>;
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000205}
206
207multiclass MIMG_Atomic <mimg op, string asm,
208 RegisterClass data_rc_32 = VGPR_32, // 32-bit atomics
209 RegisterClass data_rc_64 = VReg_64> { // 64-bit atomics
210 // _V* variants have different dst size, but the size is encoded implicitly,
211 // using dmask and tfe. Only 32-bit variant is registered with disassembler.
212 // Other variants are reconstructed by disassembler using dmask and tfe.
213 defm _V1 : MIMG_Atomic_Addr_Helper_m <op, asm # "_V1", asm, data_rc_32, 1, 1>;
214 defm _V2 : MIMG_Atomic_Addr_Helper_m <op, asm # "_V2", asm, data_rc_64, 0>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000215}
216
217class MIMG_Sampler_Helper <bits<7> op, string asm,
218 RegisterClass dst_rc,
219 RegisterClass src_rc,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000220 bit wqm, bit has_d16,
221 string dns="">
222 : MIMG_Helper <(outs dst_rc:$vdata),
223 !con((ins src_rc:$vaddr, SReg_256:$srsrc, SReg_128:$ssamp,
224 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
225 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
226 !if(has_d16, (ins D16:$d16), (ins))),
227 asm#" $vdata, $vaddr, $srsrc, $ssamp$dmask$unorm$glc$slc$r128$tfe$lwe$da"
228 #!if(has_d16, "$d16", ""),
229 dns>,
230 MIMGe<op> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000231 let WQM = wqm;
Changpeng Fang4737e892018-01-18 22:08:53 +0000232
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000233 let HasD16 = has_d16;
234 let d16 = !if(HasD16, ?, 0);
Changpeng Fangb28fe032016-09-01 17:54:54 +0000235}
236
237multiclass MIMG_Sampler_Src_Helper <bits<7> op, string asm,
238 RegisterClass dst_rc,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000239 int channels, bit wqm, bit has_d16> {
240 def _V1 : MIMG_Sampler_Helper <op, asm, dst_rc, VGPR_32, wqm, has_d16,
241 !if(!eq(channels, 1), "AMDGPU", "")>,
242 MIMG_Mask<asm#"_V1", channels>;
243 def _V2 : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_64, wqm, has_d16>,
244 MIMG_Mask<asm#"_V2", channels>;
245 def _V3 : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_96, wqm, has_d16>,
246 MIMG_Mask<asm#"_V3", channels>;
247 def _V4 : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_128, wqm, has_d16>,
248 MIMG_Mask<asm#"_V4", channels>;
249 def _V8 : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_256, wqm, has_d16>,
250 MIMG_Mask<asm#"_V8", channels>;
251 def _V16 : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_512, wqm, has_d16>,
252 MIMG_Mask<asm#"_V16", channels>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000253}
254
Nicolai Haehnle2367f032018-06-21 13:36:13 +0000255multiclass MIMG_Sampler <bits<7> op, AMDGPUSampleVariant sample, bit wqm = 0,
256 bit has_d16 = 1,
257 string asm = "image_sample"#sample.LowerCaseMod> {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000258 defm _V1 : MIMG_Sampler_Src_Helper<op, asm, VGPR_32, 1, wqm, has_d16>;
259 defm _V2 : MIMG_Sampler_Src_Helper<op, asm, VReg_64, 2, wqm, has_d16>;
260 defm _V3 : MIMG_Sampler_Src_Helper<op, asm, VReg_96, 3, wqm, has_d16>;
261 defm _V4 : MIMG_Sampler_Src_Helper<op, asm, VReg_128, 4, wqm, has_d16>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000262}
263
Nicolai Haehnle2367f032018-06-21 13:36:13 +0000264multiclass MIMG_Sampler_WQM <bits<7> op, AMDGPUSampleVariant sample> : MIMG_Sampler<op, sample, 1>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000265
266class MIMG_Gather_Helper <bits<7> op, string asm,
267 RegisterClass dst_rc,
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000268 RegisterClass src_rc,
269 bit wqm,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000270 string dns="">
271 : MIMG <(outs dst_rc:$vdata),
272 (ins src_rc:$vaddr, SReg_256:$srsrc, SReg_128:$ssamp,
273 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
274 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da, D16:$d16),
275 asm#" $vdata, $vaddr, $srsrc, $ssamp$dmask$unorm$glc$slc$r128$tfe$lwe$da$d16",
276 []>,
277 MIMGe<op> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000278 let mayLoad = 1;
279 let mayStore = 0;
280
281 // DMASK was repurposed for GATHER4. 4 components are always
282 // returned and DMASK works like a swizzle - it selects
283 // the component to fetch. The only useful DMASK values are
284 // 1=red, 2=green, 4=blue, 8=alpha. (e.g. 1 returns
285 // (red,red,red,red) etc.) The ISA document doesn't mention
286 // this.
287 // Therefore, disable all code which updates DMASK by setting this:
288 let Gather4 = 1;
289 let hasPostISelHook = 0;
290 let WQM = wqm;
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000291 let HasD16 = 1;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000292
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000293 let DecoderNamespace = dns;
294 let isAsmParserOnly = !if(!eq(dns,""), 1, 0);
Changpeng Fangb28fe032016-09-01 17:54:54 +0000295}
296
Changpeng Fang4737e892018-01-18 22:08:53 +0000297
Changpeng Fangb28fe032016-09-01 17:54:54 +0000298multiclass MIMG_Gather_Src_Helper <bits<7> op, string asm,
299 RegisterClass dst_rc,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000300 int channels, bit wqm> {
301 def _V1 : MIMG_Gather_Helper <op, asm, dst_rc, VGPR_32, wqm,
302 !if(!eq(channels, 4), "AMDGPU", "")>,
303 MIMG_Gather_Size<asm#"_V1", channels>;
304 def _V2 : MIMG_Gather_Helper <op, asm, dst_rc, VReg_64, wqm>,
305 MIMG_Gather_Size<asm#"_V2", channels>;
306 def _V3 : MIMG_Gather_Helper <op, asm, dst_rc, VReg_96, wqm>,
307 MIMG_Gather_Size<asm#"_V3", channels>;
308 def _V4 : MIMG_Gather_Helper <op, asm, dst_rc, VReg_128, wqm>,
309 MIMG_Gather_Size<asm#"_V4", channels>;
310 def _V8 : MIMG_Gather_Helper <op, asm, dst_rc, VReg_256, wqm>,
311 MIMG_Gather_Size<asm#"_V8", channels>;
312 def _V16 : MIMG_Gather_Helper <op, asm, dst_rc, VReg_512, wqm>,
313 MIMG_Gather_Size<asm#"_V16", channels>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000314}
315
Nicolai Haehnle2367f032018-06-21 13:36:13 +0000316multiclass MIMG_Gather <bits<7> op, AMDGPUSampleVariant sample, bit wqm = 0,
317 string asm = "image_gather4"#sample.LowerCaseMod> {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000318 defm _V2 : MIMG_Gather_Src_Helper<op, asm, VReg_64, 2, wqm>; /* for packed D16 only */
319 defm _V4 : MIMG_Gather_Src_Helper<op, asm, VReg_128, 4, wqm>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000320}
321
Nicolai Haehnle2367f032018-06-21 13:36:13 +0000322multiclass MIMG_Gather_WQM <bits<7> op, AMDGPUSampleVariant sample>
323 : MIMG_Gather<op, sample, 1>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000324
325//===----------------------------------------------------------------------===//
326// MIMG Instructions
327//===----------------------------------------------------------------------===//
328let SubtargetPredicate = isGCN in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000329defm IMAGE_LOAD : MIMG_NoSampler <0x00000000, "image_load", 1>;
330defm IMAGE_LOAD_MIP : MIMG_NoSampler <0x00000001, "image_load_mip", 1>;
331defm IMAGE_LOAD_PCK : MIMG_NoSampler <0x00000002, "image_load_pck", 0>;
332defm IMAGE_LOAD_PCK_SGN : MIMG_NoSampler <0x00000003, "image_load_pck_sgn", 0>;
333defm IMAGE_LOAD_MIP_PCK : MIMG_NoSampler <0x00000004, "image_load_mip_pck", 0>;
334defm IMAGE_LOAD_MIP_PCK_SGN : MIMG_NoSampler <0x00000005, "image_load_mip_pck_sgn", 0>;
335defm IMAGE_STORE : MIMG_Store <0x00000008, "image_store", 1>;
336defm IMAGE_STORE_MIP : MIMG_Store <0x00000009, "image_store_mip", 1>;
337defm IMAGE_STORE_PCK : MIMG_Store <0x0000000a, "image_store_pck", 0>;
338defm IMAGE_STORE_MIP_PCK : MIMG_Store <0x0000000b, "image_store_mip_pck", 0>;
Matt Arsenault856777d2017-12-08 20:00:57 +0000339
340let mayLoad = 0, mayStore = 0 in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000341defm IMAGE_GET_RESINFO : MIMG_NoSampler <0x0000000e, "image_get_resinfo", 0>;
Matt Arsenault856777d2017-12-08 20:00:57 +0000342}
343
Changpeng Fangb28fe032016-09-01 17:54:54 +0000344defm IMAGE_ATOMIC_SWAP : MIMG_Atomic <mimg<0x0f, 0x10>, "image_atomic_swap">;
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000345defm IMAGE_ATOMIC_CMPSWAP : MIMG_Atomic <mimg<0x10, 0x11>, "image_atomic_cmpswap", VReg_64, VReg_128>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000346defm IMAGE_ATOMIC_ADD : MIMG_Atomic <mimg<0x11, 0x12>, "image_atomic_add">;
347defm IMAGE_ATOMIC_SUB : MIMG_Atomic <mimg<0x12, 0x13>, "image_atomic_sub">;
348//def IMAGE_ATOMIC_RSUB : MIMG_NoPattern_ <"image_atomic_rsub", 0x00000013>; -- not on VI
349defm IMAGE_ATOMIC_SMIN : MIMG_Atomic <mimg<0x14>, "image_atomic_smin">;
350defm IMAGE_ATOMIC_UMIN : MIMG_Atomic <mimg<0x15>, "image_atomic_umin">;
351defm IMAGE_ATOMIC_SMAX : MIMG_Atomic <mimg<0x16>, "image_atomic_smax">;
352defm IMAGE_ATOMIC_UMAX : MIMG_Atomic <mimg<0x17>, "image_atomic_umax">;
353defm IMAGE_ATOMIC_AND : MIMG_Atomic <mimg<0x18>, "image_atomic_and">;
354defm IMAGE_ATOMIC_OR : MIMG_Atomic <mimg<0x19>, "image_atomic_or">;
355defm IMAGE_ATOMIC_XOR : MIMG_Atomic <mimg<0x1a>, "image_atomic_xor">;
356defm IMAGE_ATOMIC_INC : MIMG_Atomic <mimg<0x1b>, "image_atomic_inc">;
357defm IMAGE_ATOMIC_DEC : MIMG_Atomic <mimg<0x1c>, "image_atomic_dec">;
358//def IMAGE_ATOMIC_FCMPSWAP : MIMG_NoPattern_ <"image_atomic_fcmpswap", 0x0000001d>; -- not on VI
359//def IMAGE_ATOMIC_FMIN : MIMG_NoPattern_ <"image_atomic_fmin", 0x0000001e>; -- not on VI
360//def IMAGE_ATOMIC_FMAX : MIMG_NoPattern_ <"image_atomic_fmax", 0x0000001f>; -- not on VI
Nicolai Haehnle2367f032018-06-21 13:36:13 +0000361defm IMAGE_SAMPLE : MIMG_Sampler_WQM <0x00000020, AMDGPUSample>;
362defm IMAGE_SAMPLE_CL : MIMG_Sampler_WQM <0x00000021, AMDGPUSample_cl>;
363defm IMAGE_SAMPLE_D : MIMG_Sampler <0x00000022, AMDGPUSample_d>;
364defm IMAGE_SAMPLE_D_CL : MIMG_Sampler <0x00000023, AMDGPUSample_d_cl>;
365defm IMAGE_SAMPLE_L : MIMG_Sampler <0x00000024, AMDGPUSample_l>;
366defm IMAGE_SAMPLE_B : MIMG_Sampler_WQM <0x00000025, AMDGPUSample_b>;
367defm IMAGE_SAMPLE_B_CL : MIMG_Sampler_WQM <0x00000026, AMDGPUSample_b_cl>;
368defm IMAGE_SAMPLE_LZ : MIMG_Sampler <0x00000027, AMDGPUSample_lz>;
369defm IMAGE_SAMPLE_C : MIMG_Sampler_WQM <0x00000028, AMDGPUSample_c>;
370defm IMAGE_SAMPLE_C_CL : MIMG_Sampler_WQM <0x00000029, AMDGPUSample_c_cl>;
371defm IMAGE_SAMPLE_C_D : MIMG_Sampler <0x0000002a, AMDGPUSample_c_d>;
372defm IMAGE_SAMPLE_C_D_CL : MIMG_Sampler <0x0000002b, AMDGPUSample_c_d_cl>;
373defm IMAGE_SAMPLE_C_L : MIMG_Sampler <0x0000002c, AMDGPUSample_c_l>;
374defm IMAGE_SAMPLE_C_B : MIMG_Sampler_WQM <0x0000002d, AMDGPUSample_c_b>;
375defm IMAGE_SAMPLE_C_B_CL : MIMG_Sampler_WQM <0x0000002e, AMDGPUSample_c_b_cl>;
376defm IMAGE_SAMPLE_C_LZ : MIMG_Sampler <0x0000002f, AMDGPUSample_c_lz>;
377defm IMAGE_SAMPLE_O : MIMG_Sampler_WQM <0x00000030, AMDGPUSample_o>;
378defm IMAGE_SAMPLE_CL_O : MIMG_Sampler_WQM <0x00000031, AMDGPUSample_cl_o>;
379defm IMAGE_SAMPLE_D_O : MIMG_Sampler <0x00000032, AMDGPUSample_d_o>;
380defm IMAGE_SAMPLE_D_CL_O : MIMG_Sampler <0x00000033, AMDGPUSample_d_cl_o>;
381defm IMAGE_SAMPLE_L_O : MIMG_Sampler <0x00000034, AMDGPUSample_l_o>;
382defm IMAGE_SAMPLE_B_O : MIMG_Sampler_WQM <0x00000035, AMDGPUSample_b_o>;
383defm IMAGE_SAMPLE_B_CL_O : MIMG_Sampler_WQM <0x00000036, AMDGPUSample_b_cl_o>;
384defm IMAGE_SAMPLE_LZ_O : MIMG_Sampler <0x00000037, AMDGPUSample_lz_o>;
385defm IMAGE_SAMPLE_C_O : MIMG_Sampler_WQM <0x00000038, AMDGPUSample_c_o>;
386defm IMAGE_SAMPLE_C_CL_O : MIMG_Sampler_WQM <0x00000039, AMDGPUSample_c_cl_o>;
387defm IMAGE_SAMPLE_C_D_O : MIMG_Sampler <0x0000003a, AMDGPUSample_c_d_o>;
388defm IMAGE_SAMPLE_C_D_CL_O : MIMG_Sampler <0x0000003b, AMDGPUSample_c_d_cl_o>;
389defm IMAGE_SAMPLE_C_L_O : MIMG_Sampler <0x0000003c, AMDGPUSample_c_l_o>;
390defm IMAGE_SAMPLE_C_B_CL_O : MIMG_Sampler_WQM <0x0000003e, AMDGPUSample_c_b_cl_o>;
391defm IMAGE_SAMPLE_C_B_O : MIMG_Sampler_WQM <0x0000003d, AMDGPUSample_c_b_o>;
392defm IMAGE_SAMPLE_C_LZ_O : MIMG_Sampler <0x0000003f, AMDGPUSample_c_lz_o>;
393defm IMAGE_GATHER4 : MIMG_Gather_WQM <0x00000040, AMDGPUSample>;
394defm IMAGE_GATHER4_CL : MIMG_Gather_WQM <0x00000041, AMDGPUSample_cl>;
395defm IMAGE_GATHER4_L : MIMG_Gather <0x00000044, AMDGPUSample_l>;
396defm IMAGE_GATHER4_B : MIMG_Gather_WQM <0x00000045, AMDGPUSample_b>;
397defm IMAGE_GATHER4_B_CL : MIMG_Gather_WQM <0x00000046, AMDGPUSample_b_cl>;
398defm IMAGE_GATHER4_LZ : MIMG_Gather <0x00000047, AMDGPUSample_lz>;
399defm IMAGE_GATHER4_C : MIMG_Gather_WQM <0x00000048, AMDGPUSample_c>;
400defm IMAGE_GATHER4_C_CL : MIMG_Gather_WQM <0x00000049, AMDGPUSample_c_cl>;
401defm IMAGE_GATHER4_C_L : MIMG_Gather <0x0000004c, AMDGPUSample_c_l>;
402defm IMAGE_GATHER4_C_B : MIMG_Gather_WQM <0x0000004d, AMDGPUSample_c_b>;
403defm IMAGE_GATHER4_C_B_CL : MIMG_Gather_WQM <0x0000004e, AMDGPUSample_c_b_cl>;
404defm IMAGE_GATHER4_C_LZ : MIMG_Gather <0x0000004f, AMDGPUSample_c_lz>;
405defm IMAGE_GATHER4_O : MIMG_Gather_WQM <0x00000050, AMDGPUSample_o>;
406defm IMAGE_GATHER4_CL_O : MIMG_Gather_WQM <0x00000051, AMDGPUSample_cl_o>;
407defm IMAGE_GATHER4_L_O : MIMG_Gather <0x00000054, AMDGPUSample_l_o>;
408defm IMAGE_GATHER4_B_O : MIMG_Gather_WQM <0x00000055, AMDGPUSample_b_o>;
409defm IMAGE_GATHER4_B_CL_O : MIMG_Gather <0x00000056, AMDGPUSample_b_cl_o>;
410defm IMAGE_GATHER4_LZ_O : MIMG_Gather <0x00000057, AMDGPUSample_lz_o>;
411defm IMAGE_GATHER4_C_O : MIMG_Gather_WQM <0x00000058, AMDGPUSample_c_o>;
412defm IMAGE_GATHER4_C_CL_O : MIMG_Gather_WQM <0x00000059, AMDGPUSample_c_cl_o>;
413defm IMAGE_GATHER4_C_L_O : MIMG_Gather <0x0000005c, AMDGPUSample_c_l_o>;
414defm IMAGE_GATHER4_C_B_O : MIMG_Gather_WQM <0x0000005d, AMDGPUSample_c_b_o>;
415defm IMAGE_GATHER4_C_B_CL_O : MIMG_Gather_WQM <0x0000005e, AMDGPUSample_c_b_cl_o>;
416defm IMAGE_GATHER4_C_LZ_O : MIMG_Gather <0x0000005f, AMDGPUSample_c_lz_o>;
Matt Arsenault856777d2017-12-08 20:00:57 +0000417
418let mayLoad = 0, mayStore = 0 in {
Nicolai Haehnle2367f032018-06-21 13:36:13 +0000419defm IMAGE_GET_LOD : MIMG_Sampler <0x00000060, AMDGPUSample, 1, 0, "image_get_lod">;
Matt Arsenault856777d2017-12-08 20:00:57 +0000420}
421
Nicolai Haehnle2367f032018-06-21 13:36:13 +0000422defm IMAGE_SAMPLE_CD : MIMG_Sampler <0x00000068, AMDGPUSample_cd>;
423defm IMAGE_SAMPLE_CD_CL : MIMG_Sampler <0x00000069, AMDGPUSample_cd_cl>;
424defm IMAGE_SAMPLE_C_CD : MIMG_Sampler <0x0000006a, AMDGPUSample_c_cd>;
425defm IMAGE_SAMPLE_C_CD_CL : MIMG_Sampler <0x0000006b, AMDGPUSample_c_cd_cl>;
426defm IMAGE_SAMPLE_CD_O : MIMG_Sampler <0x0000006c, AMDGPUSample_cd_o>;
427defm IMAGE_SAMPLE_CD_CL_O : MIMG_Sampler <0x0000006d, AMDGPUSample_cd_cl_o>;
428defm IMAGE_SAMPLE_C_CD_O : MIMG_Sampler <0x0000006e, AMDGPUSample_c_cd_o>;
429defm IMAGE_SAMPLE_C_CD_CL_O : MIMG_Sampler <0x0000006f, AMDGPUSample_c_cd_cl_o>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000430//def IMAGE_RSRC256 : MIMG_NoPattern_RSRC256 <"image_rsrc256", 0x0000007e>;
431//def IMAGE_SAMPLER : MIMG_NoPattern_ <"image_sampler", 0x0000007f>;
432}
433
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000434/********** ============================== **********/
435/********** Dimension-aware image patterns **********/
436/********** ============================== **********/
437
438class getDwordsType<int dwords> {
439 int NumDwords = dwords;
440 string suffix = !if(!lt(dwords, 1), ?,
441 !if(!eq(dwords, 1), "_V1",
442 !if(!eq(dwords, 2), "_V2",
443 !if(!le(dwords, 4), "_V4",
444 !if(!le(dwords, 8), "_V8",
445 !if(!le(dwords, 16), "_V16", ?))))));
446 ValueType VT = !if(!lt(dwords, 1), ?,
447 !if(!eq(dwords, 1), f32,
448 !if(!eq(dwords, 2), v2f32,
449 !if(!le(dwords, 4), v4f32,
450 !if(!le(dwords, 8), v8f32,
451 !if(!le(dwords, 16), v16f32, ?))))));
452 RegisterClass VReg = !if(!lt(dwords, 1), ?,
453 !if(!eq(dwords, 1), VGPR_32,
454 !if(!eq(dwords, 2), VReg_64,
455 !if(!le(dwords, 4), VReg_128,
456 !if(!le(dwords, 8), VReg_256,
457 !if(!le(dwords, 16), VReg_512, ?))))));
458}
459
460class makeRegSequence_Fold<int i, dag d> {
461 int idx = i;
462 dag lhs = d;
463}
464
465// Generate a dag node which returns a vector register of class RC into which
466// the source operands given by names have been inserted (assuming that each
467// name corresponds to an operand whose size is equal to a subregister).
468class makeRegSequence<ValueType vt, RegisterClass RC, list<string> names> {
469 dag ret =
470 !if(!eq(!size(names), 1),
471 !dag(COPY_TO_REGCLASS, [?, RC], [names[0], ?]),
472 !foldl(makeRegSequence_Fold<0, (vt (IMPLICIT_DEF))>, names, f, name,
473 makeRegSequence_Fold<
474 !add(f.idx, 1),
475 !con((INSERT_SUBREG f.lhs),
476 !dag(INSERT_SUBREG, [?, !cast<SubRegIndex>("sub"#f.idx)],
477 [name, ?]))>).lhs);
478}
479
480class ImageDimPattern<AMDGPUImageDimIntrinsic I,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000481 string dop, ValueType dty, bit d16,
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000482 string suffix = ""> : GCNPat<(undef), (undef)> {
483 list<AMDGPUArg> AddrArgs = I.P.AddrDefaultArgs;
484 getDwordsType AddrDwords = getDwordsType<!size(AddrArgs)>;
485
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000486 MIMG MI =
487 !cast<MIMG>(!strconcat("IMAGE_", I.P.OpMod, dop, AddrDwords.suffix, suffix));
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000488
489 // DAG fragment to match data arguments (vdata for store/atomic, dmask
490 // for non-atomic).
491 dag MatchDataDag =
492 !con(!dag(I, !foreach(arg, I.P.DataArgs, dty),
493 !foreach(arg, I.P.DataArgs, arg.Name)),
494 !if(I.P.IsAtomic, (I), (I i32:$dmask)));
495
496 // DAG fragment to match vaddr arguments.
497 dag MatchAddrDag = !dag(I, !foreach(arg, AddrArgs, arg.Type.VT),
498 !foreach(arg, AddrArgs, arg.Name));
499
500 // DAG fragment to match sampler resource and unorm arguments.
501 dag MatchSamplerDag = !if(I.P.IsSample, (I v4i32:$sampler, i1:$unorm), (I));
502
503 // DAG node that generates the MI vdata for store/atomic
504 getDwordsType DataDwords = getDwordsType<!size(I.P.DataArgs)>;
505 dag GenDataDag =
506 !if(I.P.IsAtomic, (MI makeRegSequence<DataDwords.VT, DataDwords.VReg,
507 !foreach(arg, I.P.DataArgs, arg.Name)>.ret),
508 !if(!size(I.P.DataArgs), (MI $vdata), (MI)));
509
510 // DAG node that generates the MI vaddr
511 dag GenAddrDag = makeRegSequence<AddrDwords.VT, AddrDwords.VReg,
512 !foreach(arg, AddrArgs, arg.Name)>.ret;
513 // DAG fragments that generate various inline flags
514 dag GenDmask =
515 !if(I.P.IsAtomic, (MI !add(!shl(1, DataDwords.NumDwords), -1)),
516 (MI (as_i32imm $dmask)));
517 dag GenGLC =
518 !if(I.P.IsAtomic, (MI 1),
519 (MI (bitextract_imm<0> $cachepolicy)));
520
521 dag MatchIntrinsic = !con(MatchDataDag,
522 MatchAddrDag,
523 (I v8i32:$rsrc),
524 MatchSamplerDag,
525 (I 0/*texfailctrl*/,
526 i32:$cachepolicy));
527 let PatternToMatch =
528 !if(!size(I.RetTypes), (dty MatchIntrinsic), MatchIntrinsic);
529
530 bit IsCmpSwap = !and(I.P.IsAtomic, !eq(!size(I.P.DataArgs), 2));
531 dag ImageInstruction =
532 !con(GenDataDag,
533 (MI GenAddrDag),
534 (MI $rsrc),
535 !if(I.P.IsSample, (MI $sampler), (MI)),
536 GenDmask,
537 !if(I.P.IsSample, (MI (as_i1imm $unorm)), (MI 1)),
538 GenGLC,
539 (MI (bitextract_imm<1> $cachepolicy),
540 0, /* r128 */
541 0, /* tfe */
542 0 /*(as_i1imm $lwe)*/,
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000543 { I.P.Dim.DA }),
544 !if(MI.HasD16, (MI d16), (MI)));
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000545 let ResultInstrs = [
546 !if(IsCmpSwap, (EXTRACT_SUBREG ImageInstruction, sub0), ImageInstruction)
547 ];
548}
549
550foreach intr = !listconcat(AMDGPUImageDimIntrinsics,
551 AMDGPUImageDimGetResInfoIntrinsics) in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000552 def intr#_pat_v1 : ImageDimPattern<intr, "_V1", f32, 0>;
553 def intr#_pat_v2 : ImageDimPattern<intr, "_V2", v2f32, 0>;
554 def intr#_pat_v4 : ImageDimPattern<intr, "_V4", v4f32, 0>;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000555}
556
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000557multiclass ImageDimD16Helper<AMDGPUImageDimIntrinsic I,
558 AMDGPUImageDimIntrinsic d16helper> {
559 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000560 def _unpacked_v1 : ImageDimPattern<I, "_V1", f16, 1>;
561 def _unpacked_v2 : ImageDimPattern<d16helper, "_V2", v2i32, 1>;
562 def _unpacked_v4 : ImageDimPattern<d16helper, "_V4", v4i32, 1>;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000563 } // End HasUnpackedD16VMem.
564
565 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000566 def _packed_v1 : ImageDimPattern<I, "_V1", f16, 1>;
567 def _packed_v2 : ImageDimPattern<I, "_V1", v2f16, 1>;
568 def _packed_v4 : ImageDimPattern<I, "_V2", v4f16, 1>;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000569 } // End HasPackedD16VMem.
570}
571
572foreach intr = AMDGPUImageDimIntrinsics in {
573 def intr#_d16helper_profile : AMDGPUDimProfileCopy<intr.P> {
574 let RetTypes = !foreach(ty, intr.P.RetTypes, llvm_any_ty);
575 let DataArgs = !foreach(arg, intr.P.DataArgs, AMDGPUArg<llvm_any_ty, arg.Name>);
576 }
577
578 let TargetPrefix = "SI", isTarget = 1 in
579 def int_SI_image_d16helper_ # intr.P.OpMod # intr.P.Dim.Name :
580 AMDGPUImageDimIntrinsic<!cast<AMDGPUDimProfile>(intr#"_d16helper_profile"),
581 intr.IntrProperties, intr.Properties>;
582
583 defm intr#_d16 :
584 ImageDimD16Helper<
585 intr, !cast<AMDGPUImageDimIntrinsic>(
586 "int_SI_image_d16helper_" # intr.P.OpMod # intr.P.Dim.Name)>;
587}
588
589foreach intr = AMDGPUImageDimGatherIntrinsics in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000590 def intr#_pat3 : ImageDimPattern<intr, "_V4", v4f32, 0>;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000591
592 def intr#_d16helper_profile : AMDGPUDimProfileCopy<intr.P> {
593 let RetTypes = !foreach(ty, intr.P.RetTypes, llvm_any_ty);
594 let DataArgs = !foreach(arg, intr.P.DataArgs, AMDGPUArg<llvm_any_ty, arg.Name>);
595 }
596
597 let TargetPrefix = "SI", isTarget = 1 in
598 def int_SI_image_d16helper_ # intr.P.OpMod # intr.P.Dim.Name :
599 AMDGPUImageDimIntrinsic<!cast<AMDGPUDimProfile>(intr#"_d16helper_profile"),
600 intr.IntrProperties, intr.Properties>;
601
602 let SubtargetPredicate = HasUnpackedD16VMem in {
603 def intr#_unpacked_v4 :
604 ImageDimPattern<!cast<AMDGPUImageDimIntrinsic>(
605 "int_SI_image_d16helper_" # intr.P.OpMod # intr.P.Dim.Name),
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000606 "_V4", v4i32, 1>;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000607 } // End HasUnpackedD16VMem.
608
609 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000610 def intr#_packed_v4 : ImageDimPattern<intr, "_V2", v4f16, 1>;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000611 } // End HasPackedD16VMem.
612}
613
614foreach intr = AMDGPUImageDimAtomicIntrinsics in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000615 def intr#_pat1 : ImageDimPattern<intr, "_V1", i32, 0>;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000616}
617
Changpeng Fangb28fe032016-09-01 17:54:54 +0000618/********** ======================= **********/
619/********** Image sampling patterns **********/
620/********** ======================= **********/
621
Changpeng Fang4737e892018-01-18 22:08:53 +0000622// ImageSample for amdgcn
Changpeng Fangb28fe032016-09-01 17:54:54 +0000623// TODO:
Changpeng Fang4737e892018-01-18 22:08:53 +0000624// 1. Handle v4i32 rsrc type (Register Class for the instruction to be SReg_128).
625// 2. Add A16 support when we pass address of half type.
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000626multiclass ImageSamplePattern<SDPatternOperator name, MIMG opcode,
627 ValueType dt, ValueType vt, bit d16> {
Matt Arsenault90c75932017-10-03 00:06:41 +0000628 def : GCNPat<
Changpeng Fang8236fe12016-11-14 18:33:18 +0000629 (dt (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, i32:$dmask, i1:$unorm, i1:$glc,
Changpeng Fangb28fe032016-09-01 17:54:54 +0000630 i1:$slc, i1:$lwe, i1:$da)),
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000631 !con((opcode $addr, $rsrc, $sampler, (as_i32imm $dmask), (as_i1imm $unorm),
632 (as_i1imm $glc), (as_i1imm $slc), 0, 0, (as_i1imm $lwe),
633 (as_i1imm $da)),
634 !if(opcode.HasD16, (opcode d16), (opcode)))
Changpeng Fangb28fe032016-09-01 17:54:54 +0000635 >;
636}
637
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000638multiclass ImageSampleDataPatterns<SDPatternOperator name, string opcode,
639 ValueType dt, bit d16> {
640 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V1), dt, f32, d16>;
641 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V2), dt, v2f32, d16>;
642 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V4), dt, v4f32, d16>;
643 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V8), dt, v8f32, d16>;
644 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V16), dt, v16f32, d16>;
Changpeng Fang8236fe12016-11-14 18:33:18 +0000645}
646
Changpeng Fang4737e892018-01-18 22:08:53 +0000647// ImageSample patterns.
648multiclass ImageSamplePatterns<SDPatternOperator name, string opcode> {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000649 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), f32, 0>;
650 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v2f32, 0>;
651 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4f32, 0>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000652
653 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000654 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000655 } // End HasUnpackedD16VMem.
656
657 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000658 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>;
659 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), v2f16, 1>;
660 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v4f16, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000661 } // End HasPackedD16VMem.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000662}
663
Changpeng Fang4737e892018-01-18 22:08:53 +0000664// ImageSample alternative patterns for illegal vector half Types.
665multiclass ImageSampleAltPatterns<SDPatternOperator name, string opcode> {
666 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000667 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>;
668 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000669 } // End HasUnpackedD16VMem.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000670}
671
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000672// ImageGather4 patterns.
673multiclass ImageGather4Patterns<SDPatternOperator name, string opcode> {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000674 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4f32, 0>;
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000675
676 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000677 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v4f16, 1>;
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000678 } // End HasPackedD16VMem.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000679}
680
681// ImageGather4 alternative patterns for illegal vector half Types.
682multiclass ImageGather4AltPatterns<SDPatternOperator name, string opcode> {
683 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000684 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>;
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000685 } // End HasUnpackedD16VMem.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000686}
687
Changpeng Fang4737e892018-01-18 22:08:53 +0000688// ImageLoad for amdgcn.
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000689multiclass ImageLoadPattern<SDPatternOperator name, MIMG opcode,
690 ValueType dt, ValueType vt, bit d16> {
Matt Arsenault90c75932017-10-03 00:06:41 +0000691 def : GCNPat <
Changpeng Fang8236fe12016-11-14 18:33:18 +0000692 (dt (name vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc, i1:$lwe,
Tom Stellardfac248c2016-10-12 16:35:29 +0000693 i1:$da)),
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000694 !con((opcode $addr, $rsrc, (as_i32imm $dmask), 1, (as_i1imm $glc),
695 (as_i1imm $slc), 0, 0, (as_i1imm $lwe), (as_i1imm $da)),
696 !if(opcode.HasD16, (opcode d16), (opcode)))
Tom Stellardfac248c2016-10-12 16:35:29 +0000697 >;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000698}
699
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000700multiclass ImageLoadDataPatterns<SDPatternOperator name, string opcode,
701 ValueType dt, bit d16> {
702 defm : ImageLoadPattern<name, !cast<MIMG>(opcode # _V1), dt, i32, d16>;
703 defm : ImageLoadPattern<name, !cast<MIMG>(opcode # _V2), dt, v2i32, d16>;
704 defm : ImageLoadPattern<name, !cast<MIMG>(opcode # _V4), dt, v4i32, d16>;
Tom Stellardfac248c2016-10-12 16:35:29 +0000705}
706
Changpeng Fang4737e892018-01-18 22:08:53 +0000707// ImageLoad patterns.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000708// TODO: support v3f32.
709multiclass ImageLoadPatterns<SDPatternOperator name, string opcode> {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000710 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), f32, 0>;
711 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V2), v2f32, 0>;
712 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V4), v4f32, 0>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000713
714 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000715 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000716 } // End HasUnpackedD16VMem.
717
718 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000719 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>;
720 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), v2f16, 1>;
721 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V2), v4f16, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000722 } // End HasPackedD16VMem.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000723}
724
Changpeng Fang4737e892018-01-18 22:08:53 +0000725// ImageLoad alternative patterns for illegal vector half Types.
726multiclass ImageLoadAltPatterns<SDPatternOperator name, string opcode> {
727 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000728 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>;
729 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000730 } // End HasUnPackedD16VMem.
Changpeng Fang4737e892018-01-18 22:08:53 +0000731}
732
733// ImageStore for amdgcn.
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000734multiclass ImageStorePattern<SDPatternOperator name, MIMG opcode,
735 ValueType dt, ValueType vt, bit d16> {
Matt Arsenault90c75932017-10-03 00:06:41 +0000736 def : GCNPat <
Changpeng Fang8236fe12016-11-14 18:33:18 +0000737 (name dt:$data, vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc,
Tom Stellardfac248c2016-10-12 16:35:29 +0000738 i1:$lwe, i1:$da),
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000739 !con((opcode $data, $addr, $rsrc, (as_i32imm $dmask), 1, (as_i1imm $glc),
740 (as_i1imm $slc), 0, 0, (as_i1imm $lwe), (as_i1imm $da)),
741 !if(opcode.HasD16, (opcode d16), (opcode)))
Tom Stellardfac248c2016-10-12 16:35:29 +0000742 >;
743}
Changpeng Fangb28fe032016-09-01 17:54:54 +0000744
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000745multiclass ImageStoreDataPatterns<SDPatternOperator name, string opcode,
746 ValueType dt, bit d16> {
747 defm : ImageStorePattern<name, !cast<MIMG>(opcode # _V1), dt, i32, d16>;
748 defm : ImageStorePattern<name, !cast<MIMG>(opcode # _V2), dt, v2i32, d16>;
749 defm : ImageStorePattern<name, !cast<MIMG>(opcode # _V4), dt, v4i32, d16>;
Changpeng Fang8236fe12016-11-14 18:33:18 +0000750}
751
Changpeng Fang4737e892018-01-18 22:08:53 +0000752// ImageStore patterns.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000753// TODO: support v3f32.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000754multiclass ImageStorePatterns<SDPatternOperator name, string opcode> {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000755 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), f32, 0>;
756 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2f32, 0>;
757 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V4), v4f32, 0>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000758
759 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000760 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000761 } // End HasUnpackedD16VMem.
762
763 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000764 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>;
765 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), v2f16, 1>;
766 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v4f16, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000767 } // End HasPackedD16VMem.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000768}
769
Changpeng Fang4737e892018-01-18 22:08:53 +0000770// ImageStore alternative patterns.
771multiclass ImageStoreAltPatterns<SDPatternOperator name, string opcode> {
772 let SubtargetPredicate = HasUnpackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000773 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>;
774 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000775 } // End HasUnpackedD16VMem.
776
777 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnlef2674312018-06-21 13:36:01 +0000778 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), i32, 1>;
779 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000780 } // End HasPackedD16VMem.
781}
782
783// ImageAtomic for amdgcn.
Matt Arsenault90c75932017-10-03 00:06:41 +0000784class ImageAtomicPattern<SDPatternOperator name, MIMG opcode, ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000785 (name i32:$vdata, vt:$addr, v8i32:$rsrc, imm:$r128, imm:$da, imm:$slc),
786 (opcode $vdata, $addr, $rsrc, 1, 1, 1, (as_i1imm $slc), (as_i1imm $r128), 0, 0, (as_i1imm $da))
787>;
788
Changpeng Fang4737e892018-01-18 22:08:53 +0000789// ImageAtomic patterns.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000790multiclass ImageAtomicPatterns<SDPatternOperator name, string opcode> {
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000791 def : ImageAtomicPattern<name, !cast<MIMG>(opcode # _V1_V1), i32>;
792 def : ImageAtomicPattern<name, !cast<MIMG>(opcode # _V1_V2), v2i32>;
793 def : ImageAtomicPattern<name, !cast<MIMG>(opcode # _V1_V4), v4i32>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000794}
795
Changpeng Fang4737e892018-01-18 22:08:53 +0000796// ImageAtomicCmpSwap for amdgcn.
Matt Arsenault90c75932017-10-03 00:06:41 +0000797class ImageAtomicCmpSwapPattern<MIMG opcode, ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000798 (int_amdgcn_image_atomic_cmpswap i32:$vsrc, i32:$vcmp, vt:$addr, v8i32:$rsrc,
799 imm:$r128, imm:$da, imm:$slc),
800 (EXTRACT_SUBREG
801 (opcode (REG_SEQUENCE VReg_64, $vsrc, sub0, $vcmp, sub1),
802 $addr, $rsrc, 3, 1, 1, (as_i1imm $slc), (as_i1imm $r128), 0, 0, (as_i1imm $da)),
803 sub0)
804>;
805
Changpeng Fangb28fe032016-09-01 17:54:54 +0000806// ======= amdgcn Image Intrinsics ==============
807
Changpeng Fang4737e892018-01-18 22:08:53 +0000808// Image load.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000809defm : ImageLoadPatterns<int_amdgcn_image_load, "IMAGE_LOAD">;
810defm : ImageLoadPatterns<int_amdgcn_image_load_mip, "IMAGE_LOAD_MIP">;
Changpeng Fang8236fe12016-11-14 18:33:18 +0000811defm : ImageLoadPatterns<int_amdgcn_image_getresinfo, "IMAGE_GET_RESINFO">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000812defm : ImageLoadAltPatterns<SIImage_load, "IMAGE_LOAD">;
813defm : ImageLoadAltPatterns<SIImage_load_mip, "IMAGE_LOAD_MIP">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000814
Changpeng Fang4737e892018-01-18 22:08:53 +0000815// Image store.
Matt Arsenault1349a042018-05-22 06:32:10 +0000816defm : ImageStorePatterns<int_amdgcn_image_store, "IMAGE_STORE">;
817defm : ImageStorePatterns<int_amdgcn_image_store_mip, "IMAGE_STORE_MIP">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000818defm : ImageStoreAltPatterns<SIImage_store, "IMAGE_STORE">;
819defm : ImageStoreAltPatterns<SIImage_store_mip, "IMAGE_STORE_MIP">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000820
Changpeng Fang4737e892018-01-18 22:08:53 +0000821// Basic sample.
822defm : ImageSamplePatterns<int_amdgcn_image_sample, "IMAGE_SAMPLE">;
823defm : ImageSamplePatterns<int_amdgcn_image_sample_cl, "IMAGE_SAMPLE_CL">;
824defm : ImageSamplePatterns<int_amdgcn_image_sample_d, "IMAGE_SAMPLE_D">;
825defm : ImageSamplePatterns<int_amdgcn_image_sample_d_cl, "IMAGE_SAMPLE_D_CL">;
826defm : ImageSamplePatterns<int_amdgcn_image_sample_l, "IMAGE_SAMPLE_L">;
827defm : ImageSamplePatterns<int_amdgcn_image_sample_b, "IMAGE_SAMPLE_B">;
828defm : ImageSamplePatterns<int_amdgcn_image_sample_b_cl, "IMAGE_SAMPLE_B_CL">;
829defm : ImageSamplePatterns<int_amdgcn_image_sample_lz, "IMAGE_SAMPLE_LZ">;
830defm : ImageSamplePatterns<int_amdgcn_image_sample_cd, "IMAGE_SAMPLE_CD">;
831defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_cl, "IMAGE_SAMPLE_CD_CL">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000832
Changpeng Fang4737e892018-01-18 22:08:53 +0000833// Sample with comparison.
834defm : ImageSamplePatterns<int_amdgcn_image_sample_c, "IMAGE_SAMPLE_C">;
835defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cl, "IMAGE_SAMPLE_C_CL">;
836defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d, "IMAGE_SAMPLE_C_D">;
837defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_cl, "IMAGE_SAMPLE_C_D_CL">;
838defm : ImageSamplePatterns<int_amdgcn_image_sample_c_l, "IMAGE_SAMPLE_C_L">;
839defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b, "IMAGE_SAMPLE_C_B">;
840defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_cl, "IMAGE_SAMPLE_C_B_CL">;
841defm : ImageSamplePatterns<int_amdgcn_image_sample_c_lz, "IMAGE_SAMPLE_C_LZ">;
842defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd, "IMAGE_SAMPLE_C_CD">;
843defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_cl, "IMAGE_SAMPLE_C_CD_CL">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000844
Changpeng Fang4737e892018-01-18 22:08:53 +0000845// Sample with offsets.
846defm : ImageSamplePatterns<int_amdgcn_image_sample_o, "IMAGE_SAMPLE_O">;
847defm : ImageSamplePatterns<int_amdgcn_image_sample_cl_o, "IMAGE_SAMPLE_CL_O">;
848defm : ImageSamplePatterns<int_amdgcn_image_sample_d_o, "IMAGE_SAMPLE_D_O">;
849defm : ImageSamplePatterns<int_amdgcn_image_sample_d_cl_o, "IMAGE_SAMPLE_D_CL_O">;
850defm : ImageSamplePatterns<int_amdgcn_image_sample_l_o, "IMAGE_SAMPLE_L_O">;
851defm : ImageSamplePatterns<int_amdgcn_image_sample_b_o, "IMAGE_SAMPLE_B_O">;
852defm : ImageSamplePatterns<int_amdgcn_image_sample_b_cl_o, "IMAGE_SAMPLE_B_CL_O">;
853defm : ImageSamplePatterns<int_amdgcn_image_sample_lz_o, "IMAGE_SAMPLE_LZ_O">;
854defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_o, "IMAGE_SAMPLE_CD_O">;
855defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_cl_o, "IMAGE_SAMPLE_CD_CL_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000856
Changpeng Fang4737e892018-01-18 22:08:53 +0000857// Sample with comparison and offsets.
858defm : ImageSamplePatterns<int_amdgcn_image_sample_c_o, "IMAGE_SAMPLE_C_O">;
859defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cl_o, "IMAGE_SAMPLE_C_CL_O">;
860defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_o, "IMAGE_SAMPLE_C_D_O">;
861defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_cl_o, "IMAGE_SAMPLE_C_D_CL_O">;
862defm : ImageSamplePatterns<int_amdgcn_image_sample_c_l_o, "IMAGE_SAMPLE_C_L_O">;
863defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_o, "IMAGE_SAMPLE_C_B_O">;
864defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_cl_o, "IMAGE_SAMPLE_C_B_CL_O">;
865defm : ImageSamplePatterns<int_amdgcn_image_sample_c_lz_o, "IMAGE_SAMPLE_C_LZ_O">;
866defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_o, "IMAGE_SAMPLE_C_CD_O">;
867defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_cl_o, "IMAGE_SAMPLE_C_CD_CL_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000868
Changpeng Fang4737e892018-01-18 22:08:53 +0000869// Basic gather4.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000870defm : ImageGather4Patterns<int_amdgcn_image_gather4, "IMAGE_GATHER4">;
871defm : ImageGather4Patterns<int_amdgcn_image_gather4_cl, "IMAGE_GATHER4_CL">;
872defm : ImageGather4Patterns<int_amdgcn_image_gather4_l, "IMAGE_GATHER4_L">;
873defm : ImageGather4Patterns<int_amdgcn_image_gather4_b, "IMAGE_GATHER4_B">;
874defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_cl, "IMAGE_GATHER4_B_CL">;
875defm : ImageGather4Patterns<int_amdgcn_image_gather4_lz, "IMAGE_GATHER4_LZ">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000876
Changpeng Fang4737e892018-01-18 22:08:53 +0000877// Gather4 with comparison.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000878defm : ImageGather4Patterns<int_amdgcn_image_gather4_c, "IMAGE_GATHER4_C">;
879defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_cl, "IMAGE_GATHER4_C_CL">;
880defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_l, "IMAGE_GATHER4_C_L">;
881defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b, "IMAGE_GATHER4_C_B">;
882defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_cl, "IMAGE_GATHER4_C_B_CL">;
883defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_lz, "IMAGE_GATHER4_C_LZ">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000884
Changpeng Fang4737e892018-01-18 22:08:53 +0000885// Gather4 with offsets.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000886defm : ImageGather4Patterns<int_amdgcn_image_gather4_o, "IMAGE_GATHER4_O">;
887defm : ImageGather4Patterns<int_amdgcn_image_gather4_cl_o, "IMAGE_GATHER4_CL_O">;
888defm : ImageGather4Patterns<int_amdgcn_image_gather4_l_o, "IMAGE_GATHER4_L_O">;
889defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_o, "IMAGE_GATHER4_B_O">;
890defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_cl_o, "IMAGE_GATHER4_B_CL_O">;
891defm : ImageGather4Patterns<int_amdgcn_image_gather4_lz_o, "IMAGE_GATHER4_LZ_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000892
Changpeng Fang4737e892018-01-18 22:08:53 +0000893// Gather4 with comparison and offsets.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000894defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_o, "IMAGE_GATHER4_C_O">;
895defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_cl_o, "IMAGE_GATHER4_C_CL_O">;
896defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_l_o, "IMAGE_GATHER4_C_L_O">;
897defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_o, "IMAGE_GATHER4_C_B_O">;
898defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_cl_o, "IMAGE_GATHER4_C_B_CL_O">;
899defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_lz_o, "IMAGE_GATHER4_C_LZ_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000900
Changpeng Fang4737e892018-01-18 22:08:53 +0000901// Basic sample alternative.
902defm : ImageSampleAltPatterns<SIImage_sample, "IMAGE_SAMPLE">;
903defm : ImageSampleAltPatterns<SIImage_sample_cl, "IMAGE_SAMPLE_CL">;
904defm : ImageSampleAltPatterns<SIImage_sample_d, "IMAGE_SAMPLE_D">;
905defm : ImageSampleAltPatterns<SIImage_sample_d_cl, "IMAGE_SAMPLE_D_CL">;
906defm : ImageSampleAltPatterns<SIImage_sample_l, "IMAGE_SAMPLE_L">;
907defm : ImageSampleAltPatterns<SIImage_sample_b, "IMAGE_SAMPLE_B">;
908defm : ImageSampleAltPatterns<SIImage_sample_b_cl, "IMAGE_SAMPLE_B_CL">;
909defm : ImageSampleAltPatterns<SIImage_sample_lz, "IMAGE_SAMPLE_LZ">;
910defm : ImageSampleAltPatterns<SIImage_sample_cd, "IMAGE_SAMPLE_CD">;
911defm : ImageSampleAltPatterns<SIImage_sample_cd_cl, "IMAGE_SAMPLE_CD_CL">;
912
913// Sample with comparison alternative.
914defm : ImageSampleAltPatterns<SIImage_sample_c, "IMAGE_SAMPLE_C">;
915defm : ImageSampleAltPatterns<SIImage_sample_c_cl, "IMAGE_SAMPLE_C_CL">;
916defm : ImageSampleAltPatterns<SIImage_sample_c_d, "IMAGE_SAMPLE_C_D">;
917defm : ImageSampleAltPatterns<SIImage_sample_c_d_cl, "IMAGE_SAMPLE_C_D_CL">;
918defm : ImageSampleAltPatterns<SIImage_sample_c_l, "IMAGE_SAMPLE_C_L">;
919defm : ImageSampleAltPatterns<SIImage_sample_c_b, "IMAGE_SAMPLE_C_B">;
920defm : ImageSampleAltPatterns<SIImage_sample_c_b_cl, "IMAGE_SAMPLE_C_B_CL">;
921defm : ImageSampleAltPatterns<SIImage_sample_c_lz, "IMAGE_SAMPLE_C_LZ">;
922defm : ImageSampleAltPatterns<SIImage_sample_c_cd, "IMAGE_SAMPLE_C_CD">;
923defm : ImageSampleAltPatterns<SIImage_sample_c_cd_cl, "IMAGE_SAMPLE_C_CD_CL">;
924
925// Sample with offsets alternative.
926defm : ImageSampleAltPatterns<SIImage_sample_o, "IMAGE_SAMPLE_O">;
927defm : ImageSampleAltPatterns<SIImage_sample_cl_o, "IMAGE_SAMPLE_CL_O">;
928defm : ImageSampleAltPatterns<SIImage_sample_d_o, "IMAGE_SAMPLE_D_O">;
929defm : ImageSampleAltPatterns<SIImage_sample_d_cl_o, "IMAGE_SAMPLE_D_CL_O">;
930defm : ImageSampleAltPatterns<SIImage_sample_l_o, "IMAGE_SAMPLE_L_O">;
931defm : ImageSampleAltPatterns<SIImage_sample_b_o, "IMAGE_SAMPLE_B_O">;
932defm : ImageSampleAltPatterns<SIImage_sample_b_cl_o, "IMAGE_SAMPLE_B_CL_O">;
933defm : ImageSampleAltPatterns<SIImage_sample_lz_o, "IMAGE_SAMPLE_LZ_O">;
934defm : ImageSampleAltPatterns<SIImage_sample_cd_o, "IMAGE_SAMPLE_CD_O">;
935defm : ImageSampleAltPatterns<SIImage_sample_cd_cl_o, "IMAGE_SAMPLE_CD_CL_O">;
936
937// Sample with comparison and offsets alternative.
938defm : ImageSampleAltPatterns<SIImage_sample_c_o, "IMAGE_SAMPLE_C_O">;
939defm : ImageSampleAltPatterns<SIImage_sample_c_cl_o, "IMAGE_SAMPLE_C_CL_O">;
940defm : ImageSampleAltPatterns<SIImage_sample_c_d_o, "IMAGE_SAMPLE_C_D_O">;
941defm : ImageSampleAltPatterns<SIImage_sample_c_d_cl_o, "IMAGE_SAMPLE_C_D_CL_O">;
942defm : ImageSampleAltPatterns<SIImage_sample_c_l_o, "IMAGE_SAMPLE_C_L_O">;
943defm : ImageSampleAltPatterns<SIImage_sample_c_b_o, "IMAGE_SAMPLE_C_B_O">;
944defm : ImageSampleAltPatterns<SIImage_sample_c_b_cl_o, "IMAGE_SAMPLE_C_B_CL_O">;
945defm : ImageSampleAltPatterns<SIImage_sample_c_lz_o, "IMAGE_SAMPLE_C_LZ_O">;
946defm : ImageSampleAltPatterns<SIImage_sample_c_cd_o, "IMAGE_SAMPLE_C_CD_O">;
947defm : ImageSampleAltPatterns<SIImage_sample_c_cd_cl_o, "IMAGE_SAMPLE_C_CD_CL_O">;
948
949// Basic gather4 alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000950defm : ImageGather4AltPatterns<SIImage_gather4, "IMAGE_GATHER4">;
951defm : ImageGather4AltPatterns<SIImage_gather4_cl, "IMAGE_GATHER4_CL">;
952defm : ImageGather4AltPatterns<SIImage_gather4_l, "IMAGE_GATHER4_L">;
953defm : ImageGather4AltPatterns<SIImage_gather4_b, "IMAGE_GATHER4_B">;
954defm : ImageGather4AltPatterns<SIImage_gather4_b_cl, "IMAGE_GATHER4_B_CL">;
955defm : ImageGather4AltPatterns<SIImage_gather4_lz, "IMAGE_GATHER4_LZ">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000956
957// Gather4 with comparison alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000958defm : ImageGather4AltPatterns<SIImage_gather4_c, "IMAGE_GATHER4_C">;
959defm : ImageGather4AltPatterns<SIImage_gather4_c_cl, "IMAGE_GATHER4_C_CL">;
960defm : ImageGather4AltPatterns<SIImage_gather4_c_l, "IMAGE_GATHER4_C_L">;
961defm : ImageGather4AltPatterns<SIImage_gather4_c_b, "IMAGE_GATHER4_C_B">;
962defm : ImageGather4AltPatterns<SIImage_gather4_c_b_cl, "IMAGE_GATHER4_C_B_CL">;
963defm : ImageGather4AltPatterns<SIImage_gather4_c_lz, "IMAGE_GATHER4_C_LZ">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000964
965// Gather4 with offsets alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000966defm : ImageGather4AltPatterns<SIImage_gather4_o, "IMAGE_GATHER4_O">;
967defm : ImageGather4AltPatterns<SIImage_gather4_cl_o, "IMAGE_GATHER4_CL_O">;
968defm : ImageGather4AltPatterns<SIImage_gather4_l_o, "IMAGE_GATHER4_L_O">;
969defm : ImageGather4AltPatterns<SIImage_gather4_b_o, "IMAGE_GATHER4_B_O">;
970defm : ImageGather4AltPatterns<SIImage_gather4_b_cl_o, "IMAGE_GATHER4_B_CL_O">;
971defm : ImageGather4AltPatterns<SIImage_gather4_lz_o, "IMAGE_GATHER4_LZ_O">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000972
973// Gather4 with comparison and offsets alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000974defm : ImageGather4AltPatterns<SIImage_gather4_c_o, "IMAGE_GATHER4_C_O">;
975defm : ImageGather4AltPatterns<SIImage_gather4_c_cl_o, "IMAGE_GATHER4_C_CL_O">;
976defm : ImageGather4AltPatterns<SIImage_gather4_c_l_o, "IMAGE_GATHER4_C_L_O">;
977defm : ImageGather4AltPatterns<SIImage_gather4_c_b_o, "IMAGE_GATHER4_C_B_O">;
978defm : ImageGather4AltPatterns<SIImage_gather4_c_b_cl_o, "IMAGE_GATHER4_C_B_CL_O">;
979defm : ImageGather4AltPatterns<SIImage_gather4_c_lz_o, "IMAGE_GATHER4_C_LZ_O">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000980
981defm : ImageSamplePatterns<int_amdgcn_image_getlod, "IMAGE_GET_LOD">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000982
983// Image atomics
984defm : ImageAtomicPatterns<int_amdgcn_image_atomic_swap, "IMAGE_ATOMIC_SWAP">;
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000985def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V1, i32>;
986def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V2, v2i32>;
987def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V4, v4i32>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000988defm : ImageAtomicPatterns<int_amdgcn_image_atomic_add, "IMAGE_ATOMIC_ADD">;
989defm : ImageAtomicPatterns<int_amdgcn_image_atomic_sub, "IMAGE_ATOMIC_SUB">;
990defm : ImageAtomicPatterns<int_amdgcn_image_atomic_smin, "IMAGE_ATOMIC_SMIN">;
991defm : ImageAtomicPatterns<int_amdgcn_image_atomic_umin, "IMAGE_ATOMIC_UMIN">;
992defm : ImageAtomicPatterns<int_amdgcn_image_atomic_smax, "IMAGE_ATOMIC_SMAX">;
993defm : ImageAtomicPatterns<int_amdgcn_image_atomic_umax, "IMAGE_ATOMIC_UMAX">;
994defm : ImageAtomicPatterns<int_amdgcn_image_atomic_and, "IMAGE_ATOMIC_AND">;
995defm : ImageAtomicPatterns<int_amdgcn_image_atomic_or, "IMAGE_ATOMIC_OR">;
996defm : ImageAtomicPatterns<int_amdgcn_image_atomic_xor, "IMAGE_ATOMIC_XOR">;
997defm : ImageAtomicPatterns<int_amdgcn_image_atomic_inc, "IMAGE_ATOMIC_INC">;
998defm : ImageAtomicPatterns<int_amdgcn_image_atomic_dec, "IMAGE_ATOMIC_DEC">;