blob: 702660c00b5903738503091d4410a5a37361cc58 [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
Changpeng Fangb28fe032016-09-01 17:54:54 +000020class mimg <bits<7> si, bits<7> vi = si> {
21 field bits<7> SI = si;
22 field bits<7> VI = vi;
23}
24
25class MIMG_Helper <dag outs, dag ins, string asm,
26 string dns=""> : MIMG<outs, ins, asm,[]> {
27 let mayLoad = 1;
28 let mayStore = 0;
29 let hasPostISelHook = 1;
30 let DecoderNamespace = dns;
31 let isAsmParserOnly = !if(!eq(dns,""), 1, 0);
32 let AsmMatchConverter = "cvtMIMG";
Tom Stellard244891d2016-12-20 15:52:17 +000033 let usesCustomInserter = 1;
Marek Olsakb83f5c92017-07-04 14:43:38 +000034 let SchedRW = [WriteVMEM];
Changpeng Fangb28fe032016-09-01 17:54:54 +000035}
36
37class MIMG_NoSampler_Helper <bits<7> op, string asm,
38 RegisterClass dst_rc,
39 RegisterClass addr_rc,
Changpeng Fang4737e892018-01-18 22:08:53 +000040 bit d16_bit=0,
Changpeng Fangb28fe032016-09-01 17:54:54 +000041 string dns=""> : MIMG_Helper <
42 (outs dst_rc:$vdata),
43 (ins addr_rc:$vaddr, SReg_256:$srsrc,
Nicolai Haehnle59198ed2018-06-04 14:45:20 +000044 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
45 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
Changpeng Fang4737e892018-01-18 22:08:53 +000046 asm#" $vdata, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""),
Changpeng Fangb28fe032016-09-01 17:54:54 +000047 dns>, MIMGe<op> {
48 let ssamp = 0;
Changpeng Fang4737e892018-01-18 22:08:53 +000049 let D16 = d16;
50}
51
52multiclass MIMG_NoSampler_Src_Helper_Helper <bits<7> op, string asm,
53 RegisterClass dst_rc,
54 int channels, bit d16_bit,
55 string suffix> {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +000056 def NAME # _V1 # suffix : MIMG_NoSampler_Helper <op, asm, dst_rc, VGPR_32, d16_bit,
57 !if(!eq(channels, 1), "AMDGPU", "")>,
58 MIMG_Mask<asm#"_V1"#suffix, channels>;
59 def NAME # _V2 # suffix : MIMG_NoSampler_Helper <op, asm, dst_rc, VReg_64, d16_bit>,
60 MIMG_Mask<asm#"_V2"#suffix, channels>;
Dmitry Preobrazhenskya0b8cd02018-04-04 13:01:17 +000061 def NAME # _V3 # suffix : MIMG_NoSampler_Helper <op, asm, dst_rc, VReg_96, d16_bit>,
62 MIMG_Mask<asm#"_V3"#suffix, channels>;
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +000063 def NAME # _V4 # suffix : MIMG_NoSampler_Helper <op, asm, dst_rc, VReg_128, d16_bit>,
64 MIMG_Mask<asm#"_V4"#suffix, channels>;
Changpeng Fangb28fe032016-09-01 17:54:54 +000065}
66
67multiclass MIMG_NoSampler_Src_Helper <bits<7> op, string asm,
68 RegisterClass dst_rc,
69 int channels> {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +000070 defm NAME : MIMG_NoSampler_Src_Helper_Helper <op, asm, dst_rc, channels, 0, "">;
Changpeng Fang4737e892018-01-18 22:08:53 +000071
72 let d16 = 1 in {
73 let SubtargetPredicate = HasPackedD16VMem in {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +000074 defm NAME : MIMG_NoSampler_Src_Helper_Helper <op, asm, dst_rc, channels, 1, "_D16">;
Changpeng Fang4737e892018-01-18 22:08:53 +000075 } // End HasPackedD16VMem.
76
77 let SubtargetPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +000078 defm NAME : MIMG_NoSampler_Src_Helper_Helper <op, asm, dst_rc, channels, 1, "_D16_gfx80">;
Changpeng Fang4737e892018-01-18 22:08:53 +000079 } // End HasUnpackedD16VMem.
80 } // End d16 = 1.
Changpeng Fangb28fe032016-09-01 17:54:54 +000081}
82
83multiclass MIMG_NoSampler <bits<7> op, string asm> {
84 defm _V1 : MIMG_NoSampler_Src_Helper <op, asm, VGPR_32, 1>;
85 defm _V2 : MIMG_NoSampler_Src_Helper <op, asm, VReg_64, 2>;
86 defm _V3 : MIMG_NoSampler_Src_Helper <op, asm, VReg_96, 3>;
87 defm _V4 : MIMG_NoSampler_Src_Helper <op, asm, VReg_128, 4>;
88}
89
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +000090multiclass MIMG_PckNoSampler <bits<7> op, string asm> {
91 defm NAME # _V1 : MIMG_NoSampler_Src_Helper_Helper <op, asm, VGPR_32, 1, 0, "">;
92 defm NAME # _V2 : MIMG_NoSampler_Src_Helper_Helper <op, asm, VReg_64, 2, 0, "">;
93 defm NAME # _V3 : MIMG_NoSampler_Src_Helper_Helper <op, asm, VReg_96, 3, 0, "">;
94 defm NAME # _V4 : MIMG_NoSampler_Src_Helper_Helper <op, asm, VReg_128, 4, 0, "">;
95}
96
Changpeng Fangb28fe032016-09-01 17:54:54 +000097class MIMG_Store_Helper <bits<7> op, string asm,
98 RegisterClass data_rc,
Matt Arsenaultcad7fa82017-12-13 21:07:51 +000099 RegisterClass addr_rc,
Changpeng Fang4737e892018-01-18 22:08:53 +0000100 bit d16_bit=0,
Matt Arsenaultcad7fa82017-12-13 21:07:51 +0000101 string dns = ""> : MIMG_Helper <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000102 (outs),
103 (ins data_rc:$vdata, addr_rc:$vaddr, SReg_256:$srsrc,
Nicolai Haehnle59198ed2018-06-04 14:45:20 +0000104 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
105 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
Changpeng Fang4737e892018-01-18 22:08:53 +0000106 asm#" $vdata, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""), dns>, MIMGe<op> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000107 let ssamp = 0;
Matt Arsenaultd94b63d2017-12-29 17:18:18 +0000108 let mayLoad = 0;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000109 let mayStore = 1;
Matt Arsenaultd94b63d2017-12-29 17:18:18 +0000110 let hasSideEffects = 0;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000111 let hasPostISelHook = 0;
112 let DisableWQM = 1;
Changpeng Fang4737e892018-01-18 22:08:53 +0000113 let D16 = d16;
114}
115
116multiclass MIMG_Store_Addr_Helper_Helper <bits<7> op, string asm,
117 RegisterClass data_rc,
118 int channels, bit d16_bit,
119 string suffix> {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000120 def NAME # _V1 # suffix : MIMG_Store_Helper <op, asm, data_rc, VGPR_32, d16_bit,
121 !if(!eq(channels, 1), "AMDGPU", "")>,
122 MIMG_Mask<asm#"_V1"#suffix, channels>;
123 def NAME # _V2 # suffix : MIMG_Store_Helper <op, asm, data_rc, VReg_64, d16_bit>,
124 MIMG_Mask<asm#"_V2"#suffix, channels>;
Dmitry Preobrazhenskya0b8cd02018-04-04 13:01:17 +0000125 def NAME # _V3 # suffix : MIMG_Store_Helper <op, asm, data_rc, VReg_96, d16_bit>,
126 MIMG_Mask<asm#"_V3"#suffix, channels>;
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000127 def NAME # _V4 # suffix : MIMG_Store_Helper <op, asm, data_rc, VReg_128, d16_bit>,
128 MIMG_Mask<asm#"_V4"#suffix, channels>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000129}
130
131multiclass MIMG_Store_Addr_Helper <bits<7> op, string asm,
132 RegisterClass data_rc,
133 int channels> {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000134 defm NAME : MIMG_Store_Addr_Helper_Helper <op, asm, data_rc, channels, 0, "">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000135
136 let d16 = 1 in {
137 let SubtargetPredicate = HasPackedD16VMem in {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000138 defm NAME : MIMG_Store_Addr_Helper_Helper <op, asm, data_rc, channels, 1, "_D16">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000139 } // End HasPackedD16VMem.
140
141 let SubtargetPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in {
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000142 defm NAME : MIMG_Store_Addr_Helper_Helper <op, asm, data_rc, channels, 1, "_D16_gfx80">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000143 } // End HasUnpackedD16VMem.
144 } // End d16 = 1.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000145}
146
147multiclass MIMG_Store <bits<7> op, string asm> {
148 defm _V1 : MIMG_Store_Addr_Helper <op, asm, VGPR_32, 1>;
149 defm _V2 : MIMG_Store_Addr_Helper <op, asm, VReg_64, 2>;
150 defm _V3 : MIMG_Store_Addr_Helper <op, asm, VReg_96, 3>;
151 defm _V4 : MIMG_Store_Addr_Helper <op, asm, VReg_128, 4>;
152}
153
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000154multiclass MIMG_PckStore <bits<7> op, string asm> {
155 defm NAME # _V1 : MIMG_Store_Addr_Helper_Helper <op, asm, VGPR_32, 1, 0, "">;
156 defm NAME # _V2 : MIMG_Store_Addr_Helper_Helper <op, asm, VReg_64, 2, 0, "">;
157 defm NAME # _V3 : MIMG_Store_Addr_Helper_Helper <op, asm, VReg_96, 3, 0, "">;
158 defm NAME # _V4 : MIMG_Store_Addr_Helper_Helper <op, asm, VReg_128, 4, 0, "">;
159}
160
Changpeng Fangb28fe032016-09-01 17:54:54 +0000161class MIMG_Atomic_Helper <string asm, RegisterClass data_rc,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000162 RegisterClass addr_rc, string dns="",
163 bit enableDasm = 0> : MIMG_Helper <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000164 (outs data_rc:$vdst),
165 (ins data_rc:$vdata, addr_rc:$vaddr, SReg_256:$srsrc,
Nicolai Haehnle59198ed2018-06-04 14:45:20 +0000166 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
167 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000168 asm#" $vdst, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da",
169 !if(enableDasm, dns, "")> {
Matt Arsenaultd94b63d2017-12-29 17:18:18 +0000170 let mayLoad = 1;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000171 let mayStore = 1;
Matt Arsenaultd94b63d2017-12-29 17:18:18 +0000172 let hasSideEffects = 1; // FIXME: Remove this
Changpeng Fangb28fe032016-09-01 17:54:54 +0000173 let hasPostISelHook = 0;
174 let DisableWQM = 1;
175 let Constraints = "$vdst = $vdata";
176 let AsmMatchConverter = "cvtMIMGAtomic";
177}
178
179class MIMG_Atomic_Real_si<mimg op, string name, string asm,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000180 RegisterClass data_rc, RegisterClass addr_rc, bit enableDasm> :
181 MIMG_Atomic_Helper<asm, data_rc, addr_rc, "SICI", enableDasm>,
Changpeng Fangb28fe032016-09-01 17:54:54 +0000182 SIMCInstr<name, SIEncodingFamily.SI>,
183 MIMGe<op.SI> {
184 let isCodeGenOnly = 0;
185 let AssemblerPredicates = [isSICI];
Changpeng Fangb28fe032016-09-01 17:54:54 +0000186 let DisableDecoder = DisableSIDecoder;
187}
188
189class MIMG_Atomic_Real_vi<mimg op, string name, string asm,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000190 RegisterClass data_rc, RegisterClass addr_rc, bit enableDasm> :
191 MIMG_Atomic_Helper<asm, data_rc, addr_rc, "VI", enableDasm>,
Changpeng Fangb28fe032016-09-01 17:54:54 +0000192 SIMCInstr<name, SIEncodingFamily.VI>,
193 MIMGe<op.VI> {
194 let isCodeGenOnly = 0;
195 let AssemblerPredicates = [isVI];
Changpeng Fangb28fe032016-09-01 17:54:54 +0000196 let DisableDecoder = DisableVIDecoder;
197}
198
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000199multiclass MIMG_Atomic_Helper_m <mimg op,
200 string name,
201 string asm,
202 string key,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000203 RegisterClass data_rc,
204 RegisterClass addr_rc,
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000205 bit is32Bit,
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000206 bit enableDasm = 0> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000207 let isPseudo = 1, isCodeGenOnly = 1 in {
208 def "" : MIMG_Atomic_Helper<asm, data_rc, addr_rc>,
209 SIMCInstr<name, SIEncodingFamily.NONE>;
210 }
211
212 let ssamp = 0 in {
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000213 def _si : MIMG_Atomic_Real_si<op, name, asm, data_rc, addr_rc, enableDasm>,
214 MIMG_Atomic_Size<key # "_si", is32Bit>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000215
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000216 def _vi : MIMG_Atomic_Real_vi<op, name, asm, data_rc, addr_rc, enableDasm>,
217 MIMG_Atomic_Size<key # "_vi", is32Bit>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000218 }
219}
220
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000221multiclass MIMG_Atomic_Addr_Helper_m <mimg op,
222 string name,
223 string asm,
224 RegisterClass data_rc,
225 bit is32Bit,
226 bit enableDasm = 0> {
Dmitry Preobrazhensky6cb42e72018-01-26 14:07:38 +0000227 // _V* variants have different address size, but the size is not encoded.
228 // So only one variant can be disassembled. V1 looks the safest to decode.
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000229 defm _V1 : MIMG_Atomic_Helper_m <op, name # "_V1", asm, asm # "_V1", data_rc, VGPR_32, is32Bit, enableDasm>;
230 defm _V2 : MIMG_Atomic_Helper_m <op, name # "_V2", asm, asm # "_V2", data_rc, VReg_64, is32Bit>;
Dmitry Preobrazhenskya0b8cd02018-04-04 13:01:17 +0000231 defm _V3 : MIMG_Atomic_Helper_m <op, name # "_V3", asm, asm # "_V3", data_rc, VReg_96, is32Bit>;
232 defm _V4 : MIMG_Atomic_Helper_m <op, name # "_V4", asm, asm # "_V4", data_rc, VReg_128, is32Bit>;
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000233}
234
235multiclass MIMG_Atomic <mimg op, string asm,
236 RegisterClass data_rc_32 = VGPR_32, // 32-bit atomics
237 RegisterClass data_rc_64 = VReg_64> { // 64-bit atomics
238 // _V* variants have different dst size, but the size is encoded implicitly,
239 // using dmask and tfe. Only 32-bit variant is registered with disassembler.
240 // Other variants are reconstructed by disassembler using dmask and tfe.
241 defm _V1 : MIMG_Atomic_Addr_Helper_m <op, asm # "_V1", asm, data_rc_32, 1, 1>;
242 defm _V2 : MIMG_Atomic_Addr_Helper_m <op, asm # "_V2", asm, data_rc_64, 0>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000243}
244
245class MIMG_Sampler_Helper <bits<7> op, string asm,
246 RegisterClass dst_rc,
247 RegisterClass src_rc,
Sam Koltonc01faa32016-11-15 13:39:07 +0000248 bit wqm,
Changpeng Fang4737e892018-01-18 22:08:53 +0000249 bit d16_bit=0,
Changpeng Fangb28fe032016-09-01 17:54:54 +0000250 string dns=""> : MIMG_Helper <
251 (outs dst_rc:$vdata),
252 (ins src_rc:$vaddr, SReg_256:$srsrc, SReg_128:$ssamp,
Nicolai Haehnle59198ed2018-06-04 14:45:20 +0000253 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
254 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
Changpeng Fang4737e892018-01-18 22:08:53 +0000255 asm#" $vdata, $vaddr, $srsrc, $ssamp$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""),
Changpeng Fangb28fe032016-09-01 17:54:54 +0000256 dns>, MIMGe<op> {
257 let WQM = wqm;
Changpeng Fang4737e892018-01-18 22:08:53 +0000258 let D16 = d16;
259}
260
261multiclass MIMG_Sampler_Src_Helper_Helper <bits<7> op, string asm,
262 RegisterClass dst_rc,
263 int channels, bit wqm,
264 bit d16_bit, string suffix> {
265 def _V1 # suffix : MIMG_Sampler_Helper <op, asm, dst_rc, VGPR_32, wqm, d16_bit,
266 !if(!eq(channels, 1), "AMDGPU", "")>,
267 MIMG_Mask<asm#"_V1"#suffix, channels>;
268 def _V2 # suffix : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_64, wqm, d16_bit>,
269 MIMG_Mask<asm#"_V2"#suffix, channels>;
Dmitry Preobrazhenskya0b8cd02018-04-04 13:01:17 +0000270 def _V3 # suffix : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_96, wqm, d16_bit>,
271 MIMG_Mask<asm#"_V3"#suffix, channels>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000272 def _V4 # suffix : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_128, wqm, d16_bit>,
273 MIMG_Mask<asm#"_V4"#suffix, channels>;
274 def _V8 # suffix : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_256, wqm, d16_bit>,
275 MIMG_Mask<asm#"_V8"#suffix, channels>;
276 def _V16 # suffix : MIMG_Sampler_Helper <op, asm, dst_rc, VReg_512, wqm, d16_bit>,
277 MIMG_Mask<asm#"_V16"#suffix, channels>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000278}
279
280multiclass MIMG_Sampler_Src_Helper <bits<7> op, string asm,
281 RegisterClass dst_rc,
Sam Koltonc01faa32016-11-15 13:39:07 +0000282 int channels, bit wqm> {
Nicolai Haehnle01d261f2018-06-04 14:26:05 +0000283 defm "" : MIMG_Sampler_Src_Helper_Helper <op, asm, dst_rc, channels, wqm, 0, "">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000284
285 let d16 = 1 in {
286 let SubtargetPredicate = HasPackedD16VMem in {
Nicolai Haehnle01d261f2018-06-04 14:26:05 +0000287 defm "" : MIMG_Sampler_Src_Helper_Helper <op, asm, dst_rc, channels, wqm, 1, "_D16">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000288 } // End HasPackedD16VMem.
289
290 let SubtargetPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in {
Nicolai Haehnle01d261f2018-06-04 14:26:05 +0000291 defm "" : MIMG_Sampler_Src_Helper_Helper <op, asm, dst_rc, channels, wqm, 1, "_D16_gfx80">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000292 } // End HasUnpackedD16VMem.
293 } // End d16 = 1.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000294}
295
Sam Koltonc01faa32016-11-15 13:39:07 +0000296multiclass MIMG_Sampler <bits<7> op, string asm, bit wqm=0> {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000297 defm _V1 : MIMG_Sampler_Src_Helper<op, asm, VGPR_32, 1, wqm>;
298 defm _V2 : MIMG_Sampler_Src_Helper<op, asm, VReg_64, 2, wqm>;
299 defm _V3 : MIMG_Sampler_Src_Helper<op, asm, VReg_96, 3, wqm>;
300 defm _V4 : MIMG_Sampler_Src_Helper<op, asm, VReg_128, 4, wqm>;
301}
302
303multiclass MIMG_Sampler_WQM <bits<7> op, string asm> : MIMG_Sampler<op, asm, 1>;
304
305class MIMG_Gather_Helper <bits<7> op, string asm,
306 RegisterClass dst_rc,
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000307 RegisterClass src_rc,
308 bit wqm,
309 bit d16_bit=0,
310 string dns=""> : MIMG <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000311 (outs dst_rc:$vdata),
312 (ins src_rc:$vaddr, SReg_256:$srsrc, SReg_128:$ssamp,
Nicolai Haehnle59198ed2018-06-04 14:45:20 +0000313 DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc,
314 R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da),
Changpeng Fang4737e892018-01-18 22:08:53 +0000315 asm#" $vdata, $vaddr, $srsrc, $ssamp$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""),
Changpeng Fangb28fe032016-09-01 17:54:54 +0000316 []>, MIMGe<op> {
317 let mayLoad = 1;
318 let mayStore = 0;
319
320 // DMASK was repurposed for GATHER4. 4 components are always
321 // returned and DMASK works like a swizzle - it selects
322 // the component to fetch. The only useful DMASK values are
323 // 1=red, 2=green, 4=blue, 8=alpha. (e.g. 1 returns
324 // (red,red,red,red) etc.) The ISA document doesn't mention
325 // this.
326 // Therefore, disable all code which updates DMASK by setting this:
327 let Gather4 = 1;
328 let hasPostISelHook = 0;
329 let WQM = wqm;
Changpeng Fang4737e892018-01-18 22:08:53 +0000330 let D16 = d16;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000331
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000332 let DecoderNamespace = dns;
333 let isAsmParserOnly = !if(!eq(dns,""), 1, 0);
Changpeng Fangb28fe032016-09-01 17:54:54 +0000334}
335
Changpeng Fang4737e892018-01-18 22:08:53 +0000336
Changpeng Fangb28fe032016-09-01 17:54:54 +0000337multiclass MIMG_Gather_Src_Helper <bits<7> op, string asm,
338 RegisterClass dst_rc,
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000339 bit wqm, bit d16_bit,
340 string prefix,
341 string suffix> {
342 def prefix # _V1 # suffix : MIMG_Gather_Helper <op, asm, dst_rc, VGPR_32, wqm, d16_bit, "AMDGPU">;
343 def prefix # _V2 # suffix : MIMG_Gather_Helper <op, asm, dst_rc, VReg_64, wqm, d16_bit>;
Dmitry Preobrazhenskya0b8cd02018-04-04 13:01:17 +0000344 def prefix # _V3 # suffix : MIMG_Gather_Helper <op, asm, dst_rc, VReg_96, wqm, d16_bit>;
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000345 def prefix # _V4 # suffix : MIMG_Gather_Helper <op, asm, dst_rc, VReg_128, wqm, d16_bit>;
346 def prefix # _V8 # suffix : MIMG_Gather_Helper <op, asm, dst_rc, VReg_256, wqm, d16_bit>;
347 def prefix # _V16 # suffix : MIMG_Gather_Helper <op, asm, dst_rc, VReg_512, wqm, d16_bit>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000348}
349
Sam Koltonc01faa32016-11-15 13:39:07 +0000350multiclass MIMG_Gather <bits<7> op, string asm, bit wqm=0> {
Nicolai Haehnle01d261f2018-06-04 14:26:05 +0000351 defm "" : MIMG_Gather_Src_Helper<op, asm, VReg_128, wqm, 0, "_V4", "">;
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000352
353 let d16 = 1 in {
354 let AssemblerPredicate = HasPackedD16VMem in {
Nicolai Haehnle01d261f2018-06-04 14:26:05 +0000355 defm "" : MIMG_Gather_Src_Helper<op, asm, VReg_64, wqm, 1, "_V2", "_D16">;
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000356 } // End HasPackedD16VMem.
357
358 let AssemblerPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in {
Nicolai Haehnle01d261f2018-06-04 14:26:05 +0000359 defm "" : MIMG_Gather_Src_Helper<op, asm, VReg_128, wqm, 1, "_V4", "_D16_gfx80">;
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000360 } // End HasUnpackedD16VMem.
361 } // End d16 = 1.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000362}
363
364multiclass MIMG_Gather_WQM <bits<7> op, string asm> : MIMG_Gather<op, asm, 1>;
365
366//===----------------------------------------------------------------------===//
367// MIMG Instructions
368//===----------------------------------------------------------------------===//
369let SubtargetPredicate = isGCN in {
370defm IMAGE_LOAD : MIMG_NoSampler <0x00000000, "image_load">;
371defm IMAGE_LOAD_MIP : MIMG_NoSampler <0x00000001, "image_load_mip">;
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000372defm IMAGE_LOAD_PCK : MIMG_PckNoSampler <0x00000002, "image_load_pck">;
373defm IMAGE_LOAD_PCK_SGN : MIMG_PckNoSampler <0x00000003, "image_load_pck_sgn">;
374defm IMAGE_LOAD_MIP_PCK : MIMG_PckNoSampler <0x00000004, "image_load_mip_pck">;
375defm IMAGE_LOAD_MIP_PCK_SGN : MIMG_PckNoSampler <0x00000005, "image_load_mip_pck_sgn">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000376defm IMAGE_STORE : MIMG_Store <0x00000008, "image_store">;
377defm IMAGE_STORE_MIP : MIMG_Store <0x00000009, "image_store_mip">;
Dmitry Preobrazhensky2456ac62018-03-28 15:44:16 +0000378defm IMAGE_STORE_PCK : MIMG_PckStore <0x0000000a, "image_store_pck">;
379defm IMAGE_STORE_MIP_PCK : MIMG_PckStore <0x0000000b, "image_store_mip_pck">;
Matt Arsenault856777d2017-12-08 20:00:57 +0000380
381let mayLoad = 0, mayStore = 0 in {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000382defm IMAGE_GET_RESINFO : MIMG_NoSampler <0x0000000e, "image_get_resinfo">;
Matt Arsenault856777d2017-12-08 20:00:57 +0000383}
384
Changpeng Fangb28fe032016-09-01 17:54:54 +0000385defm IMAGE_ATOMIC_SWAP : MIMG_Atomic <mimg<0x0f, 0x10>, "image_atomic_swap">;
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000386defm IMAGE_ATOMIC_CMPSWAP : MIMG_Atomic <mimg<0x10, 0x11>, "image_atomic_cmpswap", VReg_64, VReg_128>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000387defm IMAGE_ATOMIC_ADD : MIMG_Atomic <mimg<0x11, 0x12>, "image_atomic_add">;
388defm IMAGE_ATOMIC_SUB : MIMG_Atomic <mimg<0x12, 0x13>, "image_atomic_sub">;
389//def IMAGE_ATOMIC_RSUB : MIMG_NoPattern_ <"image_atomic_rsub", 0x00000013>; -- not on VI
390defm IMAGE_ATOMIC_SMIN : MIMG_Atomic <mimg<0x14>, "image_atomic_smin">;
391defm IMAGE_ATOMIC_UMIN : MIMG_Atomic <mimg<0x15>, "image_atomic_umin">;
392defm IMAGE_ATOMIC_SMAX : MIMG_Atomic <mimg<0x16>, "image_atomic_smax">;
393defm IMAGE_ATOMIC_UMAX : MIMG_Atomic <mimg<0x17>, "image_atomic_umax">;
394defm IMAGE_ATOMIC_AND : MIMG_Atomic <mimg<0x18>, "image_atomic_and">;
395defm IMAGE_ATOMIC_OR : MIMG_Atomic <mimg<0x19>, "image_atomic_or">;
396defm IMAGE_ATOMIC_XOR : MIMG_Atomic <mimg<0x1a>, "image_atomic_xor">;
397defm IMAGE_ATOMIC_INC : MIMG_Atomic <mimg<0x1b>, "image_atomic_inc">;
398defm IMAGE_ATOMIC_DEC : MIMG_Atomic <mimg<0x1c>, "image_atomic_dec">;
399//def IMAGE_ATOMIC_FCMPSWAP : MIMG_NoPattern_ <"image_atomic_fcmpswap", 0x0000001d>; -- not on VI
400//def IMAGE_ATOMIC_FMIN : MIMG_NoPattern_ <"image_atomic_fmin", 0x0000001e>; -- not on VI
401//def IMAGE_ATOMIC_FMAX : MIMG_NoPattern_ <"image_atomic_fmax", 0x0000001f>; -- not on VI
402defm IMAGE_SAMPLE : MIMG_Sampler_WQM <0x00000020, "image_sample">;
403defm IMAGE_SAMPLE_CL : MIMG_Sampler_WQM <0x00000021, "image_sample_cl">;
404defm IMAGE_SAMPLE_D : MIMG_Sampler <0x00000022, "image_sample_d">;
405defm IMAGE_SAMPLE_D_CL : MIMG_Sampler <0x00000023, "image_sample_d_cl">;
406defm IMAGE_SAMPLE_L : MIMG_Sampler <0x00000024, "image_sample_l">;
407defm IMAGE_SAMPLE_B : MIMG_Sampler_WQM <0x00000025, "image_sample_b">;
408defm IMAGE_SAMPLE_B_CL : MIMG_Sampler_WQM <0x00000026, "image_sample_b_cl">;
409defm IMAGE_SAMPLE_LZ : MIMG_Sampler <0x00000027, "image_sample_lz">;
410defm IMAGE_SAMPLE_C : MIMG_Sampler_WQM <0x00000028, "image_sample_c">;
411defm IMAGE_SAMPLE_C_CL : MIMG_Sampler_WQM <0x00000029, "image_sample_c_cl">;
412defm IMAGE_SAMPLE_C_D : MIMG_Sampler <0x0000002a, "image_sample_c_d">;
413defm IMAGE_SAMPLE_C_D_CL : MIMG_Sampler <0x0000002b, "image_sample_c_d_cl">;
414defm IMAGE_SAMPLE_C_L : MIMG_Sampler <0x0000002c, "image_sample_c_l">;
415defm IMAGE_SAMPLE_C_B : MIMG_Sampler_WQM <0x0000002d, "image_sample_c_b">;
416defm IMAGE_SAMPLE_C_B_CL : MIMG_Sampler_WQM <0x0000002e, "image_sample_c_b_cl">;
417defm IMAGE_SAMPLE_C_LZ : MIMG_Sampler <0x0000002f, "image_sample_c_lz">;
418defm IMAGE_SAMPLE_O : MIMG_Sampler_WQM <0x00000030, "image_sample_o">;
419defm IMAGE_SAMPLE_CL_O : MIMG_Sampler_WQM <0x00000031, "image_sample_cl_o">;
420defm IMAGE_SAMPLE_D_O : MIMG_Sampler <0x00000032, "image_sample_d_o">;
421defm IMAGE_SAMPLE_D_CL_O : MIMG_Sampler <0x00000033, "image_sample_d_cl_o">;
422defm IMAGE_SAMPLE_L_O : MIMG_Sampler <0x00000034, "image_sample_l_o">;
423defm IMAGE_SAMPLE_B_O : MIMG_Sampler_WQM <0x00000035, "image_sample_b_o">;
424defm IMAGE_SAMPLE_B_CL_O : MIMG_Sampler_WQM <0x00000036, "image_sample_b_cl_o">;
425defm IMAGE_SAMPLE_LZ_O : MIMG_Sampler <0x00000037, "image_sample_lz_o">;
426defm IMAGE_SAMPLE_C_O : MIMG_Sampler_WQM <0x00000038, "image_sample_c_o">;
427defm IMAGE_SAMPLE_C_CL_O : MIMG_Sampler_WQM <0x00000039, "image_sample_c_cl_o">;
428defm IMAGE_SAMPLE_C_D_O : MIMG_Sampler <0x0000003a, "image_sample_c_d_o">;
429defm IMAGE_SAMPLE_C_D_CL_O : MIMG_Sampler <0x0000003b, "image_sample_c_d_cl_o">;
430defm IMAGE_SAMPLE_C_L_O : MIMG_Sampler <0x0000003c, "image_sample_c_l_o">;
431defm IMAGE_SAMPLE_C_B_O : MIMG_Sampler_WQM <0x0000003d, "image_sample_c_b_o">;
432defm IMAGE_SAMPLE_C_B_CL_O : MIMG_Sampler_WQM <0x0000003e, "image_sample_c_b_cl_o">;
433defm IMAGE_SAMPLE_C_LZ_O : MIMG_Sampler <0x0000003f, "image_sample_c_lz_o">;
434defm IMAGE_GATHER4 : MIMG_Gather_WQM <0x00000040, "image_gather4">;
435defm IMAGE_GATHER4_CL : MIMG_Gather_WQM <0x00000041, "image_gather4_cl">;
436defm IMAGE_GATHER4_L : MIMG_Gather <0x00000044, "image_gather4_l">;
437defm IMAGE_GATHER4_B : MIMG_Gather_WQM <0x00000045, "image_gather4_b">;
438defm IMAGE_GATHER4_B_CL : MIMG_Gather_WQM <0x00000046, "image_gather4_b_cl">;
439defm IMAGE_GATHER4_LZ : MIMG_Gather <0x00000047, "image_gather4_lz">;
440defm IMAGE_GATHER4_C : MIMG_Gather_WQM <0x00000048, "image_gather4_c">;
441defm IMAGE_GATHER4_C_CL : MIMG_Gather_WQM <0x00000049, "image_gather4_c_cl">;
442defm IMAGE_GATHER4_C_L : MIMG_Gather <0x0000004c, "image_gather4_c_l">;
443defm IMAGE_GATHER4_C_B : MIMG_Gather_WQM <0x0000004d, "image_gather4_c_b">;
444defm IMAGE_GATHER4_C_B_CL : MIMG_Gather_WQM <0x0000004e, "image_gather4_c_b_cl">;
445defm IMAGE_GATHER4_C_LZ : MIMG_Gather <0x0000004f, "image_gather4_c_lz">;
446defm IMAGE_GATHER4_O : MIMG_Gather_WQM <0x00000050, "image_gather4_o">;
447defm IMAGE_GATHER4_CL_O : MIMG_Gather_WQM <0x00000051, "image_gather4_cl_o">;
448defm IMAGE_GATHER4_L_O : MIMG_Gather <0x00000054, "image_gather4_l_o">;
449defm IMAGE_GATHER4_B_O : MIMG_Gather_WQM <0x00000055, "image_gather4_b_o">;
450defm IMAGE_GATHER4_B_CL_O : MIMG_Gather <0x00000056, "image_gather4_b_cl_o">;
451defm IMAGE_GATHER4_LZ_O : MIMG_Gather <0x00000057, "image_gather4_lz_o">;
452defm IMAGE_GATHER4_C_O : MIMG_Gather_WQM <0x00000058, "image_gather4_c_o">;
453defm IMAGE_GATHER4_C_CL_O : MIMG_Gather_WQM <0x00000059, "image_gather4_c_cl_o">;
454defm IMAGE_GATHER4_C_L_O : MIMG_Gather <0x0000005c, "image_gather4_c_l_o">;
455defm IMAGE_GATHER4_C_B_O : MIMG_Gather_WQM <0x0000005d, "image_gather4_c_b_o">;
456defm IMAGE_GATHER4_C_B_CL_O : MIMG_Gather_WQM <0x0000005e, "image_gather4_c_b_cl_o">;
457defm IMAGE_GATHER4_C_LZ_O : MIMG_Gather <0x0000005f, "image_gather4_c_lz_o">;
Matt Arsenault856777d2017-12-08 20:00:57 +0000458
459let mayLoad = 0, mayStore = 0 in {
Changpeng Fangb28fe032016-09-01 17:54:54 +0000460defm IMAGE_GET_LOD : MIMG_Sampler_WQM <0x00000060, "image_get_lod">;
Matt Arsenault856777d2017-12-08 20:00:57 +0000461}
462
Changpeng Fangb28fe032016-09-01 17:54:54 +0000463defm IMAGE_SAMPLE_CD : MIMG_Sampler <0x00000068, "image_sample_cd">;
464defm IMAGE_SAMPLE_CD_CL : MIMG_Sampler <0x00000069, "image_sample_cd_cl">;
465defm IMAGE_SAMPLE_C_CD : MIMG_Sampler <0x0000006a, "image_sample_c_cd">;
466defm IMAGE_SAMPLE_C_CD_CL : MIMG_Sampler <0x0000006b, "image_sample_c_cd_cl">;
467defm IMAGE_SAMPLE_CD_O : MIMG_Sampler <0x0000006c, "image_sample_cd_o">;
468defm IMAGE_SAMPLE_CD_CL_O : MIMG_Sampler <0x0000006d, "image_sample_cd_cl_o">;
469defm IMAGE_SAMPLE_C_CD_O : MIMG_Sampler <0x0000006e, "image_sample_c_cd_o">;
470defm IMAGE_SAMPLE_C_CD_CL_O : MIMG_Sampler <0x0000006f, "image_sample_c_cd_cl_o">;
471//def IMAGE_RSRC256 : MIMG_NoPattern_RSRC256 <"image_rsrc256", 0x0000007e>;
472//def IMAGE_SAMPLER : MIMG_NoPattern_ <"image_sampler", 0x0000007f>;
473}
474
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000475/********** ============================== **********/
476/********** Dimension-aware image patterns **********/
477/********** ============================== **********/
478
479class getDwordsType<int dwords> {
480 int NumDwords = dwords;
481 string suffix = !if(!lt(dwords, 1), ?,
482 !if(!eq(dwords, 1), "_V1",
483 !if(!eq(dwords, 2), "_V2",
484 !if(!le(dwords, 4), "_V4",
485 !if(!le(dwords, 8), "_V8",
486 !if(!le(dwords, 16), "_V16", ?))))));
487 ValueType VT = !if(!lt(dwords, 1), ?,
488 !if(!eq(dwords, 1), f32,
489 !if(!eq(dwords, 2), v2f32,
490 !if(!le(dwords, 4), v4f32,
491 !if(!le(dwords, 8), v8f32,
492 !if(!le(dwords, 16), v16f32, ?))))));
493 RegisterClass VReg = !if(!lt(dwords, 1), ?,
494 !if(!eq(dwords, 1), VGPR_32,
495 !if(!eq(dwords, 2), VReg_64,
496 !if(!le(dwords, 4), VReg_128,
497 !if(!le(dwords, 8), VReg_256,
498 !if(!le(dwords, 16), VReg_512, ?))))));
499}
500
501class makeRegSequence_Fold<int i, dag d> {
502 int idx = i;
503 dag lhs = d;
504}
505
506// Generate a dag node which returns a vector register of class RC into which
507// the source operands given by names have been inserted (assuming that each
508// name corresponds to an operand whose size is equal to a subregister).
509class makeRegSequence<ValueType vt, RegisterClass RC, list<string> names> {
510 dag ret =
511 !if(!eq(!size(names), 1),
512 !dag(COPY_TO_REGCLASS, [?, RC], [names[0], ?]),
513 !foldl(makeRegSequence_Fold<0, (vt (IMPLICIT_DEF))>, names, f, name,
514 makeRegSequence_Fold<
515 !add(f.idx, 1),
516 !con((INSERT_SUBREG f.lhs),
517 !dag(INSERT_SUBREG, [?, !cast<SubRegIndex>("sub"#f.idx)],
518 [name, ?]))>).lhs);
519}
520
521class ImageDimPattern<AMDGPUImageDimIntrinsic I,
522 string dop, ValueType dty,
523 string suffix = ""> : GCNPat<(undef), (undef)> {
524 list<AMDGPUArg> AddrArgs = I.P.AddrDefaultArgs;
525 getDwordsType AddrDwords = getDwordsType<!size(AddrArgs)>;
526
527 Instruction MI =
528 !cast<Instruction>(!strconcat("IMAGE_", I.P.OpMod, dop, AddrDwords.suffix, suffix));
529
530 // DAG fragment to match data arguments (vdata for store/atomic, dmask
531 // for non-atomic).
532 dag MatchDataDag =
533 !con(!dag(I, !foreach(arg, I.P.DataArgs, dty),
534 !foreach(arg, I.P.DataArgs, arg.Name)),
535 !if(I.P.IsAtomic, (I), (I i32:$dmask)));
536
537 // DAG fragment to match vaddr arguments.
538 dag MatchAddrDag = !dag(I, !foreach(arg, AddrArgs, arg.Type.VT),
539 !foreach(arg, AddrArgs, arg.Name));
540
541 // DAG fragment to match sampler resource and unorm arguments.
542 dag MatchSamplerDag = !if(I.P.IsSample, (I v4i32:$sampler, i1:$unorm), (I));
543
544 // DAG node that generates the MI vdata for store/atomic
545 getDwordsType DataDwords = getDwordsType<!size(I.P.DataArgs)>;
546 dag GenDataDag =
547 !if(I.P.IsAtomic, (MI makeRegSequence<DataDwords.VT, DataDwords.VReg,
548 !foreach(arg, I.P.DataArgs, arg.Name)>.ret),
549 !if(!size(I.P.DataArgs), (MI $vdata), (MI)));
550
551 // DAG node that generates the MI vaddr
552 dag GenAddrDag = makeRegSequence<AddrDwords.VT, AddrDwords.VReg,
553 !foreach(arg, AddrArgs, arg.Name)>.ret;
554 // DAG fragments that generate various inline flags
555 dag GenDmask =
556 !if(I.P.IsAtomic, (MI !add(!shl(1, DataDwords.NumDwords), -1)),
557 (MI (as_i32imm $dmask)));
558 dag GenGLC =
559 !if(I.P.IsAtomic, (MI 1),
560 (MI (bitextract_imm<0> $cachepolicy)));
561
562 dag MatchIntrinsic = !con(MatchDataDag,
563 MatchAddrDag,
564 (I v8i32:$rsrc),
565 MatchSamplerDag,
566 (I 0/*texfailctrl*/,
567 i32:$cachepolicy));
568 let PatternToMatch =
569 !if(!size(I.RetTypes), (dty MatchIntrinsic), MatchIntrinsic);
570
571 bit IsCmpSwap = !and(I.P.IsAtomic, !eq(!size(I.P.DataArgs), 2));
572 dag ImageInstruction =
573 !con(GenDataDag,
574 (MI GenAddrDag),
575 (MI $rsrc),
576 !if(I.P.IsSample, (MI $sampler), (MI)),
577 GenDmask,
578 !if(I.P.IsSample, (MI (as_i1imm $unorm)), (MI 1)),
579 GenGLC,
580 (MI (bitextract_imm<1> $cachepolicy),
581 0, /* r128 */
582 0, /* tfe */
583 0 /*(as_i1imm $lwe)*/,
584 { I.P.Dim.DA }));
585 let ResultInstrs = [
586 !if(IsCmpSwap, (EXTRACT_SUBREG ImageInstruction, sub0), ImageInstruction)
587 ];
588}
589
590foreach intr = !listconcat(AMDGPUImageDimIntrinsics,
591 AMDGPUImageDimGetResInfoIntrinsics) in {
592 def intr#_pat_v1 : ImageDimPattern<intr, "_V1", f32>;
593 def intr#_pat_v2 : ImageDimPattern<intr, "_V2", v2f32>;
594 def intr#_pat_v4 : ImageDimPattern<intr, "_V4", v4f32>;
595}
596
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000597multiclass ImageDimD16Helper<AMDGPUImageDimIntrinsic I,
598 AMDGPUImageDimIntrinsic d16helper> {
599 let SubtargetPredicate = HasUnpackedD16VMem in {
600 def _unpacked_v1 : ImageDimPattern<I, "_V1", f16, "_D16_gfx80">;
601 def _unpacked_v2 : ImageDimPattern<d16helper, "_V2", v2i32, "_D16_gfx80">;
602 def _unpacked_v4 : ImageDimPattern<d16helper, "_V4", v4i32, "_D16_gfx80">;
603 } // End HasUnpackedD16VMem.
604
605 let SubtargetPredicate = HasPackedD16VMem in {
606 def _packed_v1 : ImageDimPattern<I, "_V1", f16, "_D16">;
Matt Arsenault1349a042018-05-22 06:32:10 +0000607 def _packed_v2 : ImageDimPattern<I, "_V1", v2f16, "_D16">;
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000608 def _packed_v4 : ImageDimPattern<I, "_V2", v4f16, "_D16">;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000609 } // End HasPackedD16VMem.
610}
611
612foreach intr = AMDGPUImageDimIntrinsics in {
613 def intr#_d16helper_profile : AMDGPUDimProfileCopy<intr.P> {
614 let RetTypes = !foreach(ty, intr.P.RetTypes, llvm_any_ty);
615 let DataArgs = !foreach(arg, intr.P.DataArgs, AMDGPUArg<llvm_any_ty, arg.Name>);
616 }
617
618 let TargetPrefix = "SI", isTarget = 1 in
619 def int_SI_image_d16helper_ # intr.P.OpMod # intr.P.Dim.Name :
620 AMDGPUImageDimIntrinsic<!cast<AMDGPUDimProfile>(intr#"_d16helper_profile"),
621 intr.IntrProperties, intr.Properties>;
622
623 defm intr#_d16 :
624 ImageDimD16Helper<
625 intr, !cast<AMDGPUImageDimIntrinsic>(
626 "int_SI_image_d16helper_" # intr.P.OpMod # intr.P.Dim.Name)>;
627}
628
629foreach intr = AMDGPUImageDimGatherIntrinsics in {
630 def intr#_pat3 : ImageDimPattern<intr, "_V4", v4f32>;
631
632 def intr#_d16helper_profile : AMDGPUDimProfileCopy<intr.P> {
633 let RetTypes = !foreach(ty, intr.P.RetTypes, llvm_any_ty);
634 let DataArgs = !foreach(arg, intr.P.DataArgs, AMDGPUArg<llvm_any_ty, arg.Name>);
635 }
636
637 let TargetPrefix = "SI", isTarget = 1 in
638 def int_SI_image_d16helper_ # intr.P.OpMod # intr.P.Dim.Name :
639 AMDGPUImageDimIntrinsic<!cast<AMDGPUDimProfile>(intr#"_d16helper_profile"),
640 intr.IntrProperties, intr.Properties>;
641
642 let SubtargetPredicate = HasUnpackedD16VMem in {
643 def intr#_unpacked_v4 :
644 ImageDimPattern<!cast<AMDGPUImageDimIntrinsic>(
645 "int_SI_image_d16helper_" # intr.P.OpMod # intr.P.Dim.Name),
646 "_V4", v4i32, "_D16_gfx80">;
647 } // End HasUnpackedD16VMem.
648
649 let SubtargetPredicate = HasPackedD16VMem in {
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000650 def intr#_packed_v4 : ImageDimPattern<intr, "_V2", v4f16, "_D16">;
Nicolai Haehnle2f5a7382018-04-04 10:58:54 +0000651 } // End HasPackedD16VMem.
652}
653
654foreach intr = AMDGPUImageDimAtomicIntrinsics in {
655 def intr#_pat1 : ImageDimPattern<intr, "_V1", i32>;
656}
657
Changpeng Fangb28fe032016-09-01 17:54:54 +0000658/********** ======================= **********/
659/********** Image sampling patterns **********/
660/********** ======================= **********/
661
Changpeng Fang4737e892018-01-18 22:08:53 +0000662// ImageSample for amdgcn
Changpeng Fangb28fe032016-09-01 17:54:54 +0000663// TODO:
Changpeng Fang4737e892018-01-18 22:08:53 +0000664// 1. Handle v4i32 rsrc type (Register Class for the instruction to be SReg_128).
665// 2. Add A16 support when we pass address of half type.
666multiclass ImageSamplePattern<SDPatternOperator name, MIMG opcode, ValueType dt, ValueType vt> {
Matt Arsenault90c75932017-10-03 00:06:41 +0000667 def : GCNPat<
Changpeng Fang8236fe12016-11-14 18:33:18 +0000668 (dt (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, i32:$dmask, i1:$unorm, i1:$glc,
Changpeng Fangb28fe032016-09-01 17:54:54 +0000669 i1:$slc, i1:$lwe, i1:$da)),
670 (opcode $addr, $rsrc, $sampler,
671 (as_i32imm $dmask), (as_i1imm $unorm), (as_i1imm $glc), (as_i1imm $slc),
672 0, 0, (as_i1imm $lwe), (as_i1imm $da))
673 >;
674}
675
Changpeng Fang4737e892018-01-18 22:08:53 +0000676multiclass ImageSampleDataPatterns<SDPatternOperator name, string opcode, ValueType dt, string suffix = ""> {
677 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V1 # suffix), dt, f32>;
678 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V2 # suffix), dt, v2f32>;
679 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V4 # suffix), dt, v4f32>;
680 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V8 # suffix), dt, v8f32>;
681 defm : ImageSamplePattern<name, !cast<MIMG>(opcode # _V16 # suffix), dt, v16f32>;
Changpeng Fang8236fe12016-11-14 18:33:18 +0000682}
683
Changpeng Fang4737e892018-01-18 22:08:53 +0000684// ImageSample patterns.
685multiclass ImageSamplePatterns<SDPatternOperator name, string opcode> {
686 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), f32>;
687 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v2f32>;
688 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4f32>;
689
690 let SubtargetPredicate = HasUnpackedD16VMem in {
691 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), f16, "_D16_gfx80">;
692 } // End HasUnpackedD16VMem.
693
694 let SubtargetPredicate = HasPackedD16VMem in {
695 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), f16, "_D16">;
696 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), v2f16, "_D16">;
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000697 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v4f16, "_D16">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000698 } // End HasPackedD16VMem.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000699}
700
Changpeng Fang4737e892018-01-18 22:08:53 +0000701// ImageSample alternative patterns for illegal vector half Types.
702multiclass ImageSampleAltPatterns<SDPatternOperator name, string opcode> {
703 let SubtargetPredicate = HasUnpackedD16VMem in {
704 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v2i32, "_D16_gfx80">;
705 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4i32, "_D16_gfx80">;
706 } // End HasUnpackedD16VMem.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000707}
708
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000709// ImageGather4 patterns.
710multiclass ImageGather4Patterns<SDPatternOperator name, string opcode> {
711 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4f32>;
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000712
713 let SubtargetPredicate = HasPackedD16VMem in {
714 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v4f16, "_D16">;
715 } // End HasPackedD16VMem.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000716}
717
718// ImageGather4 alternative patterns for illegal vector half Types.
719multiclass ImageGather4AltPatterns<SDPatternOperator name, string opcode> {
720 let SubtargetPredicate = HasUnpackedD16VMem in {
721 defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4i32, "_D16_gfx80">;
722 } // End HasUnpackedD16VMem.
723
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000724}
725
Changpeng Fang4737e892018-01-18 22:08:53 +0000726// ImageLoad for amdgcn.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000727multiclass ImageLoadPattern<SDPatternOperator name, MIMG opcode, ValueType dt, ValueType vt> {
Matt Arsenault90c75932017-10-03 00:06:41 +0000728 def : GCNPat <
Changpeng Fang8236fe12016-11-14 18:33:18 +0000729 (dt (name vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc, i1:$lwe,
Tom Stellardfac248c2016-10-12 16:35:29 +0000730 i1:$da)),
731 (opcode $addr, $rsrc,
Changpeng Fangb28fe032016-09-01 17:54:54 +0000732 (as_i32imm $dmask), 1, (as_i1imm $glc), (as_i1imm $slc),
Tom Stellardfac248c2016-10-12 16:35:29 +0000733 0, 0, (as_i1imm $lwe), (as_i1imm $da))
734 >;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000735}
736
Changpeng Fang4737e892018-01-18 22:08:53 +0000737multiclass ImageLoadDataPatterns<SDPatternOperator name, string opcode, ValueType dt, string suffix = ""> {
738 defm : ImageLoadPattern<name, !cast<MIMG>(opcode # _V1 # suffix), dt, i32>;
739 defm : ImageLoadPattern<name, !cast<MIMG>(opcode # _V2 # suffix), dt, v2i32>;
740 defm : ImageLoadPattern<name, !cast<MIMG>(opcode # _V4 # suffix), dt, v4i32>;
Tom Stellardfac248c2016-10-12 16:35:29 +0000741}
742
Changpeng Fang4737e892018-01-18 22:08:53 +0000743// ImageLoad patterns.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000744// TODO: support v3f32.
745multiclass ImageLoadPatterns<SDPatternOperator name, string opcode> {
746 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), f32>;
747 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V2), v2f32>;
748 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V4), v4f32>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000749
750 let SubtargetPredicate = HasUnpackedD16VMem in {
751 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), f16, "_D16_gfx80">;
752 } // End HasUnpackedD16VMem.
753
754 let SubtargetPredicate = HasPackedD16VMem in {
755 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), f16, "_D16">;
756 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), v2f16, "_D16">;
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000757 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V2), v4f16, "_D16">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000758 } // End HasPackedD16VMem.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000759}
760
Changpeng Fang4737e892018-01-18 22:08:53 +0000761// ImageLoad alternative patterns for illegal vector half Types.
762multiclass ImageLoadAltPatterns<SDPatternOperator name, string opcode> {
763 let SubtargetPredicate = HasUnpackedD16VMem in {
764 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V2), v2i32, "_D16_gfx80">;
765 defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V4), v4i32, "_D16_gfx80">;
766 } // End HasUnPackedD16VMem.
Changpeng Fang4737e892018-01-18 22:08:53 +0000767}
768
769// ImageStore for amdgcn.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000770multiclass ImageStorePattern<SDPatternOperator name, MIMG opcode, ValueType dt, ValueType vt> {
Matt Arsenault90c75932017-10-03 00:06:41 +0000771 def : GCNPat <
Changpeng Fang8236fe12016-11-14 18:33:18 +0000772 (name dt:$data, vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc,
Tom Stellardfac248c2016-10-12 16:35:29 +0000773 i1:$lwe, i1:$da),
774 (opcode $data, $addr, $rsrc,
Changpeng Fangb28fe032016-09-01 17:54:54 +0000775 (as_i32imm $dmask), 1, (as_i1imm $glc), (as_i1imm $slc),
Tom Stellardfac248c2016-10-12 16:35:29 +0000776 0, 0, (as_i1imm $lwe), (as_i1imm $da))
777 >;
778}
Changpeng Fangb28fe032016-09-01 17:54:54 +0000779
Changpeng Fang4737e892018-01-18 22:08:53 +0000780multiclass ImageStoreDataPatterns<SDPatternOperator name, string opcode, ValueType dt, string suffix = ""> {
781 defm : ImageStorePattern<name, !cast<MIMG>(opcode # _V1 # suffix), dt, i32>;
782 defm : ImageStorePattern<name, !cast<MIMG>(opcode # _V2 # suffix), dt, v2i32>;
783 defm : ImageStorePattern<name, !cast<MIMG>(opcode # _V4 # suffix), dt, v4i32>;
Changpeng Fang8236fe12016-11-14 18:33:18 +0000784}
785
Changpeng Fang4737e892018-01-18 22:08:53 +0000786// ImageStore patterns.
Changpeng Fang8236fe12016-11-14 18:33:18 +0000787// TODO: support v3f32.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000788multiclass ImageStorePatterns<SDPatternOperator name, string opcode> {
Changpeng Fang8236fe12016-11-14 18:33:18 +0000789 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), f32>;
790 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2f32>;
791 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V4), v4f32>;
Changpeng Fang4737e892018-01-18 22:08:53 +0000792
793 let SubtargetPredicate = HasUnpackedD16VMem in {
794 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), f16, "_D16_gfx80">;
795 } // End HasUnpackedD16VMem.
796
797 let SubtargetPredicate = HasPackedD16VMem in {
798 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), f16, "_D16">;
799 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), v2f16, "_D16">;
Matt Arsenault02dc7e12018-06-15 15:15:46 +0000800 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v4f16, "_D16">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000801 } // End HasPackedD16VMem.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000802}
803
Changpeng Fang4737e892018-01-18 22:08:53 +0000804// ImageStore alternative patterns.
805multiclass ImageStoreAltPatterns<SDPatternOperator name, string opcode> {
806 let SubtargetPredicate = HasUnpackedD16VMem in {
807 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2i32, "_D16_gfx80">;
808 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V4), v4i32, "_D16_gfx80">;
809 } // End HasUnpackedD16VMem.
810
811 let SubtargetPredicate = HasPackedD16VMem in {
812 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), i32, "_D16">;
813 defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2i32, "_D16">;
814 } // End HasPackedD16VMem.
815}
816
817// ImageAtomic for amdgcn.
Matt Arsenault90c75932017-10-03 00:06:41 +0000818class ImageAtomicPattern<SDPatternOperator name, MIMG opcode, ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000819 (name i32:$vdata, vt:$addr, v8i32:$rsrc, imm:$r128, imm:$da, imm:$slc),
820 (opcode $vdata, $addr, $rsrc, 1, 1, 1, (as_i1imm $slc), (as_i1imm $r128), 0, 0, (as_i1imm $da))
821>;
822
Changpeng Fang4737e892018-01-18 22:08:53 +0000823// ImageAtomic patterns.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000824multiclass ImageAtomicPatterns<SDPatternOperator name, string opcode> {
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +0000825 def : ImageAtomicPattern<name, !cast<MIMG>(opcode # _V1_V1), i32>;
826 def : ImageAtomicPattern<name, !cast<MIMG>(opcode # _V1_V2), v2i32>;
827 def : ImageAtomicPattern<name, !cast<MIMG>(opcode # _V1_V4), v4i32>;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000828}
829
Changpeng Fang4737e892018-01-18 22:08:53 +0000830// ImageAtomicCmpSwap for amdgcn.
Matt Arsenault90c75932017-10-03 00:06:41 +0000831class ImageAtomicCmpSwapPattern<MIMG opcode, ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +0000832 (int_amdgcn_image_atomic_cmpswap i32:$vsrc, i32:$vcmp, vt:$addr, v8i32:$rsrc,
833 imm:$r128, imm:$da, imm:$slc),
834 (EXTRACT_SUBREG
835 (opcode (REG_SEQUENCE VReg_64, $vsrc, sub0, $vcmp, sub1),
836 $addr, $rsrc, 3, 1, 1, (as_i1imm $slc), (as_i1imm $r128), 0, 0, (as_i1imm $da)),
837 sub0)
838>;
839
Changpeng Fangb28fe032016-09-01 17:54:54 +0000840// ======= amdgcn Image Intrinsics ==============
841
Changpeng Fang4737e892018-01-18 22:08:53 +0000842// Image load.
Changpeng Fangb28fe032016-09-01 17:54:54 +0000843defm : ImageLoadPatterns<int_amdgcn_image_load, "IMAGE_LOAD">;
844defm : ImageLoadPatterns<int_amdgcn_image_load_mip, "IMAGE_LOAD_MIP">;
Changpeng Fang8236fe12016-11-14 18:33:18 +0000845defm : ImageLoadPatterns<int_amdgcn_image_getresinfo, "IMAGE_GET_RESINFO">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000846defm : ImageLoadAltPatterns<SIImage_load, "IMAGE_LOAD">;
847defm : ImageLoadAltPatterns<SIImage_load_mip, "IMAGE_LOAD_MIP">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000848
Changpeng Fang4737e892018-01-18 22:08:53 +0000849// Image store.
Matt Arsenault1349a042018-05-22 06:32:10 +0000850defm : ImageStorePatterns<int_amdgcn_image_store, "IMAGE_STORE">;
851defm : ImageStorePatterns<int_amdgcn_image_store_mip, "IMAGE_STORE_MIP">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000852defm : ImageStoreAltPatterns<SIImage_store, "IMAGE_STORE">;
853defm : ImageStoreAltPatterns<SIImage_store_mip, "IMAGE_STORE_MIP">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000854
Changpeng Fang4737e892018-01-18 22:08:53 +0000855// Basic sample.
856defm : ImageSamplePatterns<int_amdgcn_image_sample, "IMAGE_SAMPLE">;
857defm : ImageSamplePatterns<int_amdgcn_image_sample_cl, "IMAGE_SAMPLE_CL">;
858defm : ImageSamplePatterns<int_amdgcn_image_sample_d, "IMAGE_SAMPLE_D">;
859defm : ImageSamplePatterns<int_amdgcn_image_sample_d_cl, "IMAGE_SAMPLE_D_CL">;
860defm : ImageSamplePatterns<int_amdgcn_image_sample_l, "IMAGE_SAMPLE_L">;
861defm : ImageSamplePatterns<int_amdgcn_image_sample_b, "IMAGE_SAMPLE_B">;
862defm : ImageSamplePatterns<int_amdgcn_image_sample_b_cl, "IMAGE_SAMPLE_B_CL">;
863defm : ImageSamplePatterns<int_amdgcn_image_sample_lz, "IMAGE_SAMPLE_LZ">;
864defm : ImageSamplePatterns<int_amdgcn_image_sample_cd, "IMAGE_SAMPLE_CD">;
865defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_cl, "IMAGE_SAMPLE_CD_CL">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000866
Changpeng Fang4737e892018-01-18 22:08:53 +0000867// Sample with comparison.
868defm : ImageSamplePatterns<int_amdgcn_image_sample_c, "IMAGE_SAMPLE_C">;
869defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cl, "IMAGE_SAMPLE_C_CL">;
870defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d, "IMAGE_SAMPLE_C_D">;
871defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_cl, "IMAGE_SAMPLE_C_D_CL">;
872defm : ImageSamplePatterns<int_amdgcn_image_sample_c_l, "IMAGE_SAMPLE_C_L">;
873defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b, "IMAGE_SAMPLE_C_B">;
874defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_cl, "IMAGE_SAMPLE_C_B_CL">;
875defm : ImageSamplePatterns<int_amdgcn_image_sample_c_lz, "IMAGE_SAMPLE_C_LZ">;
876defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd, "IMAGE_SAMPLE_C_CD">;
877defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_cl, "IMAGE_SAMPLE_C_CD_CL">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000878
Changpeng Fang4737e892018-01-18 22:08:53 +0000879// Sample with offsets.
880defm : ImageSamplePatterns<int_amdgcn_image_sample_o, "IMAGE_SAMPLE_O">;
881defm : ImageSamplePatterns<int_amdgcn_image_sample_cl_o, "IMAGE_SAMPLE_CL_O">;
882defm : ImageSamplePatterns<int_amdgcn_image_sample_d_o, "IMAGE_SAMPLE_D_O">;
883defm : ImageSamplePatterns<int_amdgcn_image_sample_d_cl_o, "IMAGE_SAMPLE_D_CL_O">;
884defm : ImageSamplePatterns<int_amdgcn_image_sample_l_o, "IMAGE_SAMPLE_L_O">;
885defm : ImageSamplePatterns<int_amdgcn_image_sample_b_o, "IMAGE_SAMPLE_B_O">;
886defm : ImageSamplePatterns<int_amdgcn_image_sample_b_cl_o, "IMAGE_SAMPLE_B_CL_O">;
887defm : ImageSamplePatterns<int_amdgcn_image_sample_lz_o, "IMAGE_SAMPLE_LZ_O">;
888defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_o, "IMAGE_SAMPLE_CD_O">;
889defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_cl_o, "IMAGE_SAMPLE_CD_CL_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000890
Changpeng Fang4737e892018-01-18 22:08:53 +0000891// Sample with comparison and offsets.
892defm : ImageSamplePatterns<int_amdgcn_image_sample_c_o, "IMAGE_SAMPLE_C_O">;
893defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cl_o, "IMAGE_SAMPLE_C_CL_O">;
894defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_o, "IMAGE_SAMPLE_C_D_O">;
895defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_cl_o, "IMAGE_SAMPLE_C_D_CL_O">;
896defm : ImageSamplePatterns<int_amdgcn_image_sample_c_l_o, "IMAGE_SAMPLE_C_L_O">;
897defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_o, "IMAGE_SAMPLE_C_B_O">;
898defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_cl_o, "IMAGE_SAMPLE_C_B_CL_O">;
899defm : ImageSamplePatterns<int_amdgcn_image_sample_c_lz_o, "IMAGE_SAMPLE_C_LZ_O">;
900defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_o, "IMAGE_SAMPLE_C_CD_O">;
901defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_cl_o, "IMAGE_SAMPLE_C_CD_CL_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000902
Changpeng Fang4737e892018-01-18 22:08:53 +0000903// Basic gather4.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000904defm : ImageGather4Patterns<int_amdgcn_image_gather4, "IMAGE_GATHER4">;
905defm : ImageGather4Patterns<int_amdgcn_image_gather4_cl, "IMAGE_GATHER4_CL">;
906defm : ImageGather4Patterns<int_amdgcn_image_gather4_l, "IMAGE_GATHER4_L">;
907defm : ImageGather4Patterns<int_amdgcn_image_gather4_b, "IMAGE_GATHER4_B">;
908defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_cl, "IMAGE_GATHER4_B_CL">;
909defm : ImageGather4Patterns<int_amdgcn_image_gather4_lz, "IMAGE_GATHER4_LZ">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000910
Changpeng Fang4737e892018-01-18 22:08:53 +0000911// Gather4 with comparison.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000912defm : ImageGather4Patterns<int_amdgcn_image_gather4_c, "IMAGE_GATHER4_C">;
913defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_cl, "IMAGE_GATHER4_C_CL">;
914defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_l, "IMAGE_GATHER4_C_L">;
915defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b, "IMAGE_GATHER4_C_B">;
916defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_cl, "IMAGE_GATHER4_C_B_CL">;
917defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_lz, "IMAGE_GATHER4_C_LZ">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000918
Changpeng Fang4737e892018-01-18 22:08:53 +0000919// Gather4 with offsets.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000920defm : ImageGather4Patterns<int_amdgcn_image_gather4_o, "IMAGE_GATHER4_O">;
921defm : ImageGather4Patterns<int_amdgcn_image_gather4_cl_o, "IMAGE_GATHER4_CL_O">;
922defm : ImageGather4Patterns<int_amdgcn_image_gather4_l_o, "IMAGE_GATHER4_L_O">;
923defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_o, "IMAGE_GATHER4_B_O">;
924defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_cl_o, "IMAGE_GATHER4_B_CL_O">;
925defm : ImageGather4Patterns<int_amdgcn_image_gather4_lz_o, "IMAGE_GATHER4_LZ_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000926
Changpeng Fang4737e892018-01-18 22:08:53 +0000927// Gather4 with comparison and offsets.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000928defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_o, "IMAGE_GATHER4_C_O">;
929defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_cl_o, "IMAGE_GATHER4_C_CL_O">;
930defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_l_o, "IMAGE_GATHER4_C_L_O">;
931defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_o, "IMAGE_GATHER4_C_B_O">;
932defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_cl_o, "IMAGE_GATHER4_C_B_CL_O">;
933defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_lz_o, "IMAGE_GATHER4_C_LZ_O">;
Changpeng Fangb28fe032016-09-01 17:54:54 +0000934
Changpeng Fang4737e892018-01-18 22:08:53 +0000935// Basic sample alternative.
936defm : ImageSampleAltPatterns<SIImage_sample, "IMAGE_SAMPLE">;
937defm : ImageSampleAltPatterns<SIImage_sample_cl, "IMAGE_SAMPLE_CL">;
938defm : ImageSampleAltPatterns<SIImage_sample_d, "IMAGE_SAMPLE_D">;
939defm : ImageSampleAltPatterns<SIImage_sample_d_cl, "IMAGE_SAMPLE_D_CL">;
940defm : ImageSampleAltPatterns<SIImage_sample_l, "IMAGE_SAMPLE_L">;
941defm : ImageSampleAltPatterns<SIImage_sample_b, "IMAGE_SAMPLE_B">;
942defm : ImageSampleAltPatterns<SIImage_sample_b_cl, "IMAGE_SAMPLE_B_CL">;
943defm : ImageSampleAltPatterns<SIImage_sample_lz, "IMAGE_SAMPLE_LZ">;
944defm : ImageSampleAltPatterns<SIImage_sample_cd, "IMAGE_SAMPLE_CD">;
945defm : ImageSampleAltPatterns<SIImage_sample_cd_cl, "IMAGE_SAMPLE_CD_CL">;
946
947// Sample with comparison alternative.
948defm : ImageSampleAltPatterns<SIImage_sample_c, "IMAGE_SAMPLE_C">;
949defm : ImageSampleAltPatterns<SIImage_sample_c_cl, "IMAGE_SAMPLE_C_CL">;
950defm : ImageSampleAltPatterns<SIImage_sample_c_d, "IMAGE_SAMPLE_C_D">;
951defm : ImageSampleAltPatterns<SIImage_sample_c_d_cl, "IMAGE_SAMPLE_C_D_CL">;
952defm : ImageSampleAltPatterns<SIImage_sample_c_l, "IMAGE_SAMPLE_C_L">;
953defm : ImageSampleAltPatterns<SIImage_sample_c_b, "IMAGE_SAMPLE_C_B">;
954defm : ImageSampleAltPatterns<SIImage_sample_c_b_cl, "IMAGE_SAMPLE_C_B_CL">;
955defm : ImageSampleAltPatterns<SIImage_sample_c_lz, "IMAGE_SAMPLE_C_LZ">;
956defm : ImageSampleAltPatterns<SIImage_sample_c_cd, "IMAGE_SAMPLE_C_CD">;
957defm : ImageSampleAltPatterns<SIImage_sample_c_cd_cl, "IMAGE_SAMPLE_C_CD_CL">;
958
959// Sample with offsets alternative.
960defm : ImageSampleAltPatterns<SIImage_sample_o, "IMAGE_SAMPLE_O">;
961defm : ImageSampleAltPatterns<SIImage_sample_cl_o, "IMAGE_SAMPLE_CL_O">;
962defm : ImageSampleAltPatterns<SIImage_sample_d_o, "IMAGE_SAMPLE_D_O">;
963defm : ImageSampleAltPatterns<SIImage_sample_d_cl_o, "IMAGE_SAMPLE_D_CL_O">;
964defm : ImageSampleAltPatterns<SIImage_sample_l_o, "IMAGE_SAMPLE_L_O">;
965defm : ImageSampleAltPatterns<SIImage_sample_b_o, "IMAGE_SAMPLE_B_O">;
966defm : ImageSampleAltPatterns<SIImage_sample_b_cl_o, "IMAGE_SAMPLE_B_CL_O">;
967defm : ImageSampleAltPatterns<SIImage_sample_lz_o, "IMAGE_SAMPLE_LZ_O">;
968defm : ImageSampleAltPatterns<SIImage_sample_cd_o, "IMAGE_SAMPLE_CD_O">;
969defm : ImageSampleAltPatterns<SIImage_sample_cd_cl_o, "IMAGE_SAMPLE_CD_CL_O">;
970
971// Sample with comparison and offsets alternative.
972defm : ImageSampleAltPatterns<SIImage_sample_c_o, "IMAGE_SAMPLE_C_O">;
973defm : ImageSampleAltPatterns<SIImage_sample_c_cl_o, "IMAGE_SAMPLE_C_CL_O">;
974defm : ImageSampleAltPatterns<SIImage_sample_c_d_o, "IMAGE_SAMPLE_C_D_O">;
975defm : ImageSampleAltPatterns<SIImage_sample_c_d_cl_o, "IMAGE_SAMPLE_C_D_CL_O">;
976defm : ImageSampleAltPatterns<SIImage_sample_c_l_o, "IMAGE_SAMPLE_C_L_O">;
977defm : ImageSampleAltPatterns<SIImage_sample_c_b_o, "IMAGE_SAMPLE_C_B_O">;
978defm : ImageSampleAltPatterns<SIImage_sample_c_b_cl_o, "IMAGE_SAMPLE_C_B_CL_O">;
979defm : ImageSampleAltPatterns<SIImage_sample_c_lz_o, "IMAGE_SAMPLE_C_LZ_O">;
980defm : ImageSampleAltPatterns<SIImage_sample_c_cd_o, "IMAGE_SAMPLE_C_CD_O">;
981defm : ImageSampleAltPatterns<SIImage_sample_c_cd_cl_o, "IMAGE_SAMPLE_C_CD_CL_O">;
982
983// Basic gather4 alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000984defm : ImageGather4AltPatterns<SIImage_gather4, "IMAGE_GATHER4">;
985defm : ImageGather4AltPatterns<SIImage_gather4_cl, "IMAGE_GATHER4_CL">;
986defm : ImageGather4AltPatterns<SIImage_gather4_l, "IMAGE_GATHER4_L">;
987defm : ImageGather4AltPatterns<SIImage_gather4_b, "IMAGE_GATHER4_B">;
988defm : ImageGather4AltPatterns<SIImage_gather4_b_cl, "IMAGE_GATHER4_B_CL">;
989defm : ImageGather4AltPatterns<SIImage_gather4_lz, "IMAGE_GATHER4_LZ">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000990
991// Gather4 with comparison alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +0000992defm : ImageGather4AltPatterns<SIImage_gather4_c, "IMAGE_GATHER4_C">;
993defm : ImageGather4AltPatterns<SIImage_gather4_c_cl, "IMAGE_GATHER4_C_CL">;
994defm : ImageGather4AltPatterns<SIImage_gather4_c_l, "IMAGE_GATHER4_C_L">;
995defm : ImageGather4AltPatterns<SIImage_gather4_c_b, "IMAGE_GATHER4_C_B">;
996defm : ImageGather4AltPatterns<SIImage_gather4_c_b_cl, "IMAGE_GATHER4_C_B_CL">;
997defm : ImageGather4AltPatterns<SIImage_gather4_c_lz, "IMAGE_GATHER4_C_LZ">;
Changpeng Fang4737e892018-01-18 22:08:53 +0000998
999// Gather4 with offsets alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +00001000defm : ImageGather4AltPatterns<SIImage_gather4_o, "IMAGE_GATHER4_O">;
1001defm : ImageGather4AltPatterns<SIImage_gather4_cl_o, "IMAGE_GATHER4_CL_O">;
1002defm : ImageGather4AltPatterns<SIImage_gather4_l_o, "IMAGE_GATHER4_L_O">;
1003defm : ImageGather4AltPatterns<SIImage_gather4_b_o, "IMAGE_GATHER4_B_O">;
1004defm : ImageGather4AltPatterns<SIImage_gather4_b_cl_o, "IMAGE_GATHER4_B_CL_O">;
1005defm : ImageGather4AltPatterns<SIImage_gather4_lz_o, "IMAGE_GATHER4_LZ_O">;
Changpeng Fang4737e892018-01-18 22:08:53 +00001006
1007// Gather4 with comparison and offsets alternative.
Dmitry Preobrazhenskyda4a7c02018-03-12 15:03:34 +00001008defm : ImageGather4AltPatterns<SIImage_gather4_c_o, "IMAGE_GATHER4_C_O">;
1009defm : ImageGather4AltPatterns<SIImage_gather4_c_cl_o, "IMAGE_GATHER4_C_CL_O">;
1010defm : ImageGather4AltPatterns<SIImage_gather4_c_l_o, "IMAGE_GATHER4_C_L_O">;
1011defm : ImageGather4AltPatterns<SIImage_gather4_c_b_o, "IMAGE_GATHER4_C_B_O">;
1012defm : ImageGather4AltPatterns<SIImage_gather4_c_b_cl_o, "IMAGE_GATHER4_C_B_CL_O">;
1013defm : ImageGather4AltPatterns<SIImage_gather4_c_lz_o, "IMAGE_GATHER4_C_LZ_O">;
Changpeng Fang4737e892018-01-18 22:08:53 +00001014
1015defm : ImageSamplePatterns<int_amdgcn_image_getlod, "IMAGE_GET_LOD">;
Changpeng Fangb28fe032016-09-01 17:54:54 +00001016
1017// Image atomics
1018defm : ImageAtomicPatterns<int_amdgcn_image_atomic_swap, "IMAGE_ATOMIC_SWAP">;
Dmitry Preobrazhensky0b4eb1e2018-01-26 15:43:29 +00001019def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V1, i32>;
1020def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V2, v2i32>;
1021def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V4, v4i32>;
Changpeng Fangb28fe032016-09-01 17:54:54 +00001022defm : ImageAtomicPatterns<int_amdgcn_image_atomic_add, "IMAGE_ATOMIC_ADD">;
1023defm : ImageAtomicPatterns<int_amdgcn_image_atomic_sub, "IMAGE_ATOMIC_SUB">;
1024defm : ImageAtomicPatterns<int_amdgcn_image_atomic_smin, "IMAGE_ATOMIC_SMIN">;
1025defm : ImageAtomicPatterns<int_amdgcn_image_atomic_umin, "IMAGE_ATOMIC_UMIN">;
1026defm : ImageAtomicPatterns<int_amdgcn_image_atomic_smax, "IMAGE_ATOMIC_SMAX">;
1027defm : ImageAtomicPatterns<int_amdgcn_image_atomic_umax, "IMAGE_ATOMIC_UMAX">;
1028defm : ImageAtomicPatterns<int_amdgcn_image_atomic_and, "IMAGE_ATOMIC_AND">;
1029defm : ImageAtomicPatterns<int_amdgcn_image_atomic_or, "IMAGE_ATOMIC_OR">;
1030defm : ImageAtomicPatterns<int_amdgcn_image_atomic_xor, "IMAGE_ATOMIC_XOR">;
1031defm : ImageAtomicPatterns<int_amdgcn_image_atomic_inc, "IMAGE_ATOMIC_INC">;
1032defm : ImageAtomicPatterns<int_amdgcn_image_atomic_dec, "IMAGE_ATOMIC_DEC">;
1033
1034/* SIsample for simple 1D texture lookup */
Matt Arsenault90c75932017-10-03 00:06:41 +00001035def : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +00001036 (SIsample i32:$addr, v8i32:$rsrc, v4i32:$sampler, imm),
1037 (IMAGE_SAMPLE_V4_V1 $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 0)
1038>;
1039
Matt Arsenault90c75932017-10-03 00:06:41 +00001040class SamplePattern<SDNode name, MIMG opcode, ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +00001041 (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, imm),
1042 (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 0)
1043>;
1044
Matt Arsenault90c75932017-10-03 00:06:41 +00001045class SampleRectPattern<SDNode name, MIMG opcode, ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +00001046 (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_RECT),
1047 (opcode $addr, $rsrc, $sampler, 0xf, 1, 0, 0, 0, 0, 0, 0)
1048>;
1049
Matt Arsenault90c75932017-10-03 00:06:41 +00001050class SampleArrayPattern<SDNode name, MIMG opcode, ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +00001051 (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_ARRAY),
1052 (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 1)
1053>;
1054
1055class SampleShadowPattern<SDNode name, MIMG opcode,
Matt Arsenault90c75932017-10-03 00:06:41 +00001056 ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +00001057 (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_SHADOW),
1058 (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 0)
1059>;
1060
1061class SampleShadowArrayPattern<SDNode name, MIMG opcode,
Matt Arsenault90c75932017-10-03 00:06:41 +00001062 ValueType vt> : GCNPat <
Changpeng Fangb28fe032016-09-01 17:54:54 +00001063 (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_SHADOW_ARRAY),
1064 (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 1)
1065>;
1066
1067/* SIsample* for texture lookups consuming more address parameters */
1068multiclass SamplePatterns<MIMG sample, MIMG sample_c, MIMG sample_l,
1069 MIMG sample_c_l, MIMG sample_b, MIMG sample_c_b,
1070MIMG sample_d, MIMG sample_c_d, ValueType addr_type> {
1071 def : SamplePattern <SIsample, sample, addr_type>;
1072 def : SampleRectPattern <SIsample, sample, addr_type>;
1073 def : SampleArrayPattern <SIsample, sample, addr_type>;
1074 def : SampleShadowPattern <SIsample, sample_c, addr_type>;
1075 def : SampleShadowArrayPattern <SIsample, sample_c, addr_type>;
1076
1077 def : SamplePattern <SIsamplel, sample_l, addr_type>;
1078 def : SampleArrayPattern <SIsamplel, sample_l, addr_type>;
1079 def : SampleShadowPattern <SIsamplel, sample_c_l, addr_type>;
1080 def : SampleShadowArrayPattern <SIsamplel, sample_c_l, addr_type>;
1081
1082 def : SamplePattern <SIsampleb, sample_b, addr_type>;
1083 def : SampleArrayPattern <SIsampleb, sample_b, addr_type>;
1084 def : SampleShadowPattern <SIsampleb, sample_c_b, addr_type>;
1085 def : SampleShadowArrayPattern <SIsampleb, sample_c_b, addr_type>;
1086
1087 def : SamplePattern <SIsampled, sample_d, addr_type>;
1088 def : SampleArrayPattern <SIsampled, sample_d, addr_type>;
1089 def : SampleShadowPattern <SIsampled, sample_c_d, addr_type>;
1090 def : SampleShadowArrayPattern <SIsampled, sample_c_d, addr_type>;
1091}
1092
1093defm : SamplePatterns<IMAGE_SAMPLE_V4_V2, IMAGE_SAMPLE_C_V4_V2,
1094 IMAGE_SAMPLE_L_V4_V2, IMAGE_SAMPLE_C_L_V4_V2,
1095 IMAGE_SAMPLE_B_V4_V2, IMAGE_SAMPLE_C_B_V4_V2,
1096 IMAGE_SAMPLE_D_V4_V2, IMAGE_SAMPLE_C_D_V4_V2,
1097 v2i32>;
1098defm : SamplePatterns<IMAGE_SAMPLE_V4_V4, IMAGE_SAMPLE_C_V4_V4,
1099 IMAGE_SAMPLE_L_V4_V4, IMAGE_SAMPLE_C_L_V4_V4,
1100 IMAGE_SAMPLE_B_V4_V4, IMAGE_SAMPLE_C_B_V4_V4,
1101 IMAGE_SAMPLE_D_V4_V4, IMAGE_SAMPLE_C_D_V4_V4,
1102 v4i32>;
1103defm : SamplePatterns<IMAGE_SAMPLE_V4_V8, IMAGE_SAMPLE_C_V4_V8,
1104 IMAGE_SAMPLE_L_V4_V8, IMAGE_SAMPLE_C_L_V4_V8,
1105 IMAGE_SAMPLE_B_V4_V8, IMAGE_SAMPLE_C_B_V4_V8,
1106 IMAGE_SAMPLE_D_V4_V8, IMAGE_SAMPLE_C_D_V4_V8,
1107 v8i32>;
1108defm : SamplePatterns<IMAGE_SAMPLE_V4_V16, IMAGE_SAMPLE_C_V4_V16,
1109 IMAGE_SAMPLE_L_V4_V16, IMAGE_SAMPLE_C_L_V4_V16,
1110 IMAGE_SAMPLE_B_V4_V16, IMAGE_SAMPLE_C_B_V4_V16,
1111 IMAGE_SAMPLE_D_V4_V16, IMAGE_SAMPLE_C_D_V4_V16,
1112 v16i32>;