R600: Add support for i16 and i8 global stores
[oota-llvm.git] / lib / Target / R600 / AMDGPUInstructions.td
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
15 class AMDGPUInst <dag outs, dag ins, string asm, list<dag> pattern> : Instruction {
16   field bit isRegisterLoad = 0;
17   field bit isRegisterStore = 0;
18
19   let Namespace = "AMDGPU";
20   let OutOperandList = outs;
21   let InOperandList = ins;
22   let AsmString = asm;
23   let Pattern = pattern;
24   let Itinerary = NullALU;
25
26   let TSFlags{63} = isRegisterLoad;
27   let TSFlags{62} = isRegisterStore;
28 }
29
30 class AMDGPUShaderInst <dag outs, dag ins, string asm, list<dag> pattern>
31     : AMDGPUInst<outs, ins, asm, pattern> {
32
33   field bits<32> Inst = 0xffffffff;
34
35 }
36
37 def InstFlag : OperandWithDefaultOps <i32, (ops (i32 0))>;
38
39 def COND_EQ : PatLeaf <
40   (cond),
41   [{switch(N->get()){{default: return false;
42                      case ISD::SETOEQ: case ISD::SETUEQ:
43                      case ISD::SETEQ: return true;}}}]
44 >;
45
46 def COND_NE : PatLeaf <
47   (cond),
48   [{switch(N->get()){{default: return false;
49                      case ISD::SETONE: case ISD::SETUNE:
50                      case ISD::SETNE: return true;}}}]
51 >;
52 def COND_GT : PatLeaf <
53   (cond),
54   [{switch(N->get()){{default: return false;
55                      case ISD::SETOGT: case ISD::SETUGT:
56                      case ISD::SETGT: return true;}}}]
57 >;
58
59 def COND_GE : PatLeaf <
60   (cond),
61   [{switch(N->get()){{default: return false;
62                      case ISD::SETOGE: case ISD::SETUGE:
63                      case ISD::SETGE: return true;}}}]
64 >;
65
66 def COND_LT : PatLeaf <
67   (cond),
68   [{switch(N->get()){{default: return false;
69                      case ISD::SETOLT: case ISD::SETULT:
70                      case ISD::SETLT: return true;}}}]
71 >;
72
73 def COND_LE : PatLeaf <
74   (cond),
75   [{switch(N->get()){{default: return false;
76                      case ISD::SETOLE: case ISD::SETULE:
77                      case ISD::SETLE: return true;}}}]
78 >;
79
80 def COND_NULL : PatLeaf <
81   (cond),
82   [{return false;}]
83 >;
84
85 //===----------------------------------------------------------------------===//
86 // Load/Store Pattern Fragments
87 //===----------------------------------------------------------------------===//
88
89 def az_extload : PatFrag<(ops node:$ptr), (unindexedload node:$ptr), [{
90   LoadSDNode *L = cast<LoadSDNode>(N);
91   return L->getExtensionType() == ISD::ZEXTLOAD ||
92          L->getExtensionType() == ISD::EXTLOAD;
93 }]>;
94
95 def az_extloadi8 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{
96   return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i8;
97 }]>;
98
99 def sextloadi8_global : PatFrag<(ops node:$ptr), (sextloadi8 node:$ptr), [{
100     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
101 }]>;
102
103 def az_extloadi8_constant : PatFrag<(ops node:$ptr), (az_extloadi8 node:$ptr), [{
104     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
105 }]>;
106
107 def sextloadi8_constant : PatFrag<(ops node:$ptr), (sextloadi8 node:$ptr), [{
108     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
109 }]>;
110
111 def az_extloadi8_global : PatFrag<(ops node:$ptr), (az_extloadi8 node:$ptr), [{
112     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
113 }]>;
114
115 def az_extloadi16 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{
116   return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i16;
117 }]>;
118
119 def az_extloadi16_global : PatFrag<(ops node:$ptr), (az_extloadi16 node:$ptr), [{
120     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
121 }]>;
122
123 def sextloadi16_global : PatFrag<(ops node:$ptr), (sextloadi16 node:$ptr), [{
124     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
125 }]>;
126
127 def az_extloadi16_constant : PatFrag<(ops node:$ptr), (az_extloadi16 node:$ptr), [{
128     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
129 }]>;
130
131 def sextloadi16_constant : PatFrag<(ops node:$ptr), (sextloadi16 node:$ptr), [{
132     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
133 }]>;
134
135 def az_extloadi32 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{
136   return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i32;
137 }]>;
138
139 def az_extloadi32_global : PatFrag<(ops node:$ptr),
140                                    (az_extloadi32 node:$ptr), [{
141   return isGlobalLoad(dyn_cast<LoadSDNode>(N));
142 }]>;
143
144 def az_extloadi32_constant : PatFrag<(ops node:$ptr),
145                                      (az_extloadi32 node:$ptr), [{
146   return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
147 }]>;
148
149 def truncstorei8_global : PatFrag<(ops node:$val, node:$ptr),
150                                   (truncstorei8 node:$val, node:$ptr), [{
151   return isGlobalStore(dyn_cast<StoreSDNode>(N));
152 }]>;
153
154 def truncstorei16_global : PatFrag<(ops node:$val, node:$ptr),
155                                   (truncstorei16 node:$val, node:$ptr), [{
156   return isGlobalStore(dyn_cast<StoreSDNode>(N));
157 }]>;
158
159 def local_load : PatFrag<(ops node:$ptr), (load node:$ptr), [{
160     return isLocalLoad(dyn_cast<LoadSDNode>(N));
161 }]>;
162
163 def local_store : PatFrag<(ops node:$val, node:$ptr),
164                              (store node:$val, node:$ptr), [{
165     return isLocalStore(dyn_cast<StoreSDNode>(N));
166 }]>;
167
168 def mskor_global : PatFrag<(ops node:$val, node:$ptr),
169                             (AMDGPUstore_mskor node:$val, node:$ptr), [{
170   return dyn_cast<MemSDNode>(N)->getAddressSpace() == AMDGPUAS::GLOBAL_ADDRESS;
171 }]>;
172
173 class Constants {
174 int TWO_PI = 0x40c90fdb;
175 int PI = 0x40490fdb;
176 int TWO_PI_INV = 0x3e22f983;
177 int FP_UINT_MAX_PLUS_1 = 0x4f800000;    // 1 << 32 in floating point encoding
178 }
179 def CONST : Constants;
180
181 def FP_ZERO : PatLeaf <
182   (fpimm),
183   [{return N->getValueAPF().isZero();}]
184 >;
185
186 def FP_ONE : PatLeaf <
187   (fpimm),
188   [{return N->isExactlyValue(1.0);}]
189 >;
190
191 def U24 : ComplexPattern<i32, 1, "SelectU24", [], []>;
192 def I24 : ComplexPattern<i32, 1, "SelectI24", [], []>;
193
194 let isCodeGenOnly = 1, isPseudo = 1 in {
195
196 let usesCustomInserter = 1  in {
197
198 class CLAMP <RegisterClass rc> : AMDGPUShaderInst <
199   (outs rc:$dst),
200   (ins rc:$src0),
201   "CLAMP $dst, $src0",
202   [(set f32:$dst, (int_AMDIL_clamp f32:$src0, (f32 FP_ZERO), (f32 FP_ONE)))]
203 >;
204
205 class FABS <RegisterClass rc> : AMDGPUShaderInst <
206   (outs rc:$dst),
207   (ins rc:$src0),
208   "FABS $dst, $src0",
209   [(set f32:$dst, (fabs f32:$src0))]
210 >;
211
212 class FNEG <RegisterClass rc> : AMDGPUShaderInst <
213   (outs rc:$dst),
214   (ins rc:$src0),
215   "FNEG $dst, $src0",
216   [(set f32:$dst, (fneg f32:$src0))]
217 >;
218
219 } // usesCustomInserter = 1
220
221 multiclass RegisterLoadStore <RegisterClass dstClass, Operand addrClass,
222                     ComplexPattern addrPat> {
223   def RegisterLoad : AMDGPUShaderInst <
224     (outs dstClass:$dst),
225     (ins addrClass:$addr, i32imm:$chan),
226     "RegisterLoad $dst, $addr",
227     [(set i32:$dst, (AMDGPUregister_load addrPat:$addr, (i32 timm:$chan)))]
228   > {
229     let isRegisterLoad = 1;
230   }
231
232   def RegisterStore : AMDGPUShaderInst <
233     (outs),
234     (ins dstClass:$val, addrClass:$addr, i32imm:$chan),
235     "RegisterStore $val, $addr",
236     [(AMDGPUregister_store i32:$val, addrPat:$addr, (i32 timm:$chan))]
237   > {
238     let isRegisterStore = 1;
239   }
240 }
241
242 } // End isCodeGenOnly = 1, isPseudo = 1
243
244 /* Generic helper patterns for intrinsics */
245 /* -------------------------------------- */
246
247 class POW_Common <AMDGPUInst log_ieee, AMDGPUInst exp_ieee, AMDGPUInst mul>
248   : Pat <
249   (fpow f32:$src0, f32:$src1),
250   (exp_ieee (mul f32:$src1, (log_ieee f32:$src0)))
251 >;
252
253 /* Other helper patterns */
254 /* --------------------- */
255
256 /* Extract element pattern */
257 class Extract_Element <ValueType sub_type, ValueType vec_type, int sub_idx, 
258                        SubRegIndex sub_reg>
259   : Pat<
260   (sub_type (vector_extract vec_type:$src, sub_idx)),
261   (EXTRACT_SUBREG $src, sub_reg)
262 >;
263
264 /* Insert element pattern */
265 class Insert_Element <ValueType elem_type, ValueType vec_type,
266                       int sub_idx, SubRegIndex sub_reg>
267   : Pat <
268   (vector_insert vec_type:$vec, elem_type:$elem, sub_idx),
269   (INSERT_SUBREG $vec, $elem, sub_reg)
270 >;
271
272 class Vector4_Build <ValueType vecType, ValueType elemType> : Pat <
273   (vecType (build_vector elemType:$x, elemType:$y, elemType:$z, elemType:$w)),
274   (INSERT_SUBREG (INSERT_SUBREG (INSERT_SUBREG (INSERT_SUBREG
275     (vecType (IMPLICIT_DEF)), $x, sub0), $y, sub1), $z, sub2), $w, sub3)
276 >;
277
278 // XXX: Convert to new syntax and use COPY_TO_REG, once the DFAPacketizer
279 // can handle COPY instructions.
280 // bitconvert pattern
281 class BitConvert <ValueType dt, ValueType st, RegisterClass rc> : Pat <
282   (dt (bitconvert (st rc:$src0))),
283   (dt rc:$src0)
284 >;
285
286 // XXX: Convert to new syntax and use COPY_TO_REG, once the DFAPacketizer
287 // can handle COPY instructions.
288 class DwordAddrPat<ValueType vt, RegisterClass rc> : Pat <
289   (vt (AMDGPUdwordaddr (vt rc:$addr))),
290   (vt rc:$addr)
291 >;
292
293 // BFI_INT patterns
294
295 multiclass BFIPatterns <Instruction BFI_INT> {
296
297   // Definition from ISA doc:
298   // (y & x) | (z & ~x)
299   def : Pat <
300     (or (and i32:$y, i32:$x), (and i32:$z, (not i32:$x))),
301     (BFI_INT $x, $y, $z)
302   >;
303
304   // SHA-256 Ch function
305   // z ^ (x & (y ^ z))
306   def : Pat <
307     (xor i32:$z, (and i32:$x, (xor i32:$y, i32:$z))),
308     (BFI_INT $x, $y, $z)
309   >;
310
311 }
312
313 // SHA-256 Ma patterns
314
315 // ((x & z) | (y & (x | z))) -> BFI_INT (XOR x, y), z, y
316 class SHA256MaPattern <Instruction BFI_INT, Instruction XOR> : Pat <
317   (or (and i32:$x, i32:$z), (and i32:$y, (or i32:$x, i32:$z))),
318   (BFI_INT (XOR i32:$x, i32:$y), i32:$z, i32:$y)
319 >;
320
321 // Bitfield extract patterns
322
323 def legalshift32 : ImmLeaf <i32, [{return Imm >=0 && Imm < 32;}]>;
324 def bfemask : PatLeaf <(imm), [{return isMask_32(N->getZExtValue());}],
325                             SDNodeXForm<imm, [{ return CurDAG->getTargetConstant(CountTrailingOnes_32(N->getZExtValue()), MVT::i32);}]>>;
326
327 class BFEPattern <Instruction BFE> : Pat <
328   (and (srl i32:$x, legalshift32:$y), bfemask:$z),
329   (BFE $x, $y, $z)
330 >;
331
332 // rotr pattern
333 class ROTRPattern <Instruction BIT_ALIGN> : Pat <
334   (rotr i32:$src0, i32:$src1),
335   (BIT_ALIGN $src0, $src0, $src1)
336 >;
337
338 // 24-bit arithmetic patterns
339 def umul24 : PatFrag <(ops node:$x, node:$y), (mul node:$x, node:$y)>;
340
341 /*
342 class UMUL24Pattern <Instruction UMUL24> : Pat <
343   (mul U24:$x, U24:$y),
344   (UMUL24 $x, $y)
345 >;
346 */
347
348 include "R600Instructions.td"
349
350 include "SIInstrInfo.td"
351