| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 1 | //===-- AMDGPUInstructions.td - Common instruction defs ---*- tablegen -*-===// |
| 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 | // This file contains instruction defs that are common to all hw codegen |
| 11 | // targets. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| Matt Arsenault | 648e422 | 2016-07-14 05:23:23 +0000 | [diff] [blame] | 15 | class AMDGPUInst <dag outs, dag ins, string asm = "", |
| 16 | list<dag> pattern = []> : Instruction { |
| Tom Stellard | f3b2a1e | 2013-02-06 17:32:29 +0000 | [diff] [blame] | 17 | field bit isRegisterLoad = 0; |
| 18 | field bit isRegisterStore = 0; |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 19 | |
| 20 | let Namespace = "AMDGPU"; |
| 21 | let OutOperandList = outs; |
| 22 | let InOperandList = ins; |
| 23 | let AsmString = asm; |
| 24 | let Pattern = pattern; |
| 25 | let Itinerary = NullALU; |
| Tom Stellard | f3b2a1e | 2013-02-06 17:32:29 +0000 | [diff] [blame] | 26 | |
| Tom Stellard | e1818af | 2016-02-18 03:42:32 +0000 | [diff] [blame] | 27 | // SoftFail is a field the disassembler can use to provide a way for |
| 28 | // instructions to not match without killing the whole decode process. It is |
| 29 | // mainly used for ARM, but Tablegen expects this field to exist or it fails |
| 30 | // to build the decode table. |
| 31 | field bits<64> SoftFail = 0; |
| 32 | |
| 33 | let DecoderNamespace = Namespace; |
| Matt Arsenault | 37fefd6 | 2016-06-10 02:18:02 +0000 | [diff] [blame] | 34 | |
| Tom Stellard | f3b2a1e | 2013-02-06 17:32:29 +0000 | [diff] [blame] | 35 | let TSFlags{63} = isRegisterLoad; |
| 36 | let TSFlags{62} = isRegisterStore; |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 37 | } |
| 38 | |
| Matt Arsenault | 648e422 | 2016-07-14 05:23:23 +0000 | [diff] [blame] | 39 | class AMDGPUShaderInst <dag outs, dag ins, string asm = "", |
| 40 | list<dag> pattern = []> : AMDGPUInst<outs, ins, asm, pattern> { |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 41 | |
| 42 | field bits<32> Inst = 0xffffffff; |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 43 | } |
| 44 | |
| Stanislav Mekhanoshin | 06cab79 | 2017-08-30 03:03:38 +0000 | [diff] [blame] | 45 | def FP16Denormals : Predicate<"Subtarget->hasFP16Denormals()">; |
| 46 | def FP32Denormals : Predicate<"Subtarget->hasFP32Denormals()">; |
| 47 | def FP64Denormals : Predicate<"Subtarget->hasFP64Denormals()">; |
| 48 | def NoFP16Denormals : Predicate<"!Subtarget->hasFP16Denormals()">; |
| 49 | def NoFP32Denormals : Predicate<"!Subtarget->hasFP32Denormals()">; |
| 50 | def NoFP64Denormals : Predicate<"!Subtarget->hasFP64Denormals()">; |
| Matt Arsenault | 1d07774 | 2014-07-15 20:18:24 +0000 | [diff] [blame] | 51 | def UnsafeFPMath : Predicate<"TM.Options.UnsafeFPMath">; |
| Jan Vesely | 39aeab4 | 2017-12-04 23:07:28 +0000 | [diff] [blame] | 52 | def FMA : Predicate<"Subtarget->hasFMA()">; |
| Matt Arsenault | f171cf2 | 2014-07-14 23:40:49 +0000 | [diff] [blame] | 53 | |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 54 | def InstFlag : OperandWithDefaultOps <i32, (ops (i32 0))>; |
| 55 | |
| Dmitry Preobrazhensky | c7d35a0 | 2017-04-26 15:34:19 +0000 | [diff] [blame] | 56 | def u16ImmTarget : AsmOperandClass { |
| 57 | let Name = "U16Imm"; |
| 58 | let RenderMethod = "addImmOperands"; |
| 59 | } |
| 60 | |
| 61 | def s16ImmTarget : AsmOperandClass { |
| 62 | let Name = "S16Imm"; |
| 63 | let RenderMethod = "addImmOperands"; |
| 64 | } |
| 65 | |
| Tom Stellard | b02094e | 2014-07-21 15:45:01 +0000 | [diff] [blame] | 66 | let OperandType = "OPERAND_IMMEDIATE" in { |
| 67 | |
| Matt Arsenault | 4d7d383 | 2014-04-15 22:32:49 +0000 | [diff] [blame] | 68 | def u32imm : Operand<i32> { |
| 69 | let PrintMethod = "printU32ImmOperand"; |
| 70 | } |
| 71 | |
| 72 | def u16imm : Operand<i16> { |
| 73 | let PrintMethod = "printU16ImmOperand"; |
| Dmitry Preobrazhensky | c7d35a0 | 2017-04-26 15:34:19 +0000 | [diff] [blame] | 74 | let ParserMatchClass = u16ImmTarget; |
| 75 | } |
| 76 | |
| 77 | def s16imm : Operand<i16> { |
| 78 | let PrintMethod = "printU16ImmOperand"; |
| 79 | let ParserMatchClass = s16ImmTarget; |
| Matt Arsenault | 4d7d383 | 2014-04-15 22:32:49 +0000 | [diff] [blame] | 80 | } |
| 81 | |
| 82 | def u8imm : Operand<i8> { |
| 83 | let PrintMethod = "printU8ImmOperand"; |
| 84 | } |
| 85 | |
| Tom Stellard | b02094e | 2014-07-21 15:45:01 +0000 | [diff] [blame] | 86 | } // End OperandType = "OPERAND_IMMEDIATE" |
| 87 | |
| Tom Stellard | bc5b537 | 2014-06-13 16:38:59 +0000 | [diff] [blame] | 88 | //===--------------------------------------------------------------------===// |
| 89 | // Custom Operands |
| 90 | //===--------------------------------------------------------------------===// |
| 91 | def brtarget : Operand<OtherVT>; |
| 92 | |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 93 | //===----------------------------------------------------------------------===// |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 94 | // Misc. PatFrags |
| 95 | //===----------------------------------------------------------------------===// |
| 96 | |
| Matt Arsenault | eb522e6 | 2017-02-27 22:15:25 +0000 | [diff] [blame] | 97 | class HasOneUseUnaryOp<SDPatternOperator op> : PatFrag< |
| 98 | (ops node:$src0), |
| 99 | (op $src0), |
| 100 | [{ return N->hasOneUse(); }] |
| 101 | >; |
| 102 | |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 103 | class HasOneUseBinOp<SDPatternOperator op> : PatFrag< |
| 104 | (ops node:$src0, node:$src1), |
| 105 | (op $src0, $src1), |
| 106 | [{ return N->hasOneUse(); }] |
| 107 | >; |
| 108 | |
| 109 | class HasOneUseTernaryOp<SDPatternOperator op> : PatFrag< |
| 110 | (ops node:$src0, node:$src1, node:$src2), |
| 111 | (op $src0, $src1, $src2), |
| 112 | [{ return N->hasOneUse(); }] |
| 113 | >; |
| 114 | |
| Matt Arsenault | eb522e6 | 2017-02-27 22:15:25 +0000 | [diff] [blame] | 115 | def trunc_oneuse : HasOneUseUnaryOp<trunc>; |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 116 | |
| 117 | let Properties = [SDNPCommutative, SDNPAssociative] in { |
| 118 | def smax_oneuse : HasOneUseBinOp<smax>; |
| 119 | def smin_oneuse : HasOneUseBinOp<smin>; |
| 120 | def umax_oneuse : HasOneUseBinOp<umax>; |
| 121 | def umin_oneuse : HasOneUseBinOp<umin>; |
| 122 | def fminnum_oneuse : HasOneUseBinOp<fminnum>; |
| 123 | def fmaxnum_oneuse : HasOneUseBinOp<fmaxnum>; |
| 124 | def and_oneuse : HasOneUseBinOp<and>; |
| 125 | def or_oneuse : HasOneUseBinOp<or>; |
| 126 | def xor_oneuse : HasOneUseBinOp<xor>; |
| 127 | } // Properties = [SDNPCommutative, SDNPAssociative] |
| 128 | |
| Roman Lebedev | 9c17dad | 2018-06-15 09:56:39 +0000 | [diff] [blame] | 129 | def add_oneuse : HasOneUseBinOp<add>; |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 130 | def sub_oneuse : HasOneUseBinOp<sub>; |
| Matt Arsenault | eb522e6 | 2017-02-27 22:15:25 +0000 | [diff] [blame] | 131 | |
| 132 | def srl_oneuse : HasOneUseBinOp<srl>; |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 133 | def shl_oneuse : HasOneUseBinOp<shl>; |
| 134 | |
| 135 | def select_oneuse : HasOneUseTernaryOp<select>; |
| 136 | |
| Matt Arsenault | e1cd482 | 2017-11-13 00:22:09 +0000 | [diff] [blame] | 137 | def srl_16 : PatFrag< |
| 138 | (ops node:$src0), (srl_oneuse node:$src0, (i32 16)) |
| 139 | >; |
| 140 | |
| 141 | |
| 142 | def hi_i16_elt : PatFrag< |
| 143 | (ops node:$src0), (i16 (trunc (i32 (srl_16 node:$src0)))) |
| 144 | >; |
| 145 | |
| 146 | |
| 147 | def hi_f16_elt : PatLeaf< |
| 148 | (vt), [{ |
| 149 | if (N->getOpcode() != ISD::BITCAST) |
| 150 | return false; |
| 151 | SDValue Tmp = N->getOperand(0); |
| 152 | |
| 153 | if (Tmp.getOpcode() != ISD::SRL) |
| 154 | return false; |
| 155 | if (const auto *RHS = dyn_cast<ConstantSDNode>(Tmp.getOperand(1)) |
| 156 | return RHS->getZExtValue() == 16; |
| 157 | return false; |
| 158 | }]>; |
| 159 | |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 160 | //===----------------------------------------------------------------------===// |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 161 | // PatLeafs for floating-point comparisons |
| 162 | //===----------------------------------------------------------------------===// |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 163 | |
| Tom Stellard | 0351ea2 | 2013-09-28 02:50:50 +0000 | [diff] [blame] | 164 | def COND_OEQ : PatLeaf < |
| 165 | (cond), |
| 166 | [{return N->get() == ISD::SETOEQ || N->get() == ISD::SETEQ;}] |
| 167 | >; |
| 168 | |
| Matt Arsenault | 9cded7a | 2014-12-11 22:15:35 +0000 | [diff] [blame] | 169 | def COND_ONE : PatLeaf < |
| 170 | (cond), |
| 171 | [{return N->get() == ISD::SETONE || N->get() == ISD::SETNE;}] |
| 172 | >; |
| 173 | |
| Tom Stellard | 0351ea2 | 2013-09-28 02:50:50 +0000 | [diff] [blame] | 174 | def COND_OGT : PatLeaf < |
| 175 | (cond), |
| 176 | [{return N->get() == ISD::SETOGT || N->get() == ISD::SETGT;}] |
| 177 | >; |
| 178 | |
| Tom Stellard | 0351ea2 | 2013-09-28 02:50:50 +0000 | [diff] [blame] | 179 | def COND_OGE : PatLeaf < |
| 180 | (cond), |
| 181 | [{return N->get() == ISD::SETOGE || N->get() == ISD::SETGE;}] |
| 182 | >; |
| 183 | |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 184 | def COND_OLT : PatLeaf < |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 185 | (cond), |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 186 | [{return N->get() == ISD::SETOLT || N->get() == ISD::SETLT;}] |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 187 | >; |
| 188 | |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 189 | def COND_OLE : PatLeaf < |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 190 | (cond), |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 191 | [{return N->get() == ISD::SETOLE || N->get() == ISD::SETLE;}] |
| 192 | >; |
| 193 | |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 194 | def COND_O : PatLeaf <(cond), [{return N->get() == ISD::SETO;}]>; |
| 195 | def COND_UO : PatLeaf <(cond), [{return N->get() == ISD::SETUO;}]>; |
| 196 | |
| 197 | //===----------------------------------------------------------------------===// |
| Matt Arsenault | 8b989ef | 2014-12-11 22:15:39 +0000 | [diff] [blame] | 198 | // PatLeafs for unsigned / unordered comparisons |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 199 | //===----------------------------------------------------------------------===// |
| 200 | |
| Matt Arsenault | 9cded7a | 2014-12-11 22:15:35 +0000 | [diff] [blame] | 201 | def COND_UEQ : PatLeaf <(cond), [{return N->get() == ISD::SETUEQ;}]>; |
| 202 | def COND_UNE : PatLeaf <(cond), [{return N->get() == ISD::SETUNE;}]>; |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 203 | def COND_UGT : PatLeaf <(cond), [{return N->get() == ISD::SETUGT;}]>; |
| 204 | def COND_UGE : PatLeaf <(cond), [{return N->get() == ISD::SETUGE;}]>; |
| 205 | def COND_ULT : PatLeaf <(cond), [{return N->get() == ISD::SETULT;}]>; |
| 206 | def COND_ULE : PatLeaf <(cond), [{return N->get() == ISD::SETULE;}]>; |
| 207 | |
| Matt Arsenault | 9cded7a | 2014-12-11 22:15:35 +0000 | [diff] [blame] | 208 | // XXX - For some reason R600 version is preferring to use unordered |
| 209 | // for setne? |
| 210 | def COND_UNE_NE : PatLeaf < |
| 211 | (cond), |
| 212 | [{return N->get() == ISD::SETUNE || N->get() == ISD::SETNE;}] |
| 213 | >; |
| 214 | |
| Tom Stellard | c084533 | 2013-11-22 23:07:58 +0000 | [diff] [blame] | 215 | //===----------------------------------------------------------------------===// |
| 216 | // PatLeafs for signed comparisons |
| 217 | //===----------------------------------------------------------------------===// |
| 218 | |
| 219 | def COND_SGT : PatLeaf <(cond), [{return N->get() == ISD::SETGT;}]>; |
| 220 | def COND_SGE : PatLeaf <(cond), [{return N->get() == ISD::SETGE;}]>; |
| 221 | def COND_SLT : PatLeaf <(cond), [{return N->get() == ISD::SETLT;}]>; |
| 222 | def COND_SLE : PatLeaf <(cond), [{return N->get() == ISD::SETLE;}]>; |
| 223 | |
| 224 | //===----------------------------------------------------------------------===// |
| 225 | // PatLeafs for integer equality |
| 226 | //===----------------------------------------------------------------------===// |
| 227 | |
| 228 | def COND_EQ : PatLeaf < |
| 229 | (cond), |
| 230 | [{return N->get() == ISD::SETEQ || N->get() == ISD::SETUEQ;}] |
| 231 | >; |
| 232 | |
| 233 | def COND_NE : PatLeaf < |
| 234 | (cond), |
| 235 | [{return N->get() == ISD::SETNE || N->get() == ISD::SETUNE;}] |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 236 | >; |
| 237 | |
| Christian Konig | b19849a | 2013-02-21 15:17:04 +0000 | [diff] [blame] | 238 | def COND_NULL : PatLeaf < |
| 239 | (cond), |
| Tom Stellard | aa9a1a8 | 2014-08-01 02:05:57 +0000 | [diff] [blame] | 240 | [{(void)N; return false;}] |
| Christian Konig | b19849a | 2013-02-21 15:17:04 +0000 | [diff] [blame] | 241 | >; |
| 242 | |
| Matt Arsenault | c89f291 | 2016-03-07 21:54:48 +0000 | [diff] [blame] | 243 | |
| 244 | //===----------------------------------------------------------------------===// |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 245 | // Load/Store Pattern Fragments |
| 246 | //===----------------------------------------------------------------------===// |
| 247 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 248 | class Aligned8Bytes <dag ops, dag frag> : PatFrag <ops, frag, [{ |
| 249 | return cast<MemSDNode>(N)->getAlignment() % 8 == 0; |
| 250 | }]>; |
| 251 | |
| Farhana Aleen | a7cb311 | 2018-03-09 17:41:39 +0000 | [diff] [blame] | 252 | class Aligned16Bytes <dag ops, dag frag> : PatFrag <ops, frag, [{ |
| 253 | return cast<MemSDNode>(N)->getAlignment() >= 16; |
| 254 | }]>; |
| 255 | |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 256 | class LoadFrag <SDPatternOperator op> : PatFrag<(ops node:$ptr), (op node:$ptr)>; |
| Tom Stellard | b02094e | 2014-07-21 15:45:01 +0000 | [diff] [blame] | 257 | |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 258 | class StoreFrag<SDPatternOperator op> : PatFrag < |
| Tom Stellard | b02094e | 2014-07-21 15:45:01 +0000 | [diff] [blame] | 259 | (ops node:$value, node:$ptr), (op node:$value, node:$ptr) |
| 260 | >; |
| 261 | |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 262 | class StoreHi16<SDPatternOperator op> : PatFrag < |
| 263 | (ops node:$value, node:$ptr), (op (srl node:$value, (i32 16)), node:$ptr) |
| 264 | >; |
| Tom Stellard | b02094e | 2014-07-21 15:45:01 +0000 | [diff] [blame] | 265 | |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 266 | class PrivateAddress : CodePatPred<[{ |
| 267 | return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.PRIVATE_ADDRESS; |
| 268 | }]>; |
| 269 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 270 | class ConstantAddress : CodePatPred<[{ |
| 271 | return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.CONSTANT_ADDRESS; |
| 272 | }]>; |
| 273 | |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 274 | class LocalAddress : CodePatPred<[{ |
| 275 | return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.LOCAL_ADDRESS; |
| 276 | }]>; |
| 277 | |
| 278 | class GlobalAddress : CodePatPred<[{ |
| 279 | return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.GLOBAL_ADDRESS; |
| 280 | }]>; |
| 281 | |
| Matt Arsenault | b81495d | 2017-09-20 05:01:53 +0000 | [diff] [blame] | 282 | class GlobalLoadAddress : CodePatPred<[{ |
| 283 | auto AS = cast<MemSDNode>(N)->getAddressSpace(); |
| 284 | return AS == AMDGPUASI.GLOBAL_ADDRESS || AS == AMDGPUASI.CONSTANT_ADDRESS; |
| 285 | }]>; |
| 286 | |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 287 | class FlatLoadAddress : CodePatPred<[{ |
| 288 | const auto AS = cast<MemSDNode>(N)->getAddressSpace(); |
| 289 | return AS == AMDGPUASI.FLAT_ADDRESS || |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 290 | AS == AMDGPUASI.GLOBAL_ADDRESS || |
| 291 | AS == AMDGPUASI.CONSTANT_ADDRESS; |
| 292 | }]>; |
| 293 | |
| 294 | class FlatStoreAddress : CodePatPred<[{ |
| 295 | const auto AS = cast<MemSDNode>(N)->getAddressSpace(); |
| 296 | return AS == AMDGPUASI.FLAT_ADDRESS || |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 297 | AS == AMDGPUASI.GLOBAL_ADDRESS; |
| 298 | }]>; |
| 299 | |
| Tom Stellard | 381a94a | 2015-05-12 15:00:49 +0000 | [diff] [blame] | 300 | class AZExtLoadBase <SDPatternOperator ld_node>: PatFrag<(ops node:$ptr), |
| 301 | (ld_node node:$ptr), [{ |
| Tom Stellard | 31209cc | 2013-07-15 19:00:09 +0000 | [diff] [blame] | 302 | LoadSDNode *L = cast<LoadSDNode>(N); |
| 303 | return L->getExtensionType() == ISD::ZEXTLOAD || |
| 304 | L->getExtensionType() == ISD::EXTLOAD; |
| 305 | }]>; |
| 306 | |
| Tom Stellard | 381a94a | 2015-05-12 15:00:49 +0000 | [diff] [blame] | 307 | def az_extload : AZExtLoadBase <unindexedload>; |
| 308 | |
| Tom Stellard | 33dd04b | 2013-07-23 01:47:52 +0000 | [diff] [blame] | 309 | def az_extloadi8 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{ |
| 310 | return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i8; |
| 311 | }]>; |
| 312 | |
| Tom Stellard | 33dd04b | 2013-07-23 01:47:52 +0000 | [diff] [blame] | 313 | def az_extloadi16 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{ |
| 314 | return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i16; |
| 315 | }]>; |
| 316 | |
| Tom Stellard | 31209cc | 2013-07-15 19:00:09 +0000 | [diff] [blame] | 317 | def az_extloadi32 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{ |
| 318 | return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i32; |
| 319 | }]>; |
| 320 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 321 | class PrivateLoad <SDPatternOperator op> : LoadFrag <op>, PrivateAddress; |
| 322 | class PrivateStore <SDPatternOperator op> : StoreFrag <op>, PrivateAddress; |
| Tom Stellard | 31209cc | 2013-07-15 19:00:09 +0000 | [diff] [blame] | 323 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 324 | class LocalLoad <SDPatternOperator op> : LoadFrag <op>, LocalAddress; |
| 325 | class LocalStore <SDPatternOperator op> : StoreFrag <op>, LocalAddress; |
| Matt Arsenault | 3f98140 | 2014-09-15 15:41:53 +0000 | [diff] [blame] | 326 | |
| Matt Arsenault | b81495d | 2017-09-20 05:01:53 +0000 | [diff] [blame] | 327 | class GlobalLoad <SDPatternOperator op> : LoadFrag<op>, GlobalLoadAddress; |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 328 | class GlobalStore <SDPatternOperator op> : StoreFrag<op>, GlobalAddress; |
| Tom Stellard | 31209cc | 2013-07-15 19:00:09 +0000 | [diff] [blame] | 329 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 330 | class FlatLoad <SDPatternOperator op> : LoadFrag <op>, FlatLoadAddress; |
| 331 | class FlatStore <SDPatternOperator op> : StoreFrag <op>, FlatStoreAddress; |
| 332 | |
| 333 | class ConstantLoad <SDPatternOperator op> : LoadFrag <op>, ConstantAddress; |
| 334 | |
| 335 | |
| 336 | def load_private : PrivateLoad <load>; |
| 337 | def az_extloadi8_private : PrivateLoad <az_extloadi8>; |
| 338 | def sextloadi8_private : PrivateLoad <sextloadi8>; |
| 339 | def az_extloadi16_private : PrivateLoad <az_extloadi16>; |
| 340 | def sextloadi16_private : PrivateLoad <sextloadi16>; |
| 341 | |
| 342 | def store_private : PrivateStore <store>; |
| 343 | def truncstorei8_private : PrivateStore<truncstorei8>; |
| 344 | def truncstorei16_private : PrivateStore <truncstorei16>; |
| 345 | def store_hi16_private : StoreHi16 <truncstorei16>, PrivateAddress; |
| 346 | def truncstorei8_hi16_private : StoreHi16<truncstorei8>, PrivateAddress; |
| 347 | |
| 348 | |
| 349 | def load_global : GlobalLoad <load>; |
| 350 | def sextloadi8_global : GlobalLoad <sextloadi8>; |
| 351 | def az_extloadi8_global : GlobalLoad <az_extloadi8>; |
| 352 | def sextloadi16_global : GlobalLoad <sextloadi16>; |
| 353 | def az_extloadi16_global : GlobalLoad <az_extloadi16>; |
| 354 | def atomic_load_global : GlobalLoad<atomic_load>; |
| 355 | |
| 356 | def store_global : GlobalStore <store>; |
| Tom Stellard | a4b746d | 2016-07-05 16:10:44 +0000 | [diff] [blame] | 357 | def truncstorei8_global : GlobalStore <truncstorei8>; |
| 358 | def truncstorei16_global : GlobalStore <truncstorei16>; |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 359 | def store_atomic_global : GlobalStore<atomic_store>; |
| 360 | def truncstorei8_hi16_global : StoreHi16 <truncstorei8>, GlobalAddress; |
| 361 | def truncstorei16_hi16_global : StoreHi16 <truncstorei16>, GlobalAddress; |
| Tom Stellard | d3ee8c1 | 2013-08-16 01:12:06 +0000 | [diff] [blame] | 362 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 363 | def load_local : LocalLoad <load>; |
| 364 | def az_extloadi8_local : LocalLoad <az_extloadi8>; |
| 365 | def sextloadi8_local : LocalLoad <sextloadi8>; |
| 366 | def az_extloadi16_local : LocalLoad <az_extloadi16>; |
| 367 | def sextloadi16_local : LocalLoad <sextloadi16>; |
| Matt Arsenault | fcc213f | 2017-09-20 03:20:09 +0000 | [diff] [blame] | 368 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 369 | def store_local : LocalStore <store>; |
| 370 | def truncstorei8_local : LocalStore <truncstorei8>; |
| 371 | def truncstorei16_local : LocalStore <truncstorei16>; |
| 372 | def store_local_hi16 : StoreHi16 <truncstorei16>, LocalAddress; |
| 373 | def truncstorei8_local_hi16 : StoreHi16<truncstorei8>, LocalAddress; |
| Tom Stellard | d3ee8c1 | 2013-08-16 01:12:06 +0000 | [diff] [blame] | 374 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 375 | def load_align8_local : Aligned8Bytes < |
| 376 | (ops node:$ptr), (load_local node:$ptr) |
| Tom Stellard | f3fc555 | 2014-08-22 18:49:35 +0000 | [diff] [blame] | 377 | >; |
| 378 | |
| Farhana Aleen | a7cb311 | 2018-03-09 17:41:39 +0000 | [diff] [blame] | 379 | def load_align16_local : Aligned16Bytes < |
| 380 | (ops node:$ptr), (load_local node:$ptr) |
| 381 | >; |
| 382 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 383 | def store_align8_local : Aligned8Bytes < |
| 384 | (ops node:$val, node:$ptr), (store_local node:$val, node:$ptr) |
| Tom Stellard | f3fc555 | 2014-08-22 18:49:35 +0000 | [diff] [blame] | 385 | >; |
| Matt Arsenault | 7257410 | 2014-06-11 18:08:34 +0000 | [diff] [blame] | 386 | |
| Farhana Aleen | c6c9dc8 | 2018-03-16 18:12:00 +0000 | [diff] [blame] | 387 | def store_align16_local : Aligned16Bytes < |
| 388 | (ops node:$val, node:$ptr), (store_local node:$val, node:$ptr) |
| 389 | >; |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 390 | |
| 391 | def load_flat : FlatLoad <load>; |
| 392 | def az_extloadi8_flat : FlatLoad <az_extloadi8>; |
| 393 | def sextloadi8_flat : FlatLoad <sextloadi8>; |
| 394 | def az_extloadi16_flat : FlatLoad <az_extloadi16>; |
| 395 | def sextloadi16_flat : FlatLoad <sextloadi16>; |
| 396 | def atomic_load_flat : FlatLoad<atomic_load>; |
| 397 | |
| 398 | def store_flat : FlatStore <store>; |
| 399 | def truncstorei8_flat : FlatStore <truncstorei8>; |
| 400 | def truncstorei16_flat : FlatStore <truncstorei16>; |
| 401 | def atomic_store_flat : FlatStore <atomic_store>; |
| 402 | def truncstorei8_hi16_flat : StoreHi16<truncstorei8>, FlatStoreAddress; |
| 403 | def truncstorei16_hi16_flat : StoreHi16<truncstorei16>, FlatStoreAddress; |
| 404 | |
| 405 | |
| 406 | def constant_load : ConstantLoad<load>; |
| 407 | def sextloadi8_constant : ConstantLoad <sextloadi8>; |
| 408 | def az_extloadi8_constant : ConstantLoad <az_extloadi8>; |
| 409 | def sextloadi16_constant : ConstantLoad <sextloadi16>; |
| 410 | def az_extloadi16_constant : ConstantLoad <az_extloadi16>; |
| 411 | |
| 412 | |
| Matt Arsenault | 7257410 | 2014-06-11 18:08:34 +0000 | [diff] [blame] | 413 | class local_binary_atomic_op<SDNode atomic_op> : |
| 414 | PatFrag<(ops node:$ptr, node:$value), |
| 415 | (atomic_op node:$ptr, node:$value), [{ |
| Yaxun Liu | 1a14bfa | 2017-03-27 14:04:01 +0000 | [diff] [blame] | 416 | return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.LOCAL_ADDRESS; |
| Tom Stellard | 13c68ef | 2013-09-05 18:38:09 +0000 | [diff] [blame] | 417 | }]>; |
| 418 | |
| Matt Arsenault | 7257410 | 2014-06-11 18:08:34 +0000 | [diff] [blame] | 419 | def atomic_swap_local : local_binary_atomic_op<atomic_swap>; |
| 420 | def atomic_load_add_local : local_binary_atomic_op<atomic_load_add>; |
| 421 | def atomic_load_sub_local : local_binary_atomic_op<atomic_load_sub>; |
| 422 | def atomic_load_and_local : local_binary_atomic_op<atomic_load_and>; |
| 423 | def atomic_load_or_local : local_binary_atomic_op<atomic_load_or>; |
| 424 | def atomic_load_xor_local : local_binary_atomic_op<atomic_load_xor>; |
| 425 | def atomic_load_nand_local : local_binary_atomic_op<atomic_load_nand>; |
| 426 | def atomic_load_min_local : local_binary_atomic_op<atomic_load_min>; |
| 427 | def atomic_load_max_local : local_binary_atomic_op<atomic_load_max>; |
| 428 | def atomic_load_umin_local : local_binary_atomic_op<atomic_load_umin>; |
| 429 | def atomic_load_umax_local : local_binary_atomic_op<atomic_load_umax>; |
| Aaron Watry | 372cecf | 2013-09-06 20:17:42 +0000 | [diff] [blame] | 430 | |
| Tom Stellard | d3ee8c1 | 2013-08-16 01:12:06 +0000 | [diff] [blame] | 431 | def mskor_global : PatFrag<(ops node:$val, node:$ptr), |
| 432 | (AMDGPUstore_mskor node:$val, node:$ptr), [{ |
| Yaxun Liu | 1a14bfa | 2017-03-27 14:04:01 +0000 | [diff] [blame] | 433 | return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.GLOBAL_ADDRESS; |
| Tom Stellard | d3ee8c1 | 2013-08-16 01:12:06 +0000 | [diff] [blame] | 434 | }]>; |
| 435 | |
| Matt Arsenault | a030e26 | 2017-10-23 17:16:43 +0000 | [diff] [blame] | 436 | class AtomicCmpSwapLocal <SDNode cmp_swap_node> : PatFrag< |
| Tom Stellard | 381a94a | 2015-05-12 15:00:49 +0000 | [diff] [blame] | 437 | (ops node:$ptr, node:$cmp, node:$swap), |
| 438 | (cmp_swap_node node:$ptr, node:$cmp, node:$swap), [{ |
| 439 | AtomicSDNode *AN = cast<AtomicSDNode>(N); |
| Matt Arsenault | a030e26 | 2017-10-23 17:16:43 +0000 | [diff] [blame] | 440 | return AN->getAddressSpace() == AMDGPUASI.LOCAL_ADDRESS; |
| 441 | }]>; |
| Matt Arsenault | c793e1d | 2014-06-11 18:08:48 +0000 | [diff] [blame] | 442 | |
| Matt Arsenault | a030e26 | 2017-10-23 17:16:43 +0000 | [diff] [blame] | 443 | def atomic_cmp_swap_local : AtomicCmpSwapLocal <atomic_cmp_swap>; |
| Matt Arsenault | caa0ec2 | 2014-06-11 18:08:54 +0000 | [diff] [blame] | 444 | |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 445 | multiclass global_binary_atomic_op<SDNode atomic_op> { |
| 446 | def "" : PatFrag< |
| 447 | (ops node:$ptr, node:$value), |
| 448 | (atomic_op node:$ptr, node:$value), |
| Yaxun Liu | 1a14bfa | 2017-03-27 14:04:01 +0000 | [diff] [blame] | 449 | [{return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.GLOBAL_ADDRESS;}]>; |
| Tom Stellard | 7980fc8 | 2014-09-25 18:30:26 +0000 | [diff] [blame] | 450 | |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 451 | def _noret : PatFrag< |
| 452 | (ops node:$ptr, node:$value), |
| 453 | (atomic_op node:$ptr, node:$value), |
| Yaxun Liu | 1a14bfa | 2017-03-27 14:04:01 +0000 | [diff] [blame] | 454 | [{return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.GLOBAL_ADDRESS && (SDValue(N, 0).use_empty());}]>; |
| Tom Stellard | 7980fc8 | 2014-09-25 18:30:26 +0000 | [diff] [blame] | 455 | |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 456 | def _ret : PatFrag< |
| 457 | (ops node:$ptr, node:$value), |
| 458 | (atomic_op node:$ptr, node:$value), |
| Yaxun Liu | 1a14bfa | 2017-03-27 14:04:01 +0000 | [diff] [blame] | 459 | [{return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.GLOBAL_ADDRESS && (!SDValue(N, 0).use_empty());}]>; |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 460 | } |
| 461 | |
| 462 | defm atomic_swap_global : global_binary_atomic_op<atomic_swap>; |
| 463 | defm atomic_add_global : global_binary_atomic_op<atomic_load_add>; |
| 464 | defm atomic_and_global : global_binary_atomic_op<atomic_load_and>; |
| 465 | defm atomic_max_global : global_binary_atomic_op<atomic_load_max>; |
| 466 | defm atomic_min_global : global_binary_atomic_op<atomic_load_min>; |
| 467 | defm atomic_or_global : global_binary_atomic_op<atomic_load_or>; |
| 468 | defm atomic_sub_global : global_binary_atomic_op<atomic_load_sub>; |
| 469 | defm atomic_umax_global : global_binary_atomic_op<atomic_load_umax>; |
| 470 | defm atomic_umin_global : global_binary_atomic_op<atomic_load_umin>; |
| 471 | defm atomic_xor_global : global_binary_atomic_op<atomic_load_xor>; |
| 472 | |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 473 | // Legacy. |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 474 | def AMDGPUatomic_cmp_swap_global : PatFrag< |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 475 | (ops node:$ptr, node:$value), |
| 476 | (AMDGPUatomic_cmp_swap node:$ptr, node:$value)>, GlobalAddress; |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 477 | |
| 478 | def atomic_cmp_swap_global : PatFrag< |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 479 | (ops node:$ptr, node:$cmp, node:$value), |
| 480 | (atomic_cmp_swap node:$ptr, node:$cmp, node:$value)>, GlobalAddress; |
| 481 | |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 482 | |
| 483 | def atomic_cmp_swap_global_noret : PatFrag< |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 484 | (ops node:$ptr, node:$cmp, node:$value), |
| 485 | (atomic_cmp_swap node:$ptr, node:$cmp, node:$value), |
| 486 | [{return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.GLOBAL_ADDRESS && (SDValue(N, 0).use_empty());}]>; |
| Jan Vesely | 206a510 | 2016-12-23 15:34:51 +0000 | [diff] [blame] | 487 | |
| 488 | def atomic_cmp_swap_global_ret : PatFrag< |
| Matt Arsenault | bc68383 | 2017-09-20 03:43:35 +0000 | [diff] [blame] | 489 | (ops node:$ptr, node:$cmp, node:$value), |
| 490 | (atomic_cmp_swap node:$ptr, node:$cmp, node:$value), |
| 491 | [{return cast<MemSDNode>(N)->getAddressSpace() == AMDGPUASI.GLOBAL_ADDRESS && (!SDValue(N, 0).use_empty());}]>; |
| Tom Stellard | 354a43c | 2016-04-01 18:27:37 +0000 | [diff] [blame] | 492 | |
| Tom Stellard | b4a313a | 2014-08-01 00:32:39 +0000 | [diff] [blame] | 493 | //===----------------------------------------------------------------------===// |
| 494 | // Misc Pattern Fragments |
| 495 | //===----------------------------------------------------------------------===// |
| 496 | |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 497 | class Constants { |
| 498 | int TWO_PI = 0x40c90fdb; |
| 499 | int PI = 0x40490fdb; |
| 500 | int TWO_PI_INV = 0x3e22f983; |
| NAKAMURA Takumi | 4bb85f9 | 2013-10-28 04:07:23 +0000 | [diff] [blame] | 501 | int FP_UINT_MAX_PLUS_1 = 0x4f800000; // 1 << 32 in floating point encoding |
| Matt Arsenault | ce84130 | 2016-12-22 03:05:37 +0000 | [diff] [blame] | 502 | int FP16_ONE = 0x3C00; |
| Matt Arsenault | eb522e6 | 2017-02-27 22:15:25 +0000 | [diff] [blame] | 503 | int V2FP16_ONE = 0x3C003C00; |
| Matt Arsenault | aeca2fa | 2014-05-31 06:47:42 +0000 | [diff] [blame] | 504 | int FP32_ONE = 0x3f800000; |
| Matt Arsenault | 7fb961f | 2016-07-22 17:01:21 +0000 | [diff] [blame] | 505 | int FP32_NEG_ONE = 0xbf800000; |
| Matt Arsenault | 9cd9071 | 2016-04-14 01:42:16 +0000 | [diff] [blame] | 506 | int FP64_ONE = 0x3ff0000000000000; |
| Matt Arsenault | 7fb961f | 2016-07-22 17:01:21 +0000 | [diff] [blame] | 507 | int FP64_NEG_ONE = 0xbff0000000000000; |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 508 | } |
| 509 | def CONST : Constants; |
| 510 | |
| 511 | def FP_ZERO : PatLeaf < |
| 512 | (fpimm), |
| 513 | [{return N->getValueAPF().isZero();}] |
| 514 | >; |
| 515 | |
| 516 | def FP_ONE : PatLeaf < |
| 517 | (fpimm), |
| 518 | [{return N->isExactlyValue(1.0);}] |
| 519 | >; |
| 520 | |
| Matt Arsenault | eeb2a7e | 2015-01-15 23:58:35 +0000 | [diff] [blame] | 521 | def FP_HALF : PatLeaf < |
| 522 | (fpimm), |
| 523 | [{return N->isExactlyValue(0.5);}] |
| 524 | >; |
| 525 | |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 526 | /* Generic helper patterns for intrinsics */ |
| 527 | /* -------------------------------------- */ |
| 528 | |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 529 | class POW_Common <AMDGPUInst log_ieee, AMDGPUInst exp_ieee, AMDGPUInst mul> |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 530 | : AMDGPUPat < |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 531 | (fpow f32:$src0, f32:$src1), |
| 532 | (exp_ieee (mul f32:$src1, (log_ieee f32:$src0))) |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 533 | >; |
| 534 | |
| 535 | /* Other helper patterns */ |
| 536 | /* --------------------- */ |
| 537 | |
| 538 | /* Extract element pattern */ |
| Matt Arsenault | 530dde4 | 2014-02-26 23:00:58 +0000 | [diff] [blame] | 539 | class Extract_Element <ValueType sub_type, ValueType vec_type, int sub_idx, |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 540 | SubRegIndex sub_reg> |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 541 | : AMDGPUPat< |
| Matt Arsenault | fbd9bbf | 2015-12-11 19:20:16 +0000 | [diff] [blame] | 542 | (sub_type (extractelt vec_type:$src, sub_idx)), |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 543 | (EXTRACT_SUBREG $src, sub_reg) |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 544 | > { |
| 545 | let SubtargetPredicate = TruePredicate; |
| 546 | } |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 547 | |
| 548 | /* Insert element pattern */ |
| 549 | class Insert_Element <ValueType elem_type, ValueType vec_type, |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 550 | int sub_idx, SubRegIndex sub_reg> |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 551 | : AMDGPUPat < |
| Matt Arsenault | fbd9bbf | 2015-12-11 19:20:16 +0000 | [diff] [blame] | 552 | (insertelt vec_type:$vec, elem_type:$elem, sub_idx), |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 553 | (INSERT_SUBREG $vec, $elem, sub_reg) |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 554 | > { |
| 555 | let SubtargetPredicate = TruePredicate; |
| 556 | } |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 557 | |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 558 | // XXX: Convert to new syntax and use COPY_TO_REG, once the DFAPacketizer |
| 559 | // can handle COPY instructions. |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 560 | // bitconvert pattern |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 561 | class BitConvert <ValueType dt, ValueType st, RegisterClass rc> : AMDGPUPat < |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 562 | (dt (bitconvert (st rc:$src0))), |
| 563 | (dt rc:$src0) |
| 564 | >; |
| 565 | |
| Tom Stellard | 40b7f1f | 2013-05-02 15:30:12 +0000 | [diff] [blame] | 566 | // XXX: Convert to new syntax and use COPY_TO_REG, once the DFAPacketizer |
| 567 | // can handle COPY instructions. |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 568 | class DwordAddrPat<ValueType vt, RegisterClass rc> : AMDGPUPat < |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 569 | (vt (AMDGPUdwordaddr (vt rc:$addr))), |
| 570 | (vt rc:$addr) |
| 571 | >; |
| 572 | |
| Tom Stellard | 9d10c4c | 2013-04-19 02:11:06 +0000 | [diff] [blame] | 573 | // BFI_INT patterns |
| 574 | |
| Matt Arsenault | 7d858d8 | 2014-11-02 23:46:54 +0000 | [diff] [blame] | 575 | multiclass BFIPatterns <Instruction BFI_INT, |
| 576 | Instruction LoadImm32, |
| 577 | RegisterClass RC64> { |
| Tom Stellard | 9d10c4c | 2013-04-19 02:11:06 +0000 | [diff] [blame] | 578 | // Definition from ISA doc: |
| 579 | // (y & x) | (z & ~x) |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 580 | def : AMDGPUPat < |
| Tom Stellard | 9d10c4c | 2013-04-19 02:11:06 +0000 | [diff] [blame] | 581 | (or (and i32:$y, i32:$x), (and i32:$z, (not i32:$x))), |
| 582 | (BFI_INT $x, $y, $z) |
| 583 | >; |
| 584 | |
| Matt Arsenault | a18b3bc | 2018-02-07 00:21:34 +0000 | [diff] [blame] | 585 | // 64-bit version |
| 586 | def : AMDGPUPat < |
| 587 | (or (and i64:$y, i64:$x), (and i64:$z, (not i64:$x))), |
| 588 | (REG_SEQUENCE RC64, |
| 589 | (BFI_INT (i32 (EXTRACT_SUBREG $x, sub0)), |
| 590 | (i32 (EXTRACT_SUBREG $y, sub0)), |
| 591 | (i32 (EXTRACT_SUBREG $z, sub0))), sub0, |
| 592 | (BFI_INT (i32 (EXTRACT_SUBREG $x, sub1)), |
| 593 | (i32 (EXTRACT_SUBREG $y, sub1)), |
| 594 | (i32 (EXTRACT_SUBREG $z, sub1))), sub1) |
| 595 | >; |
| 596 | |
| Tom Stellard | 9d10c4c | 2013-04-19 02:11:06 +0000 | [diff] [blame] | 597 | // SHA-256 Ch function |
| 598 | // z ^ (x & (y ^ z)) |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 599 | def : AMDGPUPat < |
| Tom Stellard | 9d10c4c | 2013-04-19 02:11:06 +0000 | [diff] [blame] | 600 | (xor i32:$z, (and i32:$x, (xor i32:$y, i32:$z))), |
| 601 | (BFI_INT $x, $y, $z) |
| 602 | >; |
| 603 | |
| Matt Arsenault | a18b3bc | 2018-02-07 00:21:34 +0000 | [diff] [blame] | 604 | // 64-bit version |
| 605 | def : AMDGPUPat < |
| 606 | (xor i64:$z, (and i64:$x, (xor i64:$y, i64:$z))), |
| 607 | (REG_SEQUENCE RC64, |
| 608 | (BFI_INT (i32 (EXTRACT_SUBREG $x, sub0)), |
| 609 | (i32 (EXTRACT_SUBREG $y, sub0)), |
| 610 | (i32 (EXTRACT_SUBREG $z, sub0))), sub0, |
| 611 | (BFI_INT (i32 (EXTRACT_SUBREG $x, sub1)), |
| 612 | (i32 (EXTRACT_SUBREG $y, sub1)), |
| 613 | (i32 (EXTRACT_SUBREG $z, sub1))), sub1) |
| 614 | >; |
| 615 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 616 | def : AMDGPUPat < |
| Matt Arsenault | 6e43965 | 2014-06-10 19:00:20 +0000 | [diff] [blame] | 617 | (fcopysign f32:$src0, f32:$src1), |
| Tom Stellard | 115a615 | 2016-11-10 16:02:37 +0000 | [diff] [blame] | 618 | (BFI_INT (LoadImm32 (i32 0x7fffffff)), $src0, $src1) |
| Matt Arsenault | 6e43965 | 2014-06-10 19:00:20 +0000 | [diff] [blame] | 619 | >; |
| 620 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 621 | def : AMDGPUPat < |
| Konstantin Zhuravlyov | 7d88275 | 2017-01-13 19:49:25 +0000 | [diff] [blame] | 622 | (f32 (fcopysign f32:$src0, f64:$src1)), |
| 623 | (BFI_INT (LoadImm32 (i32 0x7fffffff)), $src0, |
| 624 | (i32 (EXTRACT_SUBREG $src1, sub1))) |
| 625 | >; |
| 626 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 627 | def : AMDGPUPat < |
| Matt Arsenault | 6e43965 | 2014-06-10 19:00:20 +0000 | [diff] [blame] | 628 | (f64 (fcopysign f64:$src0, f64:$src1)), |
| Matt Arsenault | 7d858d8 | 2014-11-02 23:46:54 +0000 | [diff] [blame] | 629 | (REG_SEQUENCE RC64, |
| 630 | (i32 (EXTRACT_SUBREG $src0, sub0)), sub0, |
| Tom Stellard | 115a615 | 2016-11-10 16:02:37 +0000 | [diff] [blame] | 631 | (BFI_INT (LoadImm32 (i32 0x7fffffff)), |
| Matt Arsenault | 6e43965 | 2014-06-10 19:00:20 +0000 | [diff] [blame] | 632 | (i32 (EXTRACT_SUBREG $src0, sub1)), |
| 633 | (i32 (EXTRACT_SUBREG $src1, sub1))), sub1) |
| 634 | >; |
| Valery Pykhtin | e55fd41 | 2016-10-20 16:17:54 +0000 | [diff] [blame] | 635 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 636 | def : AMDGPUPat < |
| Valery Pykhtin | e55fd41 | 2016-10-20 16:17:54 +0000 | [diff] [blame] | 637 | (f64 (fcopysign f64:$src0, f32:$src1)), |
| 638 | (REG_SEQUENCE RC64, |
| 639 | (i32 (EXTRACT_SUBREG $src0, sub0)), sub0, |
| Tom Stellard | 115a615 | 2016-11-10 16:02:37 +0000 | [diff] [blame] | 640 | (BFI_INT (LoadImm32 (i32 0x7fffffff)), |
| Valery Pykhtin | e55fd41 | 2016-10-20 16:17:54 +0000 | [diff] [blame] | 641 | (i32 (EXTRACT_SUBREG $src0, sub1)), |
| 642 | $src1), sub1) |
| 643 | >; |
| Tom Stellard | 9d10c4c | 2013-04-19 02:11:06 +0000 | [diff] [blame] | 644 | } |
| 645 | |
| Tom Stellard | eac65dd | 2013-05-03 17:21:20 +0000 | [diff] [blame] | 646 | // SHA-256 Ma patterns |
| 647 | |
| 648 | // ((x & z) | (y & (x | z))) -> BFI_INT (XOR x, y), z, y |
| Matt Arsenault | a18b3bc | 2018-02-07 00:21:34 +0000 | [diff] [blame] | 649 | multiclass SHA256MaPattern <Instruction BFI_INT, Instruction XOR, RegisterClass RC64> { |
| 650 | def : AMDGPUPat < |
| 651 | (or (and i32:$x, i32:$z), (and i32:$y, (or i32:$x, i32:$z))), |
| 652 | (BFI_INT (XOR i32:$x, i32:$y), i32:$z, i32:$y) |
| 653 | >; |
| 654 | |
| 655 | def : AMDGPUPat < |
| 656 | (or (and i64:$x, i64:$z), (and i64:$y, (or i64:$x, i64:$z))), |
| 657 | (REG_SEQUENCE RC64, |
| 658 | (BFI_INT (XOR (i32 (EXTRACT_SUBREG $x, sub0)), |
| 659 | (i32 (EXTRACT_SUBREG $y, sub0))), |
| 660 | (i32 (EXTRACT_SUBREG $z, sub0)), |
| 661 | (i32 (EXTRACT_SUBREG $y, sub0))), sub0, |
| 662 | (BFI_INT (XOR (i32 (EXTRACT_SUBREG $x, sub1)), |
| 663 | (i32 (EXTRACT_SUBREG $y, sub1))), |
| 664 | (i32 (EXTRACT_SUBREG $z, sub1)), |
| 665 | (i32 (EXTRACT_SUBREG $y, sub1))), sub1) |
| 666 | >; |
| 667 | } |
| Tom Stellard | eac65dd | 2013-05-03 17:21:20 +0000 | [diff] [blame] | 668 | |
| Tom Stellard | 2b971eb | 2013-05-10 02:09:45 +0000 | [diff] [blame] | 669 | // Bitfield extract patterns |
| 670 | |
| Marek Olsak | 949f5da | 2015-03-24 13:40:34 +0000 | [diff] [blame] | 671 | def IMMZeroBasedBitfieldMask : PatLeaf <(imm), [{ |
| 672 | return isMask_32(N->getZExtValue()); |
| 673 | }]>; |
| Tom Stellard | a2a4b8e | 2014-01-23 18:49:33 +0000 | [diff] [blame] | 674 | |
| Marek Olsak | 949f5da | 2015-03-24 13:40:34 +0000 | [diff] [blame] | 675 | def IMMPopCount : SDNodeXForm<imm, [{ |
| Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 676 | return CurDAG->getTargetConstant(countPopulation(N->getZExtValue()), SDLoc(N), |
| Marek Olsak | 949f5da | 2015-03-24 13:40:34 +0000 | [diff] [blame] | 677 | MVT::i32); |
| 678 | }]>; |
| Tom Stellard | a2a4b8e | 2014-01-23 18:49:33 +0000 | [diff] [blame] | 679 | |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 680 | multiclass BFEPattern <Instruction UBFE, Instruction SBFE, Instruction MOV> { |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 681 | def : AMDGPUPat < |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 682 | (i32 (and (i32 (srl i32:$src, i32:$rshift)), IMMZeroBasedBitfieldMask:$mask)), |
| 683 | (UBFE $src, $rshift, (MOV (i32 (IMMPopCount $mask)))) |
| 684 | >; |
| 685 | |
| Roman Lebedev | 9c17dad | 2018-06-15 09:56:39 +0000 | [diff] [blame] | 686 | // x & ((1 << y) - 1) |
| 687 | def : AMDGPUPat < |
| 688 | (and i32:$src, (add_oneuse (shl_oneuse 1, i32:$width), -1)), |
| 689 | (UBFE $src, (i32 0), $width) |
| 690 | >; |
| 691 | |
| Roman Lebedev | dec562c | 2018-06-15 09:56:45 +0000 | [diff] [blame^] | 692 | // x & ~(-1 << y) |
| 693 | def : AMDGPUPat < |
| 694 | (and i32:$src, (xor_oneuse (shl_oneuse -1, i32:$width), -1)), |
| 695 | (UBFE $src, (i32 0), $width) |
| 696 | >; |
| 697 | |
| Roman Lebedev | aa8587d | 2018-06-15 09:56:31 +0000 | [diff] [blame] | 698 | // x & (-1 >> (bitwidth - y)) |
| 699 | def : AMDGPUPat < |
| 700 | (and i32:$src, (srl_oneuse -1, (sub 32, i32:$width))), |
| 701 | (UBFE $src, (i32 0), $width) |
| 702 | >; |
| 703 | |
| 704 | // x << (bitwidth - y) >> (bitwidth - y) |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 705 | def : AMDGPUPat < |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 706 | (srl (shl_oneuse i32:$src, (sub 32, i32:$width)), (sub 32, i32:$width)), |
| 707 | (UBFE $src, (i32 0), $width) |
| 708 | >; |
| 709 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 710 | def : AMDGPUPat < |
| Matt Arsenault | a9e16e6 | 2017-02-23 00:23:43 +0000 | [diff] [blame] | 711 | (sra (shl_oneuse i32:$src, (sub 32, i32:$width)), (sub 32, i32:$width)), |
| 712 | (SBFE $src, (i32 0), $width) |
| 713 | >; |
| 714 | } |
| Tom Stellard | 2b971eb | 2013-05-10 02:09:45 +0000 | [diff] [blame] | 715 | |
| Tom Stellard | 5643c4a | 2013-05-20 15:02:19 +0000 | [diff] [blame] | 716 | // rotr pattern |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 717 | class ROTRPattern <Instruction BIT_ALIGN> : AMDGPUPat < |
| Tom Stellard | 5643c4a | 2013-05-20 15:02:19 +0000 | [diff] [blame] | 718 | (rotr i32:$src0, i32:$src1), |
| 719 | (BIT_ALIGN $src0, $src0, $src1) |
| 720 | >; |
| 721 | |
| Matt Arsenault | c89f291 | 2016-03-07 21:54:48 +0000 | [diff] [blame] | 722 | // This matches 16 permutations of |
| 723 | // max(min(x, y), min(max(x, y), z)) |
| 724 | class IntMed3Pat<Instruction med3Inst, |
| 725 | SDPatternOperator max, |
| 726 | SDPatternOperator max_oneuse, |
| Matt Arsenault | 10268f9 | 2017-02-27 22:40:39 +0000 | [diff] [blame] | 727 | SDPatternOperator min_oneuse, |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 728 | ValueType vt = i32> : AMDGPUPat< |
| Matt Arsenault | 10268f9 | 2017-02-27 22:40:39 +0000 | [diff] [blame] | 729 | (max (min_oneuse vt:$src0, vt:$src1), |
| 730 | (min_oneuse (max_oneuse vt:$src0, vt:$src1), vt:$src2)), |
| Matt Arsenault | c89f291 | 2016-03-07 21:54:48 +0000 | [diff] [blame] | 731 | (med3Inst $src0, $src1, $src2) |
| 732 | >; |
| 733 | |
| Matt Arsenault | eeb2a7e | 2015-01-15 23:58:35 +0000 | [diff] [blame] | 734 | // Special conversion patterns |
| 735 | |
| 736 | def cvt_rpi_i32_f32 : PatFrag < |
| 737 | (ops node:$src), |
| Matt Arsenault | 08ad328 | 2015-01-31 21:28:13 +0000 | [diff] [blame] | 738 | (fp_to_sint (ffloor (fadd $src, FP_HALF))), |
| 739 | [{ (void) N; return TM.Options.NoNaNsFPMath; }] |
| Matt Arsenault | eeb2a7e | 2015-01-15 23:58:35 +0000 | [diff] [blame] | 740 | >; |
| 741 | |
| 742 | def cvt_flr_i32_f32 : PatFrag < |
| 743 | (ops node:$src), |
| Matt Arsenault | 08ad328 | 2015-01-31 21:28:13 +0000 | [diff] [blame] | 744 | (fp_to_sint (ffloor $src)), |
| 745 | [{ (void)N; return TM.Options.NoNaNsFPMath; }] |
| Matt Arsenault | eeb2a7e | 2015-01-15 23:58:35 +0000 | [diff] [blame] | 746 | >; |
| 747 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 748 | class IMad24Pat<Instruction Inst, bit HasClamp = 0> : AMDGPUPat < |
| Matt Arsenault | eb26020 | 2014-05-22 18:00:15 +0000 | [diff] [blame] | 749 | (add (AMDGPUmul_i24 i32:$src0, i32:$src1), i32:$src2), |
| Dmitry Preobrazhensky | ff64aa5 | 2017-08-16 13:51:56 +0000 | [diff] [blame] | 750 | !if(HasClamp, (Inst $src0, $src1, $src2, (i1 0)), |
| 751 | (Inst $src0, $src1, $src2)) |
| Matt Arsenault | eb26020 | 2014-05-22 18:00:15 +0000 | [diff] [blame] | 752 | >; |
| 753 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 754 | class UMad24Pat<Instruction Inst, bit HasClamp = 0> : AMDGPUPat < |
| Matt Arsenault | eb26020 | 2014-05-22 18:00:15 +0000 | [diff] [blame] | 755 | (add (AMDGPUmul_u24 i32:$src0, i32:$src1), i32:$src2), |
| Dmitry Preobrazhensky | ff64aa5 | 2017-08-16 13:51:56 +0000 | [diff] [blame] | 756 | !if(HasClamp, (Inst $src0, $src1, $src2, (i1 0)), |
| 757 | (Inst $src0, $src1, $src2)) |
| Matt Arsenault | eb26020 | 2014-05-22 18:00:15 +0000 | [diff] [blame] | 758 | >; |
| 759 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 760 | class RcpPat<Instruction RcpInst, ValueType vt> : AMDGPUPat < |
| Matt Arsenault | a0050b0 | 2014-06-19 01:19:19 +0000 | [diff] [blame] | 761 | (fdiv FP_ONE, vt:$src), |
| 762 | (RcpInst $src) |
| 763 | >; |
| 764 | |
| Matt Arsenault | 90c7593 | 2017-10-03 00:06:41 +0000 | [diff] [blame] | 765 | class RsqPat<Instruction RsqInst, ValueType vt> : AMDGPUPat < |
| Matt Arsenault | 0bbcd8b | 2015-02-14 04:30:08 +0000 | [diff] [blame] | 766 | (AMDGPUrcp (fsqrt vt:$src)), |
| 767 | (RsqInst $src) |
| 768 | >; |
| Matt Arsenault | a0050b0 | 2014-06-19 01:19:19 +0000 | [diff] [blame] | 769 | |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 770 | include "R600Instructions.td" |
| Tom Stellard | 2c1c9de | 2014-03-24 16:07:25 +0000 | [diff] [blame] | 771 | include "R700Instructions.td" |
| 772 | include "EvergreenInstructions.td" |
| 773 | include "CaymanInstructions.td" |
| Tom Stellard | 75aadc2 | 2012-12-11 21:25:42 +0000 | [diff] [blame] | 774 | |
| 775 | include "SIInstrInfo.td" |
| 776 | |