| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 1 | //===-- 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 | |
| 10 | class MIMG_Mask <string op, int channels> { |
| 11 | string Op = op; |
| 12 | int Channels = channels; |
| 13 | } |
| 14 | |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 15 | class MIMG_Atomic_Size <string op, bit is32Bit> { |
| 16 | string Op = op; |
| 17 | int AtomicSize = !if(is32Bit, 1, 2); |
| 18 | } |
| 19 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 20 | class MIMG_Gather_Size <string op, int channels> { |
| 21 | string Op = op; |
| 22 | int Channels = channels; |
| 23 | } |
| 24 | |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 25 | class mimg <bits<7> si, bits<7> vi = si> { |
| 26 | field bits<7> SI = si; |
| 27 | field bits<7> VI = vi; |
| 28 | } |
| 29 | |
| 30 | class 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 Stellard | 244891d | 2016-12-20 15:52:17 +0000 | [diff] [blame] | 38 | let usesCustomInserter = 1; |
| Marek Olsak | b83f5c9 | 2017-07-04 14:43:38 +0000 | [diff] [blame] | 39 | let SchedRW = [WriteVMEM]; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 40 | } |
| 41 | |
| 42 | class MIMG_NoSampler_Helper <bits<7> op, string asm, |
| 43 | RegisterClass dst_rc, |
| 44 | RegisterClass addr_rc, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 45 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 56 | let ssamp = 0; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 57 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 58 | let HasD16 = has_d16; |
| 59 | let d16 = !if(HasD16, ?, 0); |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 60 | } |
| 61 | |
| 62 | multiclass MIMG_NoSampler_Src_Helper <bits<7> op, string asm, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 63 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 74 | } |
| 75 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 76 | multiclass 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 Preobrazhensky | 2456ac6 | 2018-03-28 15:44:16 +0000 | [diff] [blame] | 81 | } |
| 82 | |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 83 | class MIMG_Store_Helper <bits<7> op, string asm, |
| 84 | RegisterClass data_rc, |
| Matt Arsenault | cad7fa8 | 2017-12-13 21:07:51 +0000 | [diff] [blame] | 85 | RegisterClass addr_rc, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 86 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 97 | let ssamp = 0; |
| Matt Arsenault | d94b63d | 2017-12-29 17:18:18 +0000 | [diff] [blame] | 98 | let mayLoad = 0; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 99 | let mayStore = 1; |
| Matt Arsenault | d94b63d | 2017-12-29 17:18:18 +0000 | [diff] [blame] | 100 | let hasSideEffects = 0; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 101 | let hasPostISelHook = 0; |
| 102 | let DisableWQM = 1; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 103 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 104 | let HasD16 = has_d16; |
| 105 | let d16 = !if(HasD16, ?, 0); |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 106 | } |
| 107 | |
| 108 | multiclass MIMG_Store_Addr_Helper <bits<7> op, string asm, |
| 109 | RegisterClass data_rc, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 110 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 120 | } |
| 121 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 122 | multiclass 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 Preobrazhensky | 2456ac6 | 2018-03-28 15:44:16 +0000 | [diff] [blame] | 127 | } |
| 128 | |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 129 | class MIMG_Atomic_Helper <string asm, RegisterClass data_rc, |
| Dmitry Preobrazhensky | 6cb42e7 | 2018-01-26 14:07:38 +0000 | [diff] [blame] | 130 | RegisterClass addr_rc, string dns="", |
| 131 | bit enableDasm = 0> : MIMG_Helper < |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 132 | (outs data_rc:$vdst), |
| 133 | (ins data_rc:$vdata, addr_rc:$vaddr, SReg_256:$srsrc, |
| Nicolai Haehnle | 59198ed | 2018-06-04 14:45:20 +0000 | [diff] [blame] | 134 | DMask:$dmask, UNorm:$unorm, GLC:$glc, SLC:$slc, |
| 135 | R128:$r128, TFE:$tfe, LWE:$lwe, DA:$da), |
| Dmitry Preobrazhensky | 6cb42e7 | 2018-01-26 14:07:38 +0000 | [diff] [blame] | 136 | asm#" $vdst, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da", |
| 137 | !if(enableDasm, dns, "")> { |
| Matt Arsenault | d94b63d | 2017-12-29 17:18:18 +0000 | [diff] [blame] | 138 | let mayLoad = 1; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 139 | let mayStore = 1; |
| Matt Arsenault | d94b63d | 2017-12-29 17:18:18 +0000 | [diff] [blame] | 140 | let hasSideEffects = 1; // FIXME: Remove this |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 141 | let hasPostISelHook = 0; |
| 142 | let DisableWQM = 1; |
| 143 | let Constraints = "$vdst = $vdata"; |
| 144 | let AsmMatchConverter = "cvtMIMGAtomic"; |
| 145 | } |
| 146 | |
| 147 | class MIMG_Atomic_Real_si<mimg op, string name, string asm, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 148 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 153 | let isCodeGenOnly = 0; |
| 154 | let AssemblerPredicates = [isSICI]; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 155 | let DisableDecoder = DisableSIDecoder; |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 156 | let d16 = 0; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 157 | } |
| 158 | |
| 159 | class MIMG_Atomic_Real_vi<mimg op, string name, string asm, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 160 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 165 | let isCodeGenOnly = 0; |
| 166 | let AssemblerPredicates = [isVI]; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 167 | let DisableDecoder = DisableVIDecoder; |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 168 | let d16 = 0; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 169 | } |
| 170 | |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 171 | multiclass MIMG_Atomic_Helper_m <mimg op, |
| 172 | string name, |
| 173 | string asm, |
| 174 | string key, |
| Dmitry Preobrazhensky | 6cb42e7 | 2018-01-26 14:07:38 +0000 | [diff] [blame] | 175 | RegisterClass data_rc, |
| 176 | RegisterClass addr_rc, |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 177 | bit is32Bit, |
| Dmitry Preobrazhensky | 6cb42e7 | 2018-01-26 14:07:38 +0000 | [diff] [blame] | 178 | bit enableDasm = 0> { |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 179 | 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 Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 185 | def _si : MIMG_Atomic_Real_si<op, name, asm, data_rc, addr_rc, enableDasm>, |
| 186 | MIMG_Atomic_Size<key # "_si", is32Bit>; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 187 | |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 188 | def _vi : MIMG_Atomic_Real_vi<op, name, asm, data_rc, addr_rc, enableDasm>, |
| 189 | MIMG_Atomic_Size<key # "_vi", is32Bit>; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 190 | } |
| 191 | } |
| 192 | |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 193 | multiclass 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 Preobrazhensky | 6cb42e7 | 2018-01-26 14:07:38 +0000 | [diff] [blame] | 199 | // _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 Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 201 | 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 Preobrazhensky | a0b8cd0 | 2018-04-04 13:01:17 +0000 | [diff] [blame] | 203 | 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 Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 205 | } |
| 206 | |
| 207 | multiclass 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 215 | } |
| 216 | |
| 217 | class MIMG_Sampler_Helper <bits<7> op, string asm, |
| 218 | RegisterClass dst_rc, |
| 219 | RegisterClass src_rc, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 220 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 231 | let WQM = wqm; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 232 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 233 | let HasD16 = has_d16; |
| 234 | let d16 = !if(HasD16, ?, 0); |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 235 | } |
| 236 | |
| 237 | multiclass MIMG_Sampler_Src_Helper <bits<7> op, string asm, |
| 238 | RegisterClass dst_rc, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 239 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 253 | } |
| 254 | |
| Nicolai Haehnle | 2367f03 | 2018-06-21 13:36:13 +0000 | [diff] [blame] | 255 | multiclass MIMG_Sampler <bits<7> op, AMDGPUSampleVariant sample, bit wqm = 0, |
| 256 | bit has_d16 = 1, |
| 257 | string asm = "image_sample"#sample.LowerCaseMod> { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 258 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 262 | } |
| 263 | |
| Nicolai Haehnle | 2367f03 | 2018-06-21 13:36:13 +0000 | [diff] [blame] | 264 | multiclass MIMG_Sampler_WQM <bits<7> op, AMDGPUSampleVariant sample> : MIMG_Sampler<op, sample, 1>; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 265 | |
| 266 | class MIMG_Gather_Helper <bits<7> op, string asm, |
| 267 | RegisterClass dst_rc, |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 268 | RegisterClass src_rc, |
| 269 | bit wqm, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 270 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 278 | 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 Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 291 | let HasD16 = 1; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 292 | |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 293 | let DecoderNamespace = dns; |
| 294 | let isAsmParserOnly = !if(!eq(dns,""), 1, 0); |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 295 | } |
| 296 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 297 | |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 298 | multiclass MIMG_Gather_Src_Helper <bits<7> op, string asm, |
| 299 | RegisterClass dst_rc, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 300 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 314 | } |
| 315 | |
| Nicolai Haehnle | 2367f03 | 2018-06-21 13:36:13 +0000 | [diff] [blame] | 316 | multiclass MIMG_Gather <bits<7> op, AMDGPUSampleVariant sample, bit wqm = 0, |
| 317 | string asm = "image_gather4"#sample.LowerCaseMod> { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 318 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 320 | } |
| 321 | |
| Nicolai Haehnle | 2367f03 | 2018-06-21 13:36:13 +0000 | [diff] [blame] | 322 | multiclass MIMG_Gather_WQM <bits<7> op, AMDGPUSampleVariant sample> |
| 323 | : MIMG_Gather<op, sample, 1>; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 324 | |
| 325 | //===----------------------------------------------------------------------===// |
| 326 | // MIMG Instructions |
| 327 | //===----------------------------------------------------------------------===// |
| 328 | let SubtargetPredicate = isGCN in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 329 | defm IMAGE_LOAD : MIMG_NoSampler <0x00000000, "image_load", 1>; |
| 330 | defm IMAGE_LOAD_MIP : MIMG_NoSampler <0x00000001, "image_load_mip", 1>; |
| 331 | defm IMAGE_LOAD_PCK : MIMG_NoSampler <0x00000002, "image_load_pck", 0>; |
| 332 | defm IMAGE_LOAD_PCK_SGN : MIMG_NoSampler <0x00000003, "image_load_pck_sgn", 0>; |
| 333 | defm IMAGE_LOAD_MIP_PCK : MIMG_NoSampler <0x00000004, "image_load_mip_pck", 0>; |
| 334 | defm IMAGE_LOAD_MIP_PCK_SGN : MIMG_NoSampler <0x00000005, "image_load_mip_pck_sgn", 0>; |
| 335 | defm IMAGE_STORE : MIMG_Store <0x00000008, "image_store", 1>; |
| 336 | defm IMAGE_STORE_MIP : MIMG_Store <0x00000009, "image_store_mip", 1>; |
| 337 | defm IMAGE_STORE_PCK : MIMG_Store <0x0000000a, "image_store_pck", 0>; |
| 338 | defm IMAGE_STORE_MIP_PCK : MIMG_Store <0x0000000b, "image_store_mip_pck", 0>; |
| Matt Arsenault | 856777d | 2017-12-08 20:00:57 +0000 | [diff] [blame] | 339 | |
| 340 | let mayLoad = 0, mayStore = 0 in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 341 | defm IMAGE_GET_RESINFO : MIMG_NoSampler <0x0000000e, "image_get_resinfo", 0>; |
| Matt Arsenault | 856777d | 2017-12-08 20:00:57 +0000 | [diff] [blame] | 342 | } |
| 343 | |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 344 | defm IMAGE_ATOMIC_SWAP : MIMG_Atomic <mimg<0x0f, 0x10>, "image_atomic_swap">; |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 345 | defm IMAGE_ATOMIC_CMPSWAP : MIMG_Atomic <mimg<0x10, 0x11>, "image_atomic_cmpswap", VReg_64, VReg_128>; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 346 | defm IMAGE_ATOMIC_ADD : MIMG_Atomic <mimg<0x11, 0x12>, "image_atomic_add">; |
| 347 | defm 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 |
| 349 | defm IMAGE_ATOMIC_SMIN : MIMG_Atomic <mimg<0x14>, "image_atomic_smin">; |
| 350 | defm IMAGE_ATOMIC_UMIN : MIMG_Atomic <mimg<0x15>, "image_atomic_umin">; |
| 351 | defm IMAGE_ATOMIC_SMAX : MIMG_Atomic <mimg<0x16>, "image_atomic_smax">; |
| 352 | defm IMAGE_ATOMIC_UMAX : MIMG_Atomic <mimg<0x17>, "image_atomic_umax">; |
| 353 | defm IMAGE_ATOMIC_AND : MIMG_Atomic <mimg<0x18>, "image_atomic_and">; |
| 354 | defm IMAGE_ATOMIC_OR : MIMG_Atomic <mimg<0x19>, "image_atomic_or">; |
| 355 | defm IMAGE_ATOMIC_XOR : MIMG_Atomic <mimg<0x1a>, "image_atomic_xor">; |
| 356 | defm IMAGE_ATOMIC_INC : MIMG_Atomic <mimg<0x1b>, "image_atomic_inc">; |
| 357 | defm 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 Haehnle | 2367f03 | 2018-06-21 13:36:13 +0000 | [diff] [blame] | 361 | defm IMAGE_SAMPLE : MIMG_Sampler_WQM <0x00000020, AMDGPUSample>; |
| 362 | defm IMAGE_SAMPLE_CL : MIMG_Sampler_WQM <0x00000021, AMDGPUSample_cl>; |
| 363 | defm IMAGE_SAMPLE_D : MIMG_Sampler <0x00000022, AMDGPUSample_d>; |
| 364 | defm IMAGE_SAMPLE_D_CL : MIMG_Sampler <0x00000023, AMDGPUSample_d_cl>; |
| 365 | defm IMAGE_SAMPLE_L : MIMG_Sampler <0x00000024, AMDGPUSample_l>; |
| 366 | defm IMAGE_SAMPLE_B : MIMG_Sampler_WQM <0x00000025, AMDGPUSample_b>; |
| 367 | defm IMAGE_SAMPLE_B_CL : MIMG_Sampler_WQM <0x00000026, AMDGPUSample_b_cl>; |
| 368 | defm IMAGE_SAMPLE_LZ : MIMG_Sampler <0x00000027, AMDGPUSample_lz>; |
| 369 | defm IMAGE_SAMPLE_C : MIMG_Sampler_WQM <0x00000028, AMDGPUSample_c>; |
| 370 | defm IMAGE_SAMPLE_C_CL : MIMG_Sampler_WQM <0x00000029, AMDGPUSample_c_cl>; |
| 371 | defm IMAGE_SAMPLE_C_D : MIMG_Sampler <0x0000002a, AMDGPUSample_c_d>; |
| 372 | defm IMAGE_SAMPLE_C_D_CL : MIMG_Sampler <0x0000002b, AMDGPUSample_c_d_cl>; |
| 373 | defm IMAGE_SAMPLE_C_L : MIMG_Sampler <0x0000002c, AMDGPUSample_c_l>; |
| 374 | defm IMAGE_SAMPLE_C_B : MIMG_Sampler_WQM <0x0000002d, AMDGPUSample_c_b>; |
| 375 | defm IMAGE_SAMPLE_C_B_CL : MIMG_Sampler_WQM <0x0000002e, AMDGPUSample_c_b_cl>; |
| 376 | defm IMAGE_SAMPLE_C_LZ : MIMG_Sampler <0x0000002f, AMDGPUSample_c_lz>; |
| 377 | defm IMAGE_SAMPLE_O : MIMG_Sampler_WQM <0x00000030, AMDGPUSample_o>; |
| 378 | defm IMAGE_SAMPLE_CL_O : MIMG_Sampler_WQM <0x00000031, AMDGPUSample_cl_o>; |
| 379 | defm IMAGE_SAMPLE_D_O : MIMG_Sampler <0x00000032, AMDGPUSample_d_o>; |
| 380 | defm IMAGE_SAMPLE_D_CL_O : MIMG_Sampler <0x00000033, AMDGPUSample_d_cl_o>; |
| 381 | defm IMAGE_SAMPLE_L_O : MIMG_Sampler <0x00000034, AMDGPUSample_l_o>; |
| 382 | defm IMAGE_SAMPLE_B_O : MIMG_Sampler_WQM <0x00000035, AMDGPUSample_b_o>; |
| 383 | defm IMAGE_SAMPLE_B_CL_O : MIMG_Sampler_WQM <0x00000036, AMDGPUSample_b_cl_o>; |
| 384 | defm IMAGE_SAMPLE_LZ_O : MIMG_Sampler <0x00000037, AMDGPUSample_lz_o>; |
| 385 | defm IMAGE_SAMPLE_C_O : MIMG_Sampler_WQM <0x00000038, AMDGPUSample_c_o>; |
| 386 | defm IMAGE_SAMPLE_C_CL_O : MIMG_Sampler_WQM <0x00000039, AMDGPUSample_c_cl_o>; |
| 387 | defm IMAGE_SAMPLE_C_D_O : MIMG_Sampler <0x0000003a, AMDGPUSample_c_d_o>; |
| 388 | defm IMAGE_SAMPLE_C_D_CL_O : MIMG_Sampler <0x0000003b, AMDGPUSample_c_d_cl_o>; |
| 389 | defm IMAGE_SAMPLE_C_L_O : MIMG_Sampler <0x0000003c, AMDGPUSample_c_l_o>; |
| 390 | defm IMAGE_SAMPLE_C_B_CL_O : MIMG_Sampler_WQM <0x0000003e, AMDGPUSample_c_b_cl_o>; |
| 391 | defm IMAGE_SAMPLE_C_B_O : MIMG_Sampler_WQM <0x0000003d, AMDGPUSample_c_b_o>; |
| 392 | defm IMAGE_SAMPLE_C_LZ_O : MIMG_Sampler <0x0000003f, AMDGPUSample_c_lz_o>; |
| 393 | defm IMAGE_GATHER4 : MIMG_Gather_WQM <0x00000040, AMDGPUSample>; |
| 394 | defm IMAGE_GATHER4_CL : MIMG_Gather_WQM <0x00000041, AMDGPUSample_cl>; |
| 395 | defm IMAGE_GATHER4_L : MIMG_Gather <0x00000044, AMDGPUSample_l>; |
| 396 | defm IMAGE_GATHER4_B : MIMG_Gather_WQM <0x00000045, AMDGPUSample_b>; |
| 397 | defm IMAGE_GATHER4_B_CL : MIMG_Gather_WQM <0x00000046, AMDGPUSample_b_cl>; |
| 398 | defm IMAGE_GATHER4_LZ : MIMG_Gather <0x00000047, AMDGPUSample_lz>; |
| 399 | defm IMAGE_GATHER4_C : MIMG_Gather_WQM <0x00000048, AMDGPUSample_c>; |
| 400 | defm IMAGE_GATHER4_C_CL : MIMG_Gather_WQM <0x00000049, AMDGPUSample_c_cl>; |
| 401 | defm IMAGE_GATHER4_C_L : MIMG_Gather <0x0000004c, AMDGPUSample_c_l>; |
| 402 | defm IMAGE_GATHER4_C_B : MIMG_Gather_WQM <0x0000004d, AMDGPUSample_c_b>; |
| 403 | defm IMAGE_GATHER4_C_B_CL : MIMG_Gather_WQM <0x0000004e, AMDGPUSample_c_b_cl>; |
| 404 | defm IMAGE_GATHER4_C_LZ : MIMG_Gather <0x0000004f, AMDGPUSample_c_lz>; |
| 405 | defm IMAGE_GATHER4_O : MIMG_Gather_WQM <0x00000050, AMDGPUSample_o>; |
| 406 | defm IMAGE_GATHER4_CL_O : MIMG_Gather_WQM <0x00000051, AMDGPUSample_cl_o>; |
| 407 | defm IMAGE_GATHER4_L_O : MIMG_Gather <0x00000054, AMDGPUSample_l_o>; |
| 408 | defm IMAGE_GATHER4_B_O : MIMG_Gather_WQM <0x00000055, AMDGPUSample_b_o>; |
| 409 | defm IMAGE_GATHER4_B_CL_O : MIMG_Gather <0x00000056, AMDGPUSample_b_cl_o>; |
| 410 | defm IMAGE_GATHER4_LZ_O : MIMG_Gather <0x00000057, AMDGPUSample_lz_o>; |
| 411 | defm IMAGE_GATHER4_C_O : MIMG_Gather_WQM <0x00000058, AMDGPUSample_c_o>; |
| 412 | defm IMAGE_GATHER4_C_CL_O : MIMG_Gather_WQM <0x00000059, AMDGPUSample_c_cl_o>; |
| 413 | defm IMAGE_GATHER4_C_L_O : MIMG_Gather <0x0000005c, AMDGPUSample_c_l_o>; |
| 414 | defm IMAGE_GATHER4_C_B_O : MIMG_Gather_WQM <0x0000005d, AMDGPUSample_c_b_o>; |
| 415 | defm IMAGE_GATHER4_C_B_CL_O : MIMG_Gather_WQM <0x0000005e, AMDGPUSample_c_b_cl_o>; |
| 416 | defm IMAGE_GATHER4_C_LZ_O : MIMG_Gather <0x0000005f, AMDGPUSample_c_lz_o>; |
| Matt Arsenault | 856777d | 2017-12-08 20:00:57 +0000 | [diff] [blame] | 417 | |
| 418 | let mayLoad = 0, mayStore = 0 in { |
| Nicolai Haehnle | 2367f03 | 2018-06-21 13:36:13 +0000 | [diff] [blame] | 419 | defm IMAGE_GET_LOD : MIMG_Sampler <0x00000060, AMDGPUSample, 1, 0, "image_get_lod">; |
| Matt Arsenault | 856777d | 2017-12-08 20:00:57 +0000 | [diff] [blame] | 420 | } |
| 421 | |
| Nicolai Haehnle | 2367f03 | 2018-06-21 13:36:13 +0000 | [diff] [blame] | 422 | defm IMAGE_SAMPLE_CD : MIMG_Sampler <0x00000068, AMDGPUSample_cd>; |
| 423 | defm IMAGE_SAMPLE_CD_CL : MIMG_Sampler <0x00000069, AMDGPUSample_cd_cl>; |
| 424 | defm IMAGE_SAMPLE_C_CD : MIMG_Sampler <0x0000006a, AMDGPUSample_c_cd>; |
| 425 | defm IMAGE_SAMPLE_C_CD_CL : MIMG_Sampler <0x0000006b, AMDGPUSample_c_cd_cl>; |
| 426 | defm IMAGE_SAMPLE_CD_O : MIMG_Sampler <0x0000006c, AMDGPUSample_cd_o>; |
| 427 | defm IMAGE_SAMPLE_CD_CL_O : MIMG_Sampler <0x0000006d, AMDGPUSample_cd_cl_o>; |
| 428 | defm IMAGE_SAMPLE_C_CD_O : MIMG_Sampler <0x0000006e, AMDGPUSample_c_cd_o>; |
| 429 | defm IMAGE_SAMPLE_C_CD_CL_O : MIMG_Sampler <0x0000006f, AMDGPUSample_c_cd_cl_o>; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 430 | //def IMAGE_RSRC256 : MIMG_NoPattern_RSRC256 <"image_rsrc256", 0x0000007e>; |
| 431 | //def IMAGE_SAMPLER : MIMG_NoPattern_ <"image_sampler", 0x0000007f>; |
| 432 | } |
| 433 | |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 434 | /********** ============================== **********/ |
| 435 | /********** Dimension-aware image patterns **********/ |
| 436 | /********** ============================== **********/ |
| 437 | |
| 438 | class 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 | |
| 460 | class 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). |
| 468 | class 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 | |
| 480 | class ImageDimPattern<AMDGPUImageDimIntrinsic I, |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 481 | string dop, ValueType dty, bit d16, |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 482 | string suffix = ""> : GCNPat<(undef), (undef)> { |
| 483 | list<AMDGPUArg> AddrArgs = I.P.AddrDefaultArgs; |
| 484 | getDwordsType AddrDwords = getDwordsType<!size(AddrArgs)>; |
| 485 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 486 | MIMG MI = |
| 487 | !cast<MIMG>(!strconcat("IMAGE_", I.P.OpMod, dop, AddrDwords.suffix, suffix)); |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 488 | |
| 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 Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 543 | { I.P.Dim.DA }), |
| 544 | !if(MI.HasD16, (MI d16), (MI))); |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 545 | let ResultInstrs = [ |
| 546 | !if(IsCmpSwap, (EXTRACT_SUBREG ImageInstruction, sub0), ImageInstruction) |
| 547 | ]; |
| 548 | } |
| 549 | |
| 550 | foreach intr = !listconcat(AMDGPUImageDimIntrinsics, |
| 551 | AMDGPUImageDimGetResInfoIntrinsics) in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 552 | 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 Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 555 | } |
| 556 | |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 557 | multiclass ImageDimD16Helper<AMDGPUImageDimIntrinsic I, |
| 558 | AMDGPUImageDimIntrinsic d16helper> { |
| 559 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 560 | 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 Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 563 | } // End HasUnpackedD16VMem. |
| 564 | |
| 565 | let SubtargetPredicate = HasPackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 566 | 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 Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 569 | } // End HasPackedD16VMem. |
| 570 | } |
| 571 | |
| 572 | foreach 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 | |
| 589 | foreach intr = AMDGPUImageDimGatherIntrinsics in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 590 | def intr#_pat3 : ImageDimPattern<intr, "_V4", v4f32, 0>; |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 591 | |
| 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 Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 606 | "_V4", v4i32, 1>; |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 607 | } // End HasUnpackedD16VMem. |
| 608 | |
| 609 | let SubtargetPredicate = HasPackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 610 | def intr#_packed_v4 : ImageDimPattern<intr, "_V2", v4f16, 1>; |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 611 | } // End HasPackedD16VMem. |
| 612 | } |
| 613 | |
| 614 | foreach intr = AMDGPUImageDimAtomicIntrinsics in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 615 | def intr#_pat1 : ImageDimPattern<intr, "_V1", i32, 0>; |
| Nicolai Haehnle | 2f5a738 | 2018-04-04 10:58:54 +0000 | [diff] [blame] | 616 | } |
| 617 | |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 618 | /********** ======================= **********/ |
| 619 | /********** Image sampling patterns **********/ |
| 620 | /********** ======================= **********/ |
| 621 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 622 | // ImageSample for amdgcn |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 623 | // TODO: |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 624 | // 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 Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 626 | multiclass ImageSamplePattern<SDPatternOperator name, MIMG opcode, |
| 627 | ValueType dt, ValueType vt, bit d16> { |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 628 | def : GCNPat< |
| Changpeng Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 629 | (dt (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, i32:$dmask, i1:$unorm, i1:$glc, |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 630 | i1:$slc, i1:$lwe, i1:$da)), |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 631 | !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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 635 | >; |
| 636 | } |
| 637 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 638 | multiclass 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 Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 645 | } |
| 646 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 647 | // ImageSample patterns. |
| 648 | multiclass ImageSamplePatterns<SDPatternOperator name, string opcode> { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 649 | 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 Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 652 | |
| 653 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 654 | defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 655 | } // End HasUnpackedD16VMem. |
| 656 | |
| 657 | let SubtargetPredicate = HasPackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 658 | 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 Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 661 | } // End HasPackedD16VMem. |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 662 | } |
| 663 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 664 | // ImageSample alternative patterns for illegal vector half Types. |
| 665 | multiclass ImageSampleAltPatterns<SDPatternOperator name, string opcode> { |
| 666 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 667 | defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>; |
| 668 | defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 669 | } // End HasUnpackedD16VMem. |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 670 | } |
| 671 | |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 672 | // ImageGather4 patterns. |
| 673 | multiclass ImageGather4Patterns<SDPatternOperator name, string opcode> { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 674 | defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4f32, 0>; |
| Matt Arsenault | 02dc7e1 | 2018-06-15 15:15:46 +0000 | [diff] [blame] | 675 | |
| 676 | let SubtargetPredicate = HasPackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 677 | defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V2), v4f16, 1>; |
| Matt Arsenault | 02dc7e1 | 2018-06-15 15:15:46 +0000 | [diff] [blame] | 678 | } // End HasPackedD16VMem. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 679 | } |
| 680 | |
| 681 | // ImageGather4 alternative patterns for illegal vector half Types. |
| 682 | multiclass ImageGather4AltPatterns<SDPatternOperator name, string opcode> { |
| 683 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 684 | defm : ImageSampleDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>; |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 685 | } // End HasUnpackedD16VMem. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 686 | } |
| 687 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 688 | // ImageLoad for amdgcn. |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 689 | multiclass ImageLoadPattern<SDPatternOperator name, MIMG opcode, |
| 690 | ValueType dt, ValueType vt, bit d16> { |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 691 | def : GCNPat < |
| Changpeng Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 692 | (dt (name vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc, i1:$lwe, |
| Tom Stellard | fac248c | 2016-10-12 16:35:29 +0000 | [diff] [blame] | 693 | i1:$da)), |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 694 | !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 Stellard | fac248c | 2016-10-12 16:35:29 +0000 | [diff] [blame] | 697 | >; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 698 | } |
| 699 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 700 | multiclass 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 Stellard | fac248c | 2016-10-12 16:35:29 +0000 | [diff] [blame] | 705 | } |
| 706 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 707 | // ImageLoad patterns. |
| Changpeng Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 708 | // TODO: support v3f32. |
| 709 | multiclass ImageLoadPatterns<SDPatternOperator name, string opcode> { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 710 | 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 Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 713 | |
| 714 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 715 | defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 716 | } // End HasUnpackedD16VMem. |
| 717 | |
| 718 | let SubtargetPredicate = HasPackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 719 | 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 Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 722 | } // End HasPackedD16VMem. |
| Changpeng Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 723 | } |
| 724 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 725 | // ImageLoad alternative patterns for illegal vector half Types. |
| 726 | multiclass ImageLoadAltPatterns<SDPatternOperator name, string opcode> { |
| 727 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 728 | defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>; |
| 729 | defm : ImageLoadDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 730 | } // End HasUnPackedD16VMem. |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 731 | } |
| 732 | |
| 733 | // ImageStore for amdgcn. |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 734 | multiclass ImageStorePattern<SDPatternOperator name, MIMG opcode, |
| 735 | ValueType dt, ValueType vt, bit d16> { |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 736 | def : GCNPat < |
| Changpeng Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 737 | (name dt:$data, vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc, |
| Tom Stellard | fac248c | 2016-10-12 16:35:29 +0000 | [diff] [blame] | 738 | i1:$lwe, i1:$da), |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 739 | !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 Stellard | fac248c | 2016-10-12 16:35:29 +0000 | [diff] [blame] | 742 | >; |
| 743 | } |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 744 | |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 745 | multiclass 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 Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 750 | } |
| 751 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 752 | // ImageStore patterns. |
| Changpeng Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 753 | // TODO: support v3f32. |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 754 | multiclass ImageStorePatterns<SDPatternOperator name, string opcode> { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 755 | 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 Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 758 | |
| 759 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 760 | defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), f16, 1>; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 761 | } // End HasUnpackedD16VMem. |
| 762 | |
| 763 | let SubtargetPredicate = HasPackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 764 | 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 Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 767 | } // End HasPackedD16VMem. |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 768 | } |
| 769 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 770 | // ImageStore alternative patterns. |
| 771 | multiclass ImageStoreAltPatterns<SDPatternOperator name, string opcode> { |
| 772 | let SubtargetPredicate = HasUnpackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 773 | defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>; |
| 774 | defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V4), v4i32, 1>; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 775 | } // End HasUnpackedD16VMem. |
| 776 | |
| 777 | let SubtargetPredicate = HasPackedD16VMem in { |
| Nicolai Haehnle | f267431 | 2018-06-21 13:36:01 +0000 | [diff] [blame] | 778 | defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V1), i32, 1>; |
| 779 | defm : ImageStoreDataPatterns<name, !cast<string>(opcode # _V2), v2i32, 1>; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 780 | } // End HasPackedD16VMem. |
| 781 | } |
| 782 | |
| 783 | // ImageAtomic for amdgcn. |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 784 | class ImageAtomicPattern<SDPatternOperator name, MIMG opcode, ValueType vt> : GCNPat < |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 785 | (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 Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 789 | // ImageAtomic patterns. |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 790 | multiclass ImageAtomicPatterns<SDPatternOperator name, string opcode> { |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 791 | 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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 794 | } |
| 795 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 796 | // ImageAtomicCmpSwap for amdgcn. |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 797 | class ImageAtomicCmpSwapPattern<MIMG opcode, ValueType vt> : GCNPat < |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 798 | (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 Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 806 | // ======= amdgcn Image Intrinsics ============== |
| 807 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 808 | // Image load. |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 809 | defm : ImageLoadPatterns<int_amdgcn_image_load, "IMAGE_LOAD">; |
| 810 | defm : ImageLoadPatterns<int_amdgcn_image_load_mip, "IMAGE_LOAD_MIP">; |
| Changpeng Fang | 8236fe1 | 2016-11-14 18:33:18 +0000 | [diff] [blame] | 811 | defm : ImageLoadPatterns<int_amdgcn_image_getresinfo, "IMAGE_GET_RESINFO">; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 812 | defm : ImageLoadAltPatterns<SIImage_load, "IMAGE_LOAD">; |
| 813 | defm : ImageLoadAltPatterns<SIImage_load_mip, "IMAGE_LOAD_MIP">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 814 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 815 | // Image store. |
| Matt Arsenault | 1349a04 | 2018-05-22 06:32:10 +0000 | [diff] [blame] | 816 | defm : ImageStorePatterns<int_amdgcn_image_store, "IMAGE_STORE">; |
| 817 | defm : ImageStorePatterns<int_amdgcn_image_store_mip, "IMAGE_STORE_MIP">; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 818 | defm : ImageStoreAltPatterns<SIImage_store, "IMAGE_STORE">; |
| 819 | defm : ImageStoreAltPatterns<SIImage_store_mip, "IMAGE_STORE_MIP">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 820 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 821 | // Basic sample. |
| 822 | defm : ImageSamplePatterns<int_amdgcn_image_sample, "IMAGE_SAMPLE">; |
| 823 | defm : ImageSamplePatterns<int_amdgcn_image_sample_cl, "IMAGE_SAMPLE_CL">; |
| 824 | defm : ImageSamplePatterns<int_amdgcn_image_sample_d, "IMAGE_SAMPLE_D">; |
| 825 | defm : ImageSamplePatterns<int_amdgcn_image_sample_d_cl, "IMAGE_SAMPLE_D_CL">; |
| 826 | defm : ImageSamplePatterns<int_amdgcn_image_sample_l, "IMAGE_SAMPLE_L">; |
| 827 | defm : ImageSamplePatterns<int_amdgcn_image_sample_b, "IMAGE_SAMPLE_B">; |
| 828 | defm : ImageSamplePatterns<int_amdgcn_image_sample_b_cl, "IMAGE_SAMPLE_B_CL">; |
| 829 | defm : ImageSamplePatterns<int_amdgcn_image_sample_lz, "IMAGE_SAMPLE_LZ">; |
| 830 | defm : ImageSamplePatterns<int_amdgcn_image_sample_cd, "IMAGE_SAMPLE_CD">; |
| 831 | defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_cl, "IMAGE_SAMPLE_CD_CL">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 832 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 833 | // Sample with comparison. |
| 834 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c, "IMAGE_SAMPLE_C">; |
| 835 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cl, "IMAGE_SAMPLE_C_CL">; |
| 836 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d, "IMAGE_SAMPLE_C_D">; |
| 837 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_cl, "IMAGE_SAMPLE_C_D_CL">; |
| 838 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_l, "IMAGE_SAMPLE_C_L">; |
| 839 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b, "IMAGE_SAMPLE_C_B">; |
| 840 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_cl, "IMAGE_SAMPLE_C_B_CL">; |
| 841 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_lz, "IMAGE_SAMPLE_C_LZ">; |
| 842 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd, "IMAGE_SAMPLE_C_CD">; |
| 843 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_cl, "IMAGE_SAMPLE_C_CD_CL">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 844 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 845 | // Sample with offsets. |
| 846 | defm : ImageSamplePatterns<int_amdgcn_image_sample_o, "IMAGE_SAMPLE_O">; |
| 847 | defm : ImageSamplePatterns<int_amdgcn_image_sample_cl_o, "IMAGE_SAMPLE_CL_O">; |
| 848 | defm : ImageSamplePatterns<int_amdgcn_image_sample_d_o, "IMAGE_SAMPLE_D_O">; |
| 849 | defm : ImageSamplePatterns<int_amdgcn_image_sample_d_cl_o, "IMAGE_SAMPLE_D_CL_O">; |
| 850 | defm : ImageSamplePatterns<int_amdgcn_image_sample_l_o, "IMAGE_SAMPLE_L_O">; |
| 851 | defm : ImageSamplePatterns<int_amdgcn_image_sample_b_o, "IMAGE_SAMPLE_B_O">; |
| 852 | defm : ImageSamplePatterns<int_amdgcn_image_sample_b_cl_o, "IMAGE_SAMPLE_B_CL_O">; |
| 853 | defm : ImageSamplePatterns<int_amdgcn_image_sample_lz_o, "IMAGE_SAMPLE_LZ_O">; |
| 854 | defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_o, "IMAGE_SAMPLE_CD_O">; |
| 855 | defm : ImageSamplePatterns<int_amdgcn_image_sample_cd_cl_o, "IMAGE_SAMPLE_CD_CL_O">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 856 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 857 | // Sample with comparison and offsets. |
| 858 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_o, "IMAGE_SAMPLE_C_O">; |
| 859 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cl_o, "IMAGE_SAMPLE_C_CL_O">; |
| 860 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_o, "IMAGE_SAMPLE_C_D_O">; |
| 861 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_d_cl_o, "IMAGE_SAMPLE_C_D_CL_O">; |
| 862 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_l_o, "IMAGE_SAMPLE_C_L_O">; |
| 863 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_o, "IMAGE_SAMPLE_C_B_O">; |
| 864 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_b_cl_o, "IMAGE_SAMPLE_C_B_CL_O">; |
| 865 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_lz_o, "IMAGE_SAMPLE_C_LZ_O">; |
| 866 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_o, "IMAGE_SAMPLE_C_CD_O">; |
| 867 | defm : ImageSamplePatterns<int_amdgcn_image_sample_c_cd_cl_o, "IMAGE_SAMPLE_C_CD_CL_O">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 868 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 869 | // Basic gather4. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 870 | defm : ImageGather4Patterns<int_amdgcn_image_gather4, "IMAGE_GATHER4">; |
| 871 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_cl, "IMAGE_GATHER4_CL">; |
| 872 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_l, "IMAGE_GATHER4_L">; |
| 873 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_b, "IMAGE_GATHER4_B">; |
| 874 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_cl, "IMAGE_GATHER4_B_CL">; |
| 875 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_lz, "IMAGE_GATHER4_LZ">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 876 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 877 | // Gather4 with comparison. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 878 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c, "IMAGE_GATHER4_C">; |
| 879 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_cl, "IMAGE_GATHER4_C_CL">; |
| 880 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_l, "IMAGE_GATHER4_C_L">; |
| 881 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b, "IMAGE_GATHER4_C_B">; |
| 882 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_cl, "IMAGE_GATHER4_C_B_CL">; |
| 883 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_lz, "IMAGE_GATHER4_C_LZ">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 884 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 885 | // Gather4 with offsets. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 886 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_o, "IMAGE_GATHER4_O">; |
| 887 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_cl_o, "IMAGE_GATHER4_CL_O">; |
| 888 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_l_o, "IMAGE_GATHER4_L_O">; |
| 889 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_o, "IMAGE_GATHER4_B_O">; |
| 890 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_b_cl_o, "IMAGE_GATHER4_B_CL_O">; |
| 891 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_lz_o, "IMAGE_GATHER4_LZ_O">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 892 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 893 | // Gather4 with comparison and offsets. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 894 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_o, "IMAGE_GATHER4_C_O">; |
| 895 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_cl_o, "IMAGE_GATHER4_C_CL_O">; |
| 896 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_l_o, "IMAGE_GATHER4_C_L_O">; |
| 897 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_o, "IMAGE_GATHER4_C_B_O">; |
| 898 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_b_cl_o, "IMAGE_GATHER4_C_B_CL_O">; |
| 899 | defm : ImageGather4Patterns<int_amdgcn_image_gather4_c_lz_o, "IMAGE_GATHER4_C_LZ_O">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 900 | |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 901 | // Basic sample alternative. |
| 902 | defm : ImageSampleAltPatterns<SIImage_sample, "IMAGE_SAMPLE">; |
| 903 | defm : ImageSampleAltPatterns<SIImage_sample_cl, "IMAGE_SAMPLE_CL">; |
| 904 | defm : ImageSampleAltPatterns<SIImage_sample_d, "IMAGE_SAMPLE_D">; |
| 905 | defm : ImageSampleAltPatterns<SIImage_sample_d_cl, "IMAGE_SAMPLE_D_CL">; |
| 906 | defm : ImageSampleAltPatterns<SIImage_sample_l, "IMAGE_SAMPLE_L">; |
| 907 | defm : ImageSampleAltPatterns<SIImage_sample_b, "IMAGE_SAMPLE_B">; |
| 908 | defm : ImageSampleAltPatterns<SIImage_sample_b_cl, "IMAGE_SAMPLE_B_CL">; |
| 909 | defm : ImageSampleAltPatterns<SIImage_sample_lz, "IMAGE_SAMPLE_LZ">; |
| 910 | defm : ImageSampleAltPatterns<SIImage_sample_cd, "IMAGE_SAMPLE_CD">; |
| 911 | defm : ImageSampleAltPatterns<SIImage_sample_cd_cl, "IMAGE_SAMPLE_CD_CL">; |
| 912 | |
| 913 | // Sample with comparison alternative. |
| 914 | defm : ImageSampleAltPatterns<SIImage_sample_c, "IMAGE_SAMPLE_C">; |
| 915 | defm : ImageSampleAltPatterns<SIImage_sample_c_cl, "IMAGE_SAMPLE_C_CL">; |
| 916 | defm : ImageSampleAltPatterns<SIImage_sample_c_d, "IMAGE_SAMPLE_C_D">; |
| 917 | defm : ImageSampleAltPatterns<SIImage_sample_c_d_cl, "IMAGE_SAMPLE_C_D_CL">; |
| 918 | defm : ImageSampleAltPatterns<SIImage_sample_c_l, "IMAGE_SAMPLE_C_L">; |
| 919 | defm : ImageSampleAltPatterns<SIImage_sample_c_b, "IMAGE_SAMPLE_C_B">; |
| 920 | defm : ImageSampleAltPatterns<SIImage_sample_c_b_cl, "IMAGE_SAMPLE_C_B_CL">; |
| 921 | defm : ImageSampleAltPatterns<SIImage_sample_c_lz, "IMAGE_SAMPLE_C_LZ">; |
| 922 | defm : ImageSampleAltPatterns<SIImage_sample_c_cd, "IMAGE_SAMPLE_C_CD">; |
| 923 | defm : ImageSampleAltPatterns<SIImage_sample_c_cd_cl, "IMAGE_SAMPLE_C_CD_CL">; |
| 924 | |
| 925 | // Sample with offsets alternative. |
| 926 | defm : ImageSampleAltPatterns<SIImage_sample_o, "IMAGE_SAMPLE_O">; |
| 927 | defm : ImageSampleAltPatterns<SIImage_sample_cl_o, "IMAGE_SAMPLE_CL_O">; |
| 928 | defm : ImageSampleAltPatterns<SIImage_sample_d_o, "IMAGE_SAMPLE_D_O">; |
| 929 | defm : ImageSampleAltPatterns<SIImage_sample_d_cl_o, "IMAGE_SAMPLE_D_CL_O">; |
| 930 | defm : ImageSampleAltPatterns<SIImage_sample_l_o, "IMAGE_SAMPLE_L_O">; |
| 931 | defm : ImageSampleAltPatterns<SIImage_sample_b_o, "IMAGE_SAMPLE_B_O">; |
| 932 | defm : ImageSampleAltPatterns<SIImage_sample_b_cl_o, "IMAGE_SAMPLE_B_CL_O">; |
| 933 | defm : ImageSampleAltPatterns<SIImage_sample_lz_o, "IMAGE_SAMPLE_LZ_O">; |
| 934 | defm : ImageSampleAltPatterns<SIImage_sample_cd_o, "IMAGE_SAMPLE_CD_O">; |
| 935 | defm : ImageSampleAltPatterns<SIImage_sample_cd_cl_o, "IMAGE_SAMPLE_CD_CL_O">; |
| 936 | |
| 937 | // Sample with comparison and offsets alternative. |
| 938 | defm : ImageSampleAltPatterns<SIImage_sample_c_o, "IMAGE_SAMPLE_C_O">; |
| 939 | defm : ImageSampleAltPatterns<SIImage_sample_c_cl_o, "IMAGE_SAMPLE_C_CL_O">; |
| 940 | defm : ImageSampleAltPatterns<SIImage_sample_c_d_o, "IMAGE_SAMPLE_C_D_O">; |
| 941 | defm : ImageSampleAltPatterns<SIImage_sample_c_d_cl_o, "IMAGE_SAMPLE_C_D_CL_O">; |
| 942 | defm : ImageSampleAltPatterns<SIImage_sample_c_l_o, "IMAGE_SAMPLE_C_L_O">; |
| 943 | defm : ImageSampleAltPatterns<SIImage_sample_c_b_o, "IMAGE_SAMPLE_C_B_O">; |
| 944 | defm : ImageSampleAltPatterns<SIImage_sample_c_b_cl_o, "IMAGE_SAMPLE_C_B_CL_O">; |
| 945 | defm : ImageSampleAltPatterns<SIImage_sample_c_lz_o, "IMAGE_SAMPLE_C_LZ_O">; |
| 946 | defm : ImageSampleAltPatterns<SIImage_sample_c_cd_o, "IMAGE_SAMPLE_C_CD_O">; |
| 947 | defm : ImageSampleAltPatterns<SIImage_sample_c_cd_cl_o, "IMAGE_SAMPLE_C_CD_CL_O">; |
| 948 | |
| 949 | // Basic gather4 alternative. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 950 | defm : ImageGather4AltPatterns<SIImage_gather4, "IMAGE_GATHER4">; |
| 951 | defm : ImageGather4AltPatterns<SIImage_gather4_cl, "IMAGE_GATHER4_CL">; |
| 952 | defm : ImageGather4AltPatterns<SIImage_gather4_l, "IMAGE_GATHER4_L">; |
| 953 | defm : ImageGather4AltPatterns<SIImage_gather4_b, "IMAGE_GATHER4_B">; |
| 954 | defm : ImageGather4AltPatterns<SIImage_gather4_b_cl, "IMAGE_GATHER4_B_CL">; |
| 955 | defm : ImageGather4AltPatterns<SIImage_gather4_lz, "IMAGE_GATHER4_LZ">; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 956 | |
| 957 | // Gather4 with comparison alternative. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 958 | defm : ImageGather4AltPatterns<SIImage_gather4_c, "IMAGE_GATHER4_C">; |
| 959 | defm : ImageGather4AltPatterns<SIImage_gather4_c_cl, "IMAGE_GATHER4_C_CL">; |
| 960 | defm : ImageGather4AltPatterns<SIImage_gather4_c_l, "IMAGE_GATHER4_C_L">; |
| 961 | defm : ImageGather4AltPatterns<SIImage_gather4_c_b, "IMAGE_GATHER4_C_B">; |
| 962 | defm : ImageGather4AltPatterns<SIImage_gather4_c_b_cl, "IMAGE_GATHER4_C_B_CL">; |
| 963 | defm : ImageGather4AltPatterns<SIImage_gather4_c_lz, "IMAGE_GATHER4_C_LZ">; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 964 | |
| 965 | // Gather4 with offsets alternative. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 966 | defm : ImageGather4AltPatterns<SIImage_gather4_o, "IMAGE_GATHER4_O">; |
| 967 | defm : ImageGather4AltPatterns<SIImage_gather4_cl_o, "IMAGE_GATHER4_CL_O">; |
| 968 | defm : ImageGather4AltPatterns<SIImage_gather4_l_o, "IMAGE_GATHER4_L_O">; |
| 969 | defm : ImageGather4AltPatterns<SIImage_gather4_b_o, "IMAGE_GATHER4_B_O">; |
| 970 | defm : ImageGather4AltPatterns<SIImage_gather4_b_cl_o, "IMAGE_GATHER4_B_CL_O">; |
| 971 | defm : ImageGather4AltPatterns<SIImage_gather4_lz_o, "IMAGE_GATHER4_LZ_O">; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 972 | |
| 973 | // Gather4 with comparison and offsets alternative. |
| Dmitry Preobrazhensky | da4a7c0 | 2018-03-12 15:03:34 +0000 | [diff] [blame] | 974 | defm : ImageGather4AltPatterns<SIImage_gather4_c_o, "IMAGE_GATHER4_C_O">; |
| 975 | defm : ImageGather4AltPatterns<SIImage_gather4_c_cl_o, "IMAGE_GATHER4_C_CL_O">; |
| 976 | defm : ImageGather4AltPatterns<SIImage_gather4_c_l_o, "IMAGE_GATHER4_C_L_O">; |
| 977 | defm : ImageGather4AltPatterns<SIImage_gather4_c_b_o, "IMAGE_GATHER4_C_B_O">; |
| 978 | defm : ImageGather4AltPatterns<SIImage_gather4_c_b_cl_o, "IMAGE_GATHER4_C_B_CL_O">; |
| 979 | defm : ImageGather4AltPatterns<SIImage_gather4_c_lz_o, "IMAGE_GATHER4_C_LZ_O">; |
| Changpeng Fang | 4737e89 | 2018-01-18 22:08:53 +0000 | [diff] [blame] | 980 | |
| 981 | defm : ImageSamplePatterns<int_amdgcn_image_getlod, "IMAGE_GET_LOD">; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 982 | |
| 983 | // Image atomics |
| 984 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_swap, "IMAGE_ATOMIC_SWAP">; |
| Dmitry Preobrazhensky | 0b4eb1e | 2018-01-26 15:43:29 +0000 | [diff] [blame] | 985 | def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V1, i32>; |
| 986 | def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V2, v2i32>; |
| 987 | def : ImageAtomicCmpSwapPattern<IMAGE_ATOMIC_CMPSWAP_V1_V4, v4i32>; |
| Changpeng Fang | b28fe03 | 2016-09-01 17:54:54 +0000 | [diff] [blame] | 988 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_add, "IMAGE_ATOMIC_ADD">; |
| 989 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_sub, "IMAGE_ATOMIC_SUB">; |
| 990 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_smin, "IMAGE_ATOMIC_SMIN">; |
| 991 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_umin, "IMAGE_ATOMIC_UMIN">; |
| 992 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_smax, "IMAGE_ATOMIC_SMAX">; |
| 993 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_umax, "IMAGE_ATOMIC_UMAX">; |
| 994 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_and, "IMAGE_ATOMIC_AND">; |
| 995 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_or, "IMAGE_ATOMIC_OR">; |
| 996 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_xor, "IMAGE_ATOMIC_XOR">; |
| 997 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_inc, "IMAGE_ATOMIC_INC">; |
| 998 | defm : ImageAtomicPatterns<int_amdgcn_image_atomic_dec, "IMAGE_ATOMIC_DEC">; |