freedreno/ir3: code-motion
[mesa.git] / src / freedreno / ir3 / ir3_context.h
1 /*
2 * Copyright (C) 2015-2018 Rob Clark <robclark@freedesktop.org>
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 (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 * SOFTWARE.
22 *
23 * Authors:
24 * Rob Clark <robclark@freedesktop.org>
25 */
26
27 #ifndef IR3_CONTEXT_H_
28 #define IR3_CONTEXT_H_
29
30 #include "ir3_nir.h"
31 #include "ir3.h"
32
33 /* for conditionally setting boolean flag(s): */
34 #define COND(bool, val) ((bool) ? (val) : 0)
35
36 #define DBG(fmt, ...) \
37 do { debug_printf("%s:%d: "fmt "\n", \
38 __FUNCTION__, __LINE__, ##__VA_ARGS__); } while (0)
39
40 /**
41 * The context for compilation of a single shader.
42 */
43 struct ir3_context {
44 struct ir3_compiler *compiler;
45
46 struct nir_shader *s;
47
48 struct nir_instr *cur_instr; /* current instruction, just for debug */
49
50 struct ir3 *ir;
51 struct ir3_shader_variant *so;
52
53 struct ir3_block *block; /* the current block */
54 struct ir3_block *in_block; /* block created for shader inputs */
55
56 nir_function_impl *impl;
57
58 /* For fragment shaders, varyings are not actual shader inputs,
59 * instead the hw passes a varying-coord which is used with
60 * bary.f.
61 *
62 * But NIR doesn't know that, it still declares varyings as
63 * inputs. So we do all the input tracking normally and fix
64 * things up after compile_instructions()
65 *
66 * NOTE that frag_vcoord is the hardware position (possibly it
67 * is actually an index or tag or some such.. it is *not*
68 * values that can be directly used for gl_FragCoord..)
69 */
70 struct ir3_instruction *frag_vcoord;
71
72 /* for fragment shaders, for gl_FrontFacing and gl_FragCoord: */
73 struct ir3_instruction *frag_face, *frag_coord;
74
75 /* For vertex shaders, keep track of the system values sources */
76 struct ir3_instruction *vertex_id, *basevertex, *instance_id;
77
78 /* For fragment shaders: */
79 struct ir3_instruction *samp_id, *samp_mask_in;
80
81 /* Compute shader inputs: */
82 struct ir3_instruction *local_invocation_id, *work_group_id;
83
84 /* mapping from nir_register to defining instruction: */
85 struct hash_table *def_ht;
86
87 unsigned num_arrays;
88
89 /* a common pattern for indirect addressing is to request the
90 * same address register multiple times. To avoid generating
91 * duplicate instruction sequences (which our backend does not
92 * try to clean up, since that should be done as the NIR stage)
93 * we cache the address value generated for a given src value:
94 *
95 * Note that we have to cache these per alignment, since same
96 * src used for an array of vec1 cannot be also used for an
97 * array of vec4.
98 */
99 struct hash_table *addr_ht[4];
100
101 /* last dst array, for indirect we need to insert a var-store.
102 */
103 struct ir3_instruction **last_dst;
104 unsigned last_dst_n;
105
106 /* maps nir_block to ir3_block, mostly for the purposes of
107 * figuring out the blocks successors
108 */
109 struct hash_table *block_ht;
110
111 /* on a4xx, bitmask of samplers which need astc+srgb workaround: */
112 unsigned astc_srgb;
113
114 unsigned samples; /* bitmask of x,y sample shifts */
115
116 unsigned max_texture_index;
117
118 /* set if we encounter something we can't handle yet, so we
119 * can bail cleanly and fallback to TGSI compiler f/e
120 */
121 bool error;
122 };
123
124 struct ir3_context * ir3_context_init(struct ir3_compiler *compiler,
125 struct ir3_shader_variant *so);
126 void ir3_context_free(struct ir3_context *ctx);
127
128 /* gpu pointer size in units of 32bit registers/slots */
129 static inline
130 unsigned ir3_pointer_size(struct ir3_context *ctx)
131 {
132 return (ctx->compiler->gpu_id >= 500) ? 2 : 1;
133 }
134
135 struct ir3_instruction ** ir3_get_dst_ssa(struct ir3_context *ctx, nir_ssa_def *dst, unsigned n);
136 struct ir3_instruction ** ir3_get_dst(struct ir3_context *ctx, nir_dest *dst, unsigned n);
137 struct ir3_instruction * const * ir3_get_src(struct ir3_context *ctx, nir_src *src);
138 void put_dst(struct ir3_context *ctx, nir_dest *dst);
139 struct ir3_instruction * ir3_create_collect(struct ir3_context *ctx,
140 struct ir3_instruction *const *arr, unsigned arrsz);
141 void ir3_split_dest(struct ir3_block *block, struct ir3_instruction **dst,
142 struct ir3_instruction *src, unsigned base, unsigned n);
143
144 void ir3_context_error(struct ir3_context *ctx, const char *format, ...);
145
146 #define compile_assert(ctx, cond) do { \
147 if (!(cond)) ir3_context_error((ctx), "failed assert: "#cond"\n"); \
148 } while (0)
149
150 struct ir3_instruction * ir3_get_addr(struct ir3_context *ctx,
151 struct ir3_instruction *src, int align);
152 struct ir3_instruction * ir3_get_predicate(struct ir3_context *ctx,
153 struct ir3_instruction *src);
154
155 void ir3_declare_array(struct ir3_context *ctx, nir_register *reg);
156 struct ir3_array * ir3_get_array(struct ir3_context *ctx, nir_register *reg);
157 struct ir3_instruction *ir3_create_array_load(struct ir3_context *ctx,
158 struct ir3_array *arr, int n, struct ir3_instruction *address);
159 void ir3_create_array_store(struct ir3_context *ctx, struct ir3_array *arr, int n,
160 struct ir3_instruction *src, struct ir3_instruction *address);
161
162 static inline type_t utype_for_size(unsigned bit_size)
163 {
164 switch (bit_size) {
165 case 32: return TYPE_U32;
166 case 16: return TYPE_U16;
167 case 8: return TYPE_U8;
168 default: unreachable("bad bitsize"); return ~0;
169 }
170 }
171
172 static inline type_t utype_src(nir_src src)
173 { return utype_for_size(nir_src_bit_size(src)); }
174
175 static inline type_t utype_dst(nir_dest dst)
176 { return utype_for_size(nir_dest_bit_size(dst)); }
177
178 #endif /* IR3_CONTEXT_H_ */