nv50/ir: make Instruction::src/def container private
[mesa.git] / src / gallium / drivers / nv50 / codegen / nv50_ir.h
1 /*
2 * Copyright 2011 Christoph Bumiller
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17 * THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
18 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF
19 * OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20 * SOFTWARE.
21 */
22
23 #ifndef __NV50_IR_H__
24 #define __NV50_IR_H__
25
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <stdint.h>
29 #include <deque>
30 #include <list>
31 #include <vector>
32
33 #include "nv50_ir_util.h"
34 #include "nv50_ir_graph.h"
35
36 #include "nv50_ir_driver.h"
37
38 namespace nv50_ir {
39
40 enum operation
41 {
42 OP_NOP = 0,
43 OP_PHI,
44 OP_UNION, // unify a new definition and several source values
45 OP_SPLIT, // $r0d -> { $r0, $r1 } ($r0d and $r0/$r1 will be coalesced)
46 OP_MERGE, // opposite of split, e.g. combine 2 32 bit into a 64 bit value
47 OP_CONSTRAINT, // copy values into consecutive registers
48 OP_MOV,
49 OP_LOAD,
50 OP_STORE,
51 OP_ADD,
52 OP_SUB,
53 OP_MUL,
54 OP_DIV,
55 OP_MOD,
56 OP_MAD,
57 OP_FMA,
58 OP_SAD, // abs(src0 - src1) + src2
59 OP_ABS,
60 OP_NEG,
61 OP_NOT,
62 OP_AND,
63 OP_OR,
64 OP_XOR,
65 OP_SHL,
66 OP_SHR,
67 OP_MAX,
68 OP_MIN,
69 OP_SAT, // CLAMP(f32, 0.0, 1.0)
70 OP_CEIL,
71 OP_FLOOR,
72 OP_TRUNC,
73 OP_CVT,
74 OP_SET_AND, // dst = (src0 CMP src1) & src2
75 OP_SET_OR,
76 OP_SET_XOR,
77 OP_SET,
78 OP_SELP, // dst = src2 ? src0 : src1
79 OP_SLCT, // dst = (src2 CMP 0) ? src0 : src1
80 OP_RCP,
81 OP_RSQ,
82 OP_LG2,
83 OP_SIN,
84 OP_COS,
85 OP_EX2,
86 OP_EXP, // exponential (base M_E)
87 OP_LOG, // natural logarithm
88 OP_PRESIN,
89 OP_PREEX2,
90 OP_SQRT,
91 OP_POW,
92 OP_BRA,
93 OP_CALL,
94 OP_RET,
95 OP_CONT,
96 OP_BREAK,
97 OP_PRERET,
98 OP_PRECONT,
99 OP_PREBREAK,
100 OP_BRKPT, // breakpoint (not related to loops)
101 OP_JOINAT, // push control flow convergence point
102 OP_JOIN, // converge
103 OP_DISCARD,
104 OP_EXIT,
105 OP_MEMBAR,
106 OP_VFETCH, // indirection 0 in attribute space, indirection 1 is vertex base
107 OP_PFETCH, // fetch base address of vertex src0 (immediate) [+ src1]
108 OP_EXPORT,
109 OP_LINTERP,
110 OP_PINTERP,
111 OP_EMIT, // emit vertex
112 OP_RESTART, // restart primitive
113 OP_TEX,
114 OP_TXB, // texture bias
115 OP_TXL, // texure lod
116 OP_TXF, // texel fetch
117 OP_TXQ, // texture size query
118 OP_TXD, // texture derivatives
119 OP_TXG, // texture gather
120 OP_TEXCSAA,
121 OP_SULD, // surface load
122 OP_SUST, // surface store
123 OP_DFDX,
124 OP_DFDY,
125 OP_RDSV, // read system value
126 OP_WRSV, // write system value
127 OP_PIXLD,
128 OP_QUADOP,
129 OP_QUADON,
130 OP_QUADPOP,
131 OP_POPCNT, // bitcount(src0 & src1)
132 OP_INSBF, // insert first src1[8:15] bits of src0 into src2 at src1[0:7]
133 OP_EXTBF,
134 OP_LAST
135 };
136
137 #define NV50_IR_SUBOP_MUL_HIGH 1
138 #define NV50_IR_SUBOP_EMIT_RESTART 1
139 #define NV50_IR_SUBOP_LDC_IL 1
140 #define NV50_IR_SUBOP_LDC_IS 2
141 #define NV50_IR_SUBOP_LDC_ISL 3
142 #define NV50_IR_SUBOP_SHIFT_WRAP 1
143
144 enum DataType
145 {
146 TYPE_NONE,
147 TYPE_U8,
148 TYPE_S8,
149 TYPE_U16,
150 TYPE_S16,
151 TYPE_U32,
152 TYPE_S32,
153 TYPE_U64, // 64 bit operations are only lowered after register allocation
154 TYPE_S64,
155 TYPE_F16,
156 TYPE_F32,
157 TYPE_F64,
158 TYPE_B96,
159 TYPE_B128
160 };
161
162 enum CondCode
163 {
164 CC_FL = 0,
165 CC_NEVER = CC_FL, // when used with FILE_FLAGS
166 CC_LT = 1,
167 CC_EQ = 2,
168 CC_NOT_P = CC_EQ, // when used with FILE_PREDICATE
169 CC_LE = 3,
170 CC_GT = 4,
171 CC_NE = 5,
172 CC_P = CC_NE,
173 CC_GE = 6,
174 CC_TR = 7,
175 CC_ALWAYS = CC_TR,
176 CC_U = 8,
177 CC_LTU = 9,
178 CC_EQU = 10,
179 CC_LEU = 11,
180 CC_GTU = 12,
181 CC_NEU = 13,
182 CC_GEU = 14,
183 CC_NO = 0x10,
184 CC_NC = 0x11,
185 CC_NS = 0x12,
186 CC_NA = 0x13,
187 CC_A = 0x14,
188 CC_S = 0x15,
189 CC_C = 0x16,
190 CC_O = 0x17
191 };
192
193 enum RoundMode
194 {
195 ROUND_N, // nearest
196 ROUND_M, // towards -inf
197 ROUND_Z, // towards 0
198 ROUND_P, // towards +inf
199 ROUND_NI, // nearest integer
200 ROUND_MI, // to integer towards -inf
201 ROUND_ZI, // to integer towards 0
202 ROUND_PI, // to integer towards +inf
203 };
204
205 enum CacheMode
206 {
207 CACHE_CA, // cache at all levels
208 CACHE_WB = CACHE_CA, // cache write back
209 CACHE_CG, // cache at global level
210 CACHE_CS, // cache streaming
211 CACHE_CV, // cache as volatile
212 CACHE_WT = CACHE_CV // cache write-through
213 };
214
215 enum DataFile
216 {
217 FILE_NULL = 0,
218 FILE_GPR,
219 FILE_PREDICATE, // boolean predicate
220 FILE_FLAGS, // zero/sign/carry/overflow bits
221 FILE_ADDRESS,
222 FILE_IMMEDIATE,
223 FILE_MEMORY_CONST,
224 FILE_SHADER_INPUT,
225 FILE_SHADER_OUTPUT,
226 FILE_MEMORY_GLOBAL,
227 FILE_MEMORY_SHARED,
228 FILE_MEMORY_LOCAL,
229 FILE_SYSTEM_VALUE,
230 DATA_FILE_COUNT
231 };
232
233 enum TexTarget
234 {
235 TEX_TARGET_1D,
236 TEX_TARGET_2D,
237 TEX_TARGET_2D_MS,
238 TEX_TARGET_3D,
239 TEX_TARGET_CUBE,
240 TEX_TARGET_1D_SHADOW,
241 TEX_TARGET_2D_SHADOW,
242 TEX_TARGET_CUBE_SHADOW,
243 TEX_TARGET_1D_ARRAY,
244 TEX_TARGET_2D_ARRAY,
245 TEX_TARGET_2D_MS_ARRAY,
246 TEX_TARGET_CUBE_ARRAY,
247 TEX_TARGET_1D_ARRAY_SHADOW,
248 TEX_TARGET_2D_ARRAY_SHADOW,
249 TEX_TARGET_RECT,
250 TEX_TARGET_RECT_SHADOW,
251 TEX_TARGET_CUBE_ARRAY_SHADOW,
252 TEX_TARGET_BUFFER,
253 TEX_TARGET_COUNT
254 };
255
256 enum SVSemantic
257 {
258 SV_POSITION, // WPOS
259 SV_VERTEX_ID,
260 SV_INSTANCE_ID,
261 SV_INVOCATION_ID,
262 SV_PRIMITIVE_ID,
263 SV_VERTEX_COUNT, // gl_PatchVerticesIn
264 SV_LAYER,
265 SV_VIEWPORT_INDEX,
266 SV_YDIR,
267 SV_FACE,
268 SV_POINT_SIZE,
269 SV_POINT_COORD,
270 SV_CLIP_DISTANCE,
271 SV_SAMPLE_INDEX,
272 SV_TESS_FACTOR,
273 SV_TESS_COORD,
274 SV_TID,
275 SV_CTAID,
276 SV_NTID,
277 SV_GRIDID,
278 SV_NCTAID,
279 SV_LANEID,
280 SV_PHYSID,
281 SV_NPHYSID,
282 SV_CLOCK,
283 SV_LBASE,
284 SV_SBASE,
285 SV_UNDEFINED,
286 SV_LAST
287 };
288
289 class Program;
290 class Function;
291 class BasicBlock;
292
293 class Target;
294
295 class Instruction;
296 class CmpInstruction;
297 class TexInstruction;
298 class FlowInstruction;
299
300 class Value;
301 class LValue;
302 class Symbol;
303 class ImmediateValue;
304
305 struct Storage
306 {
307 DataFile file;
308 int8_t fileIndex; // signed, may be indirect for CONST[]
309 uint8_t size; // this should match the Instruction type's size
310 DataType type; // mainly for pretty printing
311 union {
312 uint64_t u64; // immediate values
313 uint32_t u32;
314 uint16_t u16;
315 uint8_t u8;
316 int64_t s64;
317 int32_t s32;
318 int16_t s16;
319 int8_t s8;
320 float f32;
321 double f64;
322 int32_t offset; // offset from 0 (base of address space)
323 int32_t id; // register id (< 0 if virtual/unassigned)
324 struct {
325 SVSemantic sv;
326 int index;
327 } sv;
328 } data;
329 };
330
331 // precedence: NOT after SAT after NEG after ABS
332 #define NV50_IR_MOD_ABS (1 << 0)
333 #define NV50_IR_MOD_NEG (1 << 1)
334 #define NV50_IR_MOD_SAT (1 << 2)
335 #define NV50_IR_MOD_NOT (1 << 3)
336 #define NV50_IR_MOD_NEG_ABS (NV50_IR_MOD_NEG | NV50_IR_MOD_ABS)
337
338 #define NV50_IR_INTERP_MODE_MASK 0x3
339 #define NV50_IR_INTERP_LINEAR (0 << 0)
340 #define NV50_IR_INTERP_PERSPECTIVE (1 << 0)
341 #define NV50_IR_INTERP_FLAT (2 << 0)
342 #define NV50_IR_INTERP_SC (3 << 0) // what exactly is that ?
343 #define NV50_IR_INTERP_SAMPLE_MASK 0xc
344 #define NV50_IR_INTERP_DEFAULT (0 << 2)
345 #define NV50_IR_INTERP_CENTROID (1 << 2)
346 #define NV50_IR_INTERP_OFFSET (2 << 2)
347 #define NV50_IR_INTERP_SAMPLEID (3 << 2)
348
349 // do we really want this to be a class ?
350 class Modifier
351 {
352 public:
353 Modifier() : bits(0) { }
354 Modifier(unsigned int m) : bits(m) { }
355 Modifier(operation op);
356
357 // @return new Modifier applying a after b (asserts if unrepresentable)
358 Modifier operator*(const Modifier) const;
359 Modifier operator==(const Modifier m) const { return m.bits == bits; }
360 Modifier operator!=(const Modifier m) const { return m.bits != bits; }
361
362 inline Modifier operator&(const Modifier m) const { return bits & m.bits; }
363 inline Modifier operator|(const Modifier m) const { return bits | m.bits; }
364 inline Modifier operator^(const Modifier m) const { return bits ^ m.bits; }
365
366 operation getOp() const;
367
368 inline int neg() const { return (bits & NV50_IR_MOD_NEG) ? 1 : 0; }
369 inline int abs() const { return (bits & NV50_IR_MOD_ABS) ? 1 : 0; }
370
371 inline operator bool() { return bits ? true : false; }
372
373 void applyTo(ImmediateValue &imm) const;
374
375 int print(char *buf, size_t size) const;
376
377 private:
378 uint8_t bits;
379 };
380
381 class ValueRef
382 {
383 public:
384 ValueRef();
385 ValueRef(const ValueRef&);
386 ~ValueRef();
387
388 inline bool exists() const { return value != NULL; }
389
390 void set(Value *);
391 void set(const ValueRef&);
392 inline Value *get() const { return value; }
393 inline Value *rep() const;
394
395 inline Instruction *getInsn() const { return insn; }
396 inline void setInsn(Instruction *inst) { insn = inst; }
397
398 inline bool isIndirect(int dim) const { return indirect[dim] >= 0; }
399 inline const ValueRef *getIndirect(int dim) const;
400
401 inline DataFile getFile() const;
402 inline unsigned getSize() const;
403
404 // SSA: return eventual (traverse MOVs) literal value, if it exists
405 ImmediateValue *getImmediate() const;
406
407 public:
408 Modifier mod;
409 int8_t indirect[2]; // >= 0 if relative to lvalue in insn->src(indirect[i])
410 uint8_t swizzle;
411
412 bool usedAsPtr; // for printing
413
414 private:
415 Value *value;
416 Instruction *insn;
417 };
418
419 class ValueDef
420 {
421 public:
422 ValueDef();
423 ValueDef(const ValueDef&);
424 ~ValueDef();
425
426 inline bool exists() const { return value != NULL; }
427
428 inline Value *get() const { return value; }
429 inline Value *rep() const;
430 void set(Value *);
431 void replace(Value *, bool doSet); // replace all uses of the old value
432
433 inline Instruction *getInsn() const { return insn; }
434 inline void setInsn(Instruction *inst) { insn = inst; }
435
436 inline DataFile getFile() const;
437 inline unsigned getSize() const;
438
439 inline void setSSA(LValue *);
440 inline const LValue *preSSA() const;
441
442 private:
443 Value *value; // should make this LValue * ...
444 LValue *origin; // pre SSA value
445 Instruction *insn;
446 };
447
448 class Value
449 {
450 public:
451 Value();
452
453 virtual Value *clone(Function *) const { return NULL; }
454
455 virtual int print(char *, size_t, DataType ty = TYPE_NONE) const = 0;
456
457 virtual bool equals(const Value *, bool strict = false) const;
458 virtual bool interfers(const Value *) const;
459
460 inline Value *rep() const { return join; }
461
462 inline Instruction *getUniqueInsn() const;
463 inline Instruction *getInsn() const; // use when uniqueness is certain
464
465 inline int refCount() { return uses.size(); }
466
467 inline LValue *asLValue();
468 inline Symbol *asSym();
469 inline ImmediateValue *asImm();
470 inline const Symbol *asSym() const;
471 inline const ImmediateValue *asImm() const;
472
473 bool coalesce(Value *, bool force = false);
474
475 inline bool inFile(DataFile f) { return reg.file == f; }
476
477 static inline Value *get(Iterator&);
478
479 std::list<ValueRef *> uses;
480 std::list<ValueDef *> defs;
481 typedef std::list<ValueRef *>::iterator UseIterator;
482 typedef std::list<ValueRef *>::const_iterator UseCIterator;
483 typedef std::list<ValueDef *>::iterator DefIterator;
484 typedef std::list<ValueDef *>::const_iterator DefCIterator;
485
486 int id;
487 Storage reg;
488
489 // TODO: these should be in LValue:
490 Interval livei;
491 Value *join;
492 };
493
494 class LValue : public Value
495 {
496 public:
497 LValue(Function *, DataFile file);
498 LValue(Function *, LValue *);
499
500 virtual Value *clone(Function *) const;
501
502 virtual int print(char *, size_t, DataType ty = TYPE_NONE) const;
503
504 public:
505 unsigned ssa : 1;
506
507 int affinity;
508 };
509
510 class Symbol : public Value
511 {
512 public:
513 Symbol(Program *, DataFile file = FILE_MEMORY_CONST, ubyte fileIdx = 0);
514
515 virtual Value *clone(Function *) const;
516
517 virtual bool equals(const Value *that, bool strict) const;
518
519 virtual int print(char *, size_t, DataType ty = TYPE_NONE) const;
520
521 // print with indirect values
522 int print(char *, size_t, Value *, Value *, DataType ty = TYPE_NONE) const;
523
524 inline void setFile(DataFile file, ubyte fileIndex = 0)
525 {
526 reg.file = file;
527 reg.fileIndex = fileIndex;
528 }
529
530 inline void setOffset(int32_t offset);
531 inline void setAddress(Symbol *base, int32_t offset);
532 inline void setSV(SVSemantic sv, uint32_t idx = 0);
533
534 inline const Symbol *getBase() const { return baseSym; }
535
536 private:
537 Symbol *baseSym; // array base for Symbols representing array elements
538 };
539
540 class ImmediateValue : public Value
541 {
542 public:
543 ImmediateValue(Program *, uint32_t);
544 ImmediateValue(Program *, float);
545 ImmediateValue(Program *, double);
546
547 // NOTE: not added to program with
548 ImmediateValue(const ImmediateValue *, DataType ty);
549
550 virtual bool equals(const Value *that, bool strict) const;
551
552 // these only work if 'type' is valid (we mostly use untyped literals):
553 bool isInteger(const int ival) const; // ival is cast to this' type
554 bool isNegative() const;
555 bool isPow2() const;
556
557 void applyLog2();
558
559 // for constant folding:
560 ImmediateValue operator+(const ImmediateValue&) const;
561 ImmediateValue operator-(const ImmediateValue&) const;
562 ImmediateValue operator*(const ImmediateValue&) const;
563 ImmediateValue operator/(const ImmediateValue&) const;
564
565 bool compare(CondCode cc, float fval) const;
566
567 virtual int print(char *, size_t, DataType ty = TYPE_NONE) const;
568 };
569
570 class Instruction
571 {
572 public:
573 Instruction();
574 Instruction(Function *, operation, DataType);
575 virtual ~Instruction();
576
577 virtual Instruction *clone(bool deep) const;
578
579 void setDef(int i, Value *);
580 void setSrc(int s, Value *);
581 void setSrc(int s, const ValueRef&);
582 void swapSources(int a, int b);
583 bool setIndirect(int s, int dim, Value *);
584
585 inline ValueRef& src(int s) { return srcs[s]; }
586 inline ValueDef& def(int s) { return defs[s]; }
587 inline const ValueRef& src(int s) const { return srcs[s]; }
588 inline const ValueDef& def(int s) const { return defs[s]; }
589
590 inline Value *getDef(int d) const { return defs[d].get(); }
591 inline Value *getSrc(int s) const { return srcs[s].get(); }
592 inline Value *getIndirect(int s, int dim) const;
593
594 inline bool defExists(unsigned d) const
595 {
596 return d < defs.size() && defs[d].exists();
597 }
598 inline bool srcExists(unsigned s) const
599 {
600 return s < srcs.size() && srcs[s].exists();
601 }
602
603 inline bool constrainedDefs() const { return defExists(1); }
604
605 bool setPredicate(CondCode ccode, Value *);
606 inline Value *getPredicate() const;
607 bool writesPredicate() const;
608
609 inline void setFlagsSrc(int s, Value *);
610 inline void setFlagsDef(int d, Value *);
611
612 unsigned int defCount() const { return defs.size(); };
613 unsigned int defCount(unsigned int mask) const;
614 unsigned int srcCount() const { return srcs.size(); };
615 unsigned int srcCount(unsigned int mask) const;
616
617 // save & remove / set indirect[0,1] and predicate source
618 void takeExtraSources(int s, Value *[3]);
619 void putExtraSources(int s, Value *[3]);
620
621 inline void setType(DataType type) { dType = sType = type; }
622
623 inline void setType(DataType dtype, DataType stype)
624 {
625 dType = dtype;
626 sType = stype;
627 }
628
629 inline bool isPseudo() const { return op < OP_MOV; }
630 bool isDead() const;
631 bool isNop() const;
632 bool isCommutationLegal(const Instruction *) const; // must be adjacent !
633 bool isActionEqual(const Instruction *) const;
634 bool isResultEqual(const Instruction *) const;
635
636 void print() const;
637
638 inline CmpInstruction *asCmp();
639 inline TexInstruction *asTex();
640 inline FlowInstruction *asFlow();
641 inline const TexInstruction *asTex() const;
642 inline const CmpInstruction *asCmp() const;
643 inline const FlowInstruction *asFlow() const;
644
645 public:
646 Instruction *next;
647 Instruction *prev;
648 int id;
649 int serial; // CFG order
650
651 operation op;
652 DataType dType; // destination or defining type
653 DataType sType; // source or secondary type
654 CondCode cc;
655 RoundMode rnd;
656 CacheMode cache;
657
658 uint8_t subOp; // quadop, 1 for mul-high, etc.
659
660 unsigned encSize : 4; // encoding size in bytes
661 unsigned saturate : 1; // to [0.0f, 1.0f]
662 unsigned join : 1; // converge control flow (use OP_JOIN until end)
663 unsigned fixed : 1; // prevent dead code elimination
664 unsigned terminator : 1; // end of basic block
665 unsigned atomic : 1;
666 unsigned ftz : 1; // flush denormal to zero
667 unsigned dnz : 1; // denormals, NaN are zero
668 unsigned ipa : 4; // interpolation mode
669 unsigned lanes : 4;
670 unsigned perPatch : 1;
671 unsigned exit : 1; // terminate program after insn
672
673 int8_t postFactor; // MUL/DIV(if < 0) by 1 << postFactor
674
675 int8_t predSrc;
676 int8_t flagsDef;
677 int8_t flagsSrc;
678
679 BasicBlock *bb;
680
681 protected:
682 std::deque<ValueDef> defs; // no gaps !
683 std::deque<ValueRef> srcs; // no gaps !
684
685 // instruction specific methods:
686 // (don't want to subclass, would need more constructors and memory pools)
687 public:
688 inline void setInterpolate(unsigned int mode) { ipa = mode; }
689
690 unsigned int getInterpMode() const { return ipa & 0x3; }
691 unsigned int getSampleMode() const { return ipa & 0xc; }
692
693 private:
694 void init();
695 protected:
696 void cloneBase(Instruction *clone, bool deep) const;
697 };
698
699 enum TexQuery
700 {
701 TXQ_DIMS,
702 TXQ_TYPE,
703 TXQ_SAMPLE_POSITION,
704 TXQ_FILTER,
705 TXQ_LOD,
706 TXQ_WRAP,
707 TXQ_BORDER_COLOUR
708 };
709
710 class TexInstruction : public Instruction
711 {
712 public:
713 class Target
714 {
715 public:
716 Target(TexTarget targ = TEX_TARGET_2D) : target(targ) { }
717
718 const char *getName() const { return descTable[target].name; }
719 unsigned int getArgCount() const { return descTable[target].argc; }
720 unsigned int getDim() const { return descTable[target].dim; }
721 int isArray() const { return descTable[target].array ? 1 : 0; }
722 int isCube() const { return descTable[target].cube ? 1 : 0; }
723 int isShadow() const { return descTable[target].shadow ? 1 : 0; }
724
725 Target& operator=(TexTarget targ)
726 {
727 assert(targ < TEX_TARGET_COUNT);
728 return *this;
729 }
730
731 inline bool operator==(TexTarget targ) const { return target == targ; }
732
733 private:
734 struct Desc
735 {
736 char name[19];
737 uint8_t dim;
738 uint8_t argc;
739 bool array;
740 bool cube;
741 bool shadow;
742 };
743
744 static const struct Desc descTable[TEX_TARGET_COUNT];
745
746 private:
747 enum TexTarget target;
748 };
749
750 public:
751 TexInstruction(Function *, operation);
752 virtual ~TexInstruction();
753
754 virtual Instruction *clone(bool deep) const;
755
756 inline void setTexture(Target targ, uint8_t r, uint8_t s)
757 {
758 tex.r = r;
759 tex.s = s;
760 tex.target = targ;
761 }
762
763 inline Value *getIndirectR() const;
764 inline Value *getIndirectS() const;
765
766 public:
767 struct {
768 Target target;
769
770 uint8_t r;
771 int8_t rIndirectSrc;
772 uint8_t s;
773 int8_t sIndirectSrc;
774
775 uint8_t mask;
776 uint8_t gatherComp;
777
778 bool liveOnly; // only execute on live pixels of a quad (optimization)
779 bool levelZero;
780 bool derivAll;
781
782 int8_t useOffsets; // 0, 1, or 4 for textureGatherOffsets
783 int8_t offset[4][3];
784
785 enum TexQuery query;
786 } tex;
787
788 ValueRef dPdx[3];
789 ValueRef dPdy[3];
790 };
791
792 class CmpInstruction : public Instruction
793 {
794 public:
795 CmpInstruction(Function *, operation);
796
797 virtual Instruction *clone(bool deep) const;
798
799 void setCondition(CondCode cond) { setCond = cond; }
800 CondCode getCondition() const { return setCond; }
801
802 public:
803 CondCode setCond;
804 };
805
806 class FlowInstruction : public Instruction
807 {
808 public:
809 FlowInstruction(Function *, operation, BasicBlock *target);
810
811 public:
812 unsigned allWarp : 1;
813 unsigned absolute : 1;
814 unsigned limit : 1;
815 unsigned builtin : 1; // true for calls to emulation code
816
817 union {
818 BasicBlock *bb;
819 int builtin;
820 Function *fn;
821 } target;
822 };
823
824 class BasicBlock
825 {
826 public:
827 BasicBlock(Function *);
828 ~BasicBlock();
829
830 inline int getId() const { return id; }
831 inline unsigned int getInsnCount() const { return numInsns; }
832 inline bool isTerminated() const { return exit && exit->terminator; }
833
834 bool dominatedBy(BasicBlock *bb);
835 inline bool reachableBy(BasicBlock *by, BasicBlock *term);
836
837 // returns mask of conditional out blocks
838 // e.g. 3 for IF { .. } ELSE { .. } ENDIF, 1 for IF { .. } ENDIF
839 unsigned int initiatesSimpleConditional() const;
840
841 public:
842 Function *getFunction() const { return func; }
843 Program *getProgram() const { return program; }
844
845 Instruction *getEntry() const { return entry; } // first non-phi instruction
846 Instruction *getPhi() const { return phi; }
847 Instruction *getFirst() const { return phi ? phi : entry; }
848 Instruction *getExit() const { return exit; }
849
850 void insertHead(Instruction *);
851 void insertTail(Instruction *);
852 void insertBefore(Instruction *, Instruction *);
853 void insertAfter(Instruction *, Instruction *);
854 void remove(Instruction *);
855 void permuteAdjacent(Instruction *, Instruction *);
856
857 BasicBlock *idom() const;
858
859 // NOTE: currently does not rebuild the dominator tree
860 BasicBlock *splitBefore(Instruction *, bool attach = true);
861 BasicBlock *splitAfter(Instruction *, bool attach = true);
862
863 DLList& getDF() { return df; }
864 DLList::Iterator iterDF() { return df.iterator(); }
865
866 static inline BasicBlock *get(Iterator&);
867 static inline BasicBlock *get(Graph::Node *);
868
869 public:
870 Graph::Node cfg; // first edge is branch *taken* (the ELSE branch)
871 Graph::Node dom;
872
873 BitSet liveSet;
874
875 uint32_t binPos;
876 uint32_t binSize;
877
878 Instruction *joinAt; // for quick reference
879
880 bool explicitCont; // loop headers: true if loop contains continue stmts
881
882 private:
883 int id;
884 DLList df;
885
886 Instruction *phi;
887 Instruction *entry;
888 Instruction *exit;
889
890 unsigned int numInsns;
891
892 private:
893 Function *func;
894 Program *program;
895
896 void splitCommon(Instruction *, BasicBlock *, bool attach);
897 };
898
899 class Function
900 {
901 public:
902 Function(Program *, const char *name);
903 ~Function();
904
905 inline Program *getProgram() const { return prog; }
906 inline const char *getName() const { return name; }
907 inline int getId() const { return id; }
908
909 void print();
910 void printLiveIntervals() const;
911 void printCFGraph(const char *filePath);
912
913 bool setEntry(BasicBlock *);
914 bool setExit(BasicBlock *);
915
916 unsigned int orderInstructions(ArrayList&);
917
918 inline void add(BasicBlock *bb, int& id) { allBBlocks.insert(bb, id); }
919 inline void add(Instruction *insn, int& id) { allInsns.insert(insn, id); }
920 inline void add(LValue *lval, int& id) { allLValues.insert(lval, id); }
921
922 inline LValue *getLValue(int id);
923
924 bool convertToSSA();
925
926 public:
927 Graph cfg;
928 Graph::Node *cfgExit;
929 Graph *domTree;
930 Graph::Node call; // node in the call graph
931
932 BasicBlock **bbArray; // BBs in emission order
933 int bbCount;
934
935 unsigned int loopNestingBound;
936 int regClobberMax;
937
938 uint32_t binPos;
939 uint32_t binSize;
940
941 ArrayList allBBlocks;
942 ArrayList allInsns;
943 ArrayList allLValues;
944
945 private:
946 void buildLiveSetsPreSSA(BasicBlock *, const int sequence);
947
948 private:
949 int id;
950 const char *const name;
951 Program *prog;
952 };
953
954 enum CGStage
955 {
956 CG_STAGE_PRE_SSA,
957 CG_STAGE_SSA, // expected directly before register allocation
958 CG_STAGE_POST_RA
959 };
960
961 class Program
962 {
963 public:
964 enum Type
965 {
966 TYPE_VERTEX,
967 TYPE_TESSELLATION_CONTROL,
968 TYPE_TESSELLATION_EVAL,
969 TYPE_GEOMETRY,
970 TYPE_FRAGMENT,
971 TYPE_COMPUTE
972 };
973
974 Program(Type type, Target *targ);
975 ~Program();
976
977 void print();
978
979 Type getType() const { return progType; }
980
981 inline void add(Function *fn, int& id) { allFuncs.insert(fn, id); }
982 inline void add(Value *rval, int& id) { allRValues.insert(rval, id); }
983
984 bool makeFromTGSI(struct nv50_ir_prog_info *);
985 bool makeFromSM4(struct nv50_ir_prog_info *);
986 bool convertToSSA();
987 bool optimizeSSA(int level);
988 bool optimizePostRA(int level);
989 bool registerAllocation();
990 bool emitBinary(struct nv50_ir_prog_info *);
991
992 const Target *getTarget() const { return target; }
993
994 private:
995 Type progType;
996 Target *target;
997
998 public:
999 Function *main;
1000 Graph calls;
1001
1002 ArrayList allFuncs;
1003 ArrayList allRValues;
1004
1005 uint32_t *code;
1006 uint32_t binSize;
1007
1008 int maxGPR;
1009
1010 MemoryPool mem_Instruction;
1011 MemoryPool mem_CmpInstruction;
1012 MemoryPool mem_TexInstruction;
1013 MemoryPool mem_FlowInstruction;
1014 MemoryPool mem_LValue;
1015 MemoryPool mem_Symbol;
1016 MemoryPool mem_ImmediateValue;
1017
1018 uint32_t dbgFlags;
1019
1020 void releaseInstruction(Instruction *);
1021 void releaseValue(Value *);
1022 };
1023
1024 // TODO: add const version
1025 class Pass
1026 {
1027 public:
1028 bool run(Program *, bool ordered = false, bool skipPhi = false);
1029 bool run(Function *, bool ordered = false, bool skipPhi = false);
1030
1031 private:
1032 // return false to continue with next entity on next higher level
1033 virtual bool visit(Function *) { return true; }
1034 virtual bool visit(BasicBlock *) { return true; }
1035 virtual bool visit(Instruction *) { return false; }
1036
1037 bool doRun(Program *, bool ordered, bool skipPhi);
1038 bool doRun(Function *, bool ordered, bool skipPhi);
1039
1040 protected:
1041 bool err;
1042 Function *func;
1043 Program *prog;
1044 };
1045
1046 // =============================================================================
1047
1048 #include "nv50_ir_inlines.h"
1049
1050 } // namespace nv50_ir
1051
1052 #endif // __NV50_IR_H__