R600: Add support for i8 and i16 local memory loads
[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 az_extloadi8_global : PatFrag<(ops node:$ptr), (az_extloadi8 node:$ptr), [{
100     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
101 }]>;
102
103 def sextloadi8_global : PatFrag<(ops node:$ptr), (sextloadi8 node:$ptr), [{
104     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
105 }]>;
106
107 def az_extloadi8_constant : PatFrag<(ops node:$ptr), (az_extloadi8 node:$ptr), [{
108     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
109 }]>;
110
111 def sextloadi8_constant : PatFrag<(ops node:$ptr), (sextloadi8 node:$ptr), [{
112     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
113 }]>;
114
115 def az_extloadi8_local : PatFrag<(ops node:$ptr), (az_extloadi8 node:$ptr), [{
116     return isLocalLoad(dyn_cast<LoadSDNode>(N));
117 }]>;
118
119 def sextloadi8_local : PatFrag<(ops node:$ptr), (sextloadi8 node:$ptr), [{
120     return isLocalLoad(dyn_cast<LoadSDNode>(N));
121 }]>;
122
123 def az_extloadi16 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{
124   return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i16;
125 }]>;
126
127 def az_extloadi16_global : PatFrag<(ops node:$ptr), (az_extloadi16 node:$ptr), [{
128     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
129 }]>;
130
131 def sextloadi16_global : PatFrag<(ops node:$ptr), (sextloadi16 node:$ptr), [{
132     return isGlobalLoad(dyn_cast<LoadSDNode>(N));
133 }]>;
134
135 def az_extloadi16_constant : PatFrag<(ops node:$ptr), (az_extloadi16 node:$ptr), [{
136     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
137 }]>;
138
139 def sextloadi16_constant : PatFrag<(ops node:$ptr), (sextloadi16 node:$ptr), [{
140     return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
141 }]>;
142
143 def az_extloadi16_local : PatFrag<(ops node:$ptr), (az_extloadi16 node:$ptr), [{
144     return isLocalLoad(dyn_cast<LoadSDNode>(N));
145 }]>;
146
147 def sextloadi16_local : PatFrag<(ops node:$ptr), (sextloadi16 node:$ptr), [{
148     return isLocalLoad(dyn_cast<LoadSDNode>(N));
149 }]>;
150
151 def az_extloadi32 : PatFrag<(ops node:$ptr), (az_extload node:$ptr), [{
152   return cast<LoadSDNode>(N)->getMemoryVT() == MVT::i32;
153 }]>;
154
155 def az_extloadi32_global : PatFrag<(ops node:$ptr),
156                                    (az_extloadi32 node:$ptr), [{
157   return isGlobalLoad(dyn_cast<LoadSDNode>(N));
158 }]>;
159
160 def az_extloadi32_constant : PatFrag<(ops node:$ptr),
161                                      (az_extloadi32 node:$ptr), [{
162   return isConstantLoad(dyn_cast<LoadSDNode>(N), -1);
163 }]>;
164
165 def truncstorei8_global : PatFrag<(ops node:$val, node:$ptr),
166                                   (truncstorei8 node:$val, node:$ptr), [{
167   return isGlobalStore(dyn_cast<StoreSDNode>(N));
168 }]>;
169
170 def truncstorei16_global : PatFrag<(ops node:$val, node:$ptr),
171                                   (truncstorei16 node:$val, node:$ptr), [{
172   return isGlobalStore(dyn_cast<StoreSDNode>(N));
173 }]>;
174
175 def local_store : PatFrag<(ops node:$val, node:$ptr),
176                              (store node:$val, node:$ptr), [{
177   return isLocalStore(dyn_cast<StoreSDNode>(N));
178 }]>;
179
180 def truncstorei8_local : PatFrag<(ops node:$val, node:$ptr),
181                                   (truncstorei8 node:$val, node:$ptr), [{
182   return isLocalStore(dyn_cast<StoreSDNode>(N));
183 }]>;
184
185 def truncstorei16_local : PatFrag<(ops node:$val, node:$ptr),
186                                   (truncstorei16 node:$val, node:$ptr), [{
187   return isLocalStore(dyn_cast<StoreSDNode>(N));
188 }]>;
189
190 def local_load : PatFrag<(ops node:$ptr), (load node:$ptr), [{
191     return isLocalLoad(dyn_cast<LoadSDNode>(N));
192 }]>;
193
194 def mskor_global : PatFrag<(ops node:$val, node:$ptr),
195                             (AMDGPUstore_mskor node:$val, node:$ptr), [{
196   return dyn_cast<MemSDNode>(N)->getAddressSpace() == AMDGPUAS::GLOBAL_ADDRESS;
197 }]>;
198
199 class Constants {
200 int TWO_PI = 0x40c90fdb;
201 int PI = 0x40490fdb;
202 int TWO_PI_INV = 0x3e22f983;
203 int FP_UINT_MAX_PLUS_1 = 0x4f800000;    // 1 << 32 in floating point encoding
204 }
205 def CONST : Constants;
206
207 def FP_ZERO : PatLeaf <
208   (fpimm),
209   [{return N->getValueAPF().isZero();}]
210 >;
211
212 def FP_ONE : PatLeaf <
213   (fpimm),
214   [{return N->isExactlyValue(1.0);}]
215 >;
216
217 def U24 : ComplexPattern<i32, 1, "SelectU24", [], []>;
218 def I24 : ComplexPattern<i32, 1, "SelectI24", [], []>;
219
220 let isCodeGenOnly = 1, isPseudo = 1 in {
221
222 let usesCustomInserter = 1  in {
223
224 class CLAMP <RegisterClass rc> : AMDGPUShaderInst <
225   (outs rc:$dst),
226   (ins rc:$src0),
227   "CLAMP $dst, $src0",
228   [(set f32:$dst, (int_AMDIL_clamp f32:$src0, (f32 FP_ZERO), (f32 FP_ONE)))]
229 >;
230
231 class FABS <RegisterClass rc> : AMDGPUShaderInst <
232   (outs rc:$dst),
233   (ins rc:$src0),
234   "FABS $dst, $src0",
235   [(set f32:$dst, (fabs f32:$src0))]
236 >;
237
238 class FNEG <RegisterClass rc> : AMDGPUShaderInst <
239   (outs rc:$dst),
240   (ins rc:$src0),
241   "FNEG $dst, $src0",
242   [(set f32:$dst, (fneg f32:$src0))]
243 >;
244
245 } // usesCustomInserter = 1
246
247 multiclass RegisterLoadStore <RegisterClass dstClass, Operand addrClass,
248                     ComplexPattern addrPat> {
249   def RegisterLoad : AMDGPUShaderInst <
250     (outs dstClass:$dst),
251     (ins addrClass:$addr, i32imm:$chan),
252     "RegisterLoad $dst, $addr",
253     [(set i32:$dst, (AMDGPUregister_load addrPat:$addr, (i32 timm:$chan)))]
254   > {
255     let isRegisterLoad = 1;
256   }
257
258   def RegisterStore : AMDGPUShaderInst <
259     (outs),
260     (ins dstClass:$val, addrClass:$addr, i32imm:$chan),
261     "RegisterStore $val, $addr",
262     [(AMDGPUregister_store i32:$val, addrPat:$addr, (i32 timm:$chan))]
263   > {
264     let isRegisterStore = 1;
265   }
266 }
267
268 } // End isCodeGenOnly = 1, isPseudo = 1
269
270 /* Generic helper patterns for intrinsics */
271 /* -------------------------------------- */
272
273 class POW_Common <AMDGPUInst log_ieee, AMDGPUInst exp_ieee, AMDGPUInst mul>
274   : Pat <
275   (fpow f32:$src0, f32:$src1),
276   (exp_ieee (mul f32:$src1, (log_ieee f32:$src0)))
277 >;
278
279 /* Other helper patterns */
280 /* --------------------- */
281
282 /* Extract element pattern */
283 class Extract_Element <ValueType sub_type, ValueType vec_type, int sub_idx, 
284                        SubRegIndex sub_reg>
285   : Pat<
286   (sub_type (vector_extract vec_type:$src, sub_idx)),
287   (EXTRACT_SUBREG $src, sub_reg)
288 >;
289
290 /* Insert element pattern */
291 class Insert_Element <ValueType elem_type, ValueType vec_type,
292                       int sub_idx, SubRegIndex sub_reg>
293   : Pat <
294   (vector_insert vec_type:$vec, elem_type:$elem, sub_idx),
295   (INSERT_SUBREG $vec, $elem, sub_reg)
296 >;
297
298 class Vector4_Build <ValueType vecType, ValueType elemType> : Pat <
299   (vecType (build_vector elemType:$x, elemType:$y, elemType:$z, elemType:$w)),
300   (INSERT_SUBREG (INSERT_SUBREG (INSERT_SUBREG (INSERT_SUBREG
301     (vecType (IMPLICIT_DEF)), $x, sub0), $y, sub1), $z, sub2), $w, sub3)
302 >;
303
304 // XXX: Convert to new syntax and use COPY_TO_REG, once the DFAPacketizer
305 // can handle COPY instructions.
306 // bitconvert pattern
307 class BitConvert <ValueType dt, ValueType st, RegisterClass rc> : Pat <
308   (dt (bitconvert (st rc:$src0))),
309   (dt rc:$src0)
310 >;
311
312 // XXX: Convert to new syntax and use COPY_TO_REG, once the DFAPacketizer
313 // can handle COPY instructions.
314 class DwordAddrPat<ValueType vt, RegisterClass rc> : Pat <
315   (vt (AMDGPUdwordaddr (vt rc:$addr))),
316   (vt rc:$addr)
317 >;
318
319 // BFI_INT patterns
320
321 multiclass BFIPatterns <Instruction BFI_INT> {
322
323   // Definition from ISA doc:
324   // (y & x) | (z & ~x)
325   def : Pat <
326     (or (and i32:$y, i32:$x), (and i32:$z, (not i32:$x))),
327     (BFI_INT $x, $y, $z)
328   >;
329
330   // SHA-256 Ch function
331   // z ^ (x & (y ^ z))
332   def : Pat <
333     (xor i32:$z, (and i32:$x, (xor i32:$y, i32:$z))),
334     (BFI_INT $x, $y, $z)
335   >;
336
337 }
338
339 // SHA-256 Ma patterns
340
341 // ((x & z) | (y & (x | z))) -> BFI_INT (XOR x, y), z, y
342 class SHA256MaPattern <Instruction BFI_INT, Instruction XOR> : Pat <
343   (or (and i32:$x, i32:$z), (and i32:$y, (or i32:$x, i32:$z))),
344   (BFI_INT (XOR i32:$x, i32:$y), i32:$z, i32:$y)
345 >;
346
347 // Bitfield extract patterns
348
349 def legalshift32 : ImmLeaf <i32, [{return Imm >=0 && Imm < 32;}]>;
350 def bfemask : PatLeaf <(imm), [{return isMask_32(N->getZExtValue());}],
351                             SDNodeXForm<imm, [{ return CurDAG->getTargetConstant(CountTrailingOnes_32(N->getZExtValue()), MVT::i32);}]>>;
352
353 class BFEPattern <Instruction BFE> : Pat <
354   (and (srl i32:$x, legalshift32:$y), bfemask:$z),
355   (BFE $x, $y, $z)
356 >;
357
358 // rotr pattern
359 class ROTRPattern <Instruction BIT_ALIGN> : Pat <
360   (rotr i32:$src0, i32:$src1),
361   (BIT_ALIGN $src0, $src0, $src1)
362 >;
363
364 // 24-bit arithmetic patterns
365 def umul24 : PatFrag <(ops node:$x, node:$y), (mul node:$x, node:$y)>;
366
367 /*
368 class UMUL24Pattern <Instruction UMUL24> : Pat <
369   (mul U24:$x, U24:$y),
370   (UMUL24 $x, $y)
371 >;
372 */
373
374 include "R600Instructions.td"
375
376 include "SIInstrInfo.td"
377