i965/fs: Revisit NIR atomic counter intrinsic translation.
[mesa.git] / src / mesa / drivers / dri / i965 / brw_fs.h
1 /*
2 * Copyright © 2010 Intel Corporation
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
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21 * IN THE SOFTWARE.
22 *
23 * Authors:
24 * Eric Anholt <eric@anholt.net>
25 *
26 */
27
28 #pragma once
29
30 #include "brw_shader.h"
31 #include "brw_ir_fs.h"
32 #include "brw_fs_builder.h"
33
34 extern "C" {
35
36 #include <sys/types.h>
37
38 #include "main/macros.h"
39 #include "main/shaderobj.h"
40 #include "main/uniforms.h"
41 #include "program/prog_parameter.h"
42 #include "program/prog_print.h"
43 #include "program/prog_optimize.h"
44 #include "util/register_allocate.h"
45 #include "program/hash_table.h"
46 #include "brw_context.h"
47 #include "brw_eu.h"
48 #include "brw_wm.h"
49 #include "intel_asm_annotation.h"
50 }
51 #include "glsl/glsl_types.h"
52 #include "glsl/ir.h"
53 #include "glsl/nir/nir.h"
54 #include "program/sampler.h"
55
56 struct bblock_t;
57 namespace {
58 struct acp_entry;
59 }
60
61 namespace brw {
62 class fs_live_variables;
63 }
64
65 static inline fs_reg
66 offset(fs_reg reg, const brw::fs_builder& bld, unsigned delta)
67 {
68 switch (reg.file) {
69 case BAD_FILE:
70 break;
71 case GRF:
72 case MRF:
73 case HW_REG:
74 case ATTR:
75 return byte_offset(reg,
76 delta * reg.component_size(bld.dispatch_width()));
77 case UNIFORM:
78 reg.reg_offset += delta;
79 break;
80 case IMM:
81 assert(delta == 0);
82 }
83 return reg;
84 }
85
86 /**
87 * The fragment shader front-end.
88 *
89 * Translates either GLSL IR or Mesa IR (for ARB_fragment_program) into FS IR.
90 */
91 class fs_visitor : public backend_shader
92 {
93 public:
94 fs_visitor(const struct brw_compiler *compiler, void *log_data,
95 void *mem_ctx,
96 gl_shader_stage stage,
97 const void *key,
98 struct brw_stage_prog_data *prog_data,
99 struct gl_shader_program *shader_prog,
100 struct gl_program *prog,
101 unsigned dispatch_width,
102 int shader_time_index);
103
104 ~fs_visitor();
105
106 fs_reg vgrf(const glsl_type *const type);
107 void import_uniforms(fs_visitor *v);
108 void setup_uniform_clipplane_values(gl_clip_plane *clip_planes);
109 void compute_clip_distance(gl_clip_plane *clip_planes);
110
111 uint32_t gather_channel(int orig_chan, uint32_t sampler);
112 void swizzle_result(ir_texture_opcode op, int dest_components,
113 fs_reg orig_val, uint32_t sampler);
114
115 int type_size(const struct glsl_type *type);
116 fs_inst *get_instruction_generating_reg(fs_inst *start,
117 fs_inst *end,
118 const fs_reg &reg);
119
120 void VARYING_PULL_CONSTANT_LOAD(const brw::fs_builder &bld,
121 const fs_reg &dst,
122 const fs_reg &surf_index,
123 const fs_reg &varying_offset,
124 uint32_t const_offset);
125 void DEP_RESOLVE_MOV(const brw::fs_builder &bld, int grf);
126
127 bool run_fs(bool do_rep_send);
128 bool run_vs(gl_clip_plane *clip_planes);
129 bool run_cs();
130 void optimize();
131 void allocate_registers();
132 void assign_binding_table_offsets();
133 void setup_payload_gen4();
134 void setup_payload_gen6();
135 void setup_vs_payload();
136 void setup_cs_payload();
137 void fixup_3src_null_dest();
138 void assign_curb_setup();
139 void calculate_urb_setup();
140 void assign_urb_setup();
141 void assign_vs_urb_setup();
142 bool assign_regs(bool allow_spilling);
143 void assign_regs_trivial();
144 void setup_payload_interference(struct ra_graph *g, int payload_reg_count,
145 int first_payload_node);
146 int choose_spill_reg(struct ra_graph *g);
147 void spill_reg(int spill_reg);
148 void split_virtual_grfs();
149 bool compact_virtual_grfs();
150 void move_uniform_array_access_to_pull_constants();
151 void assign_constant_locations();
152 void demote_pull_constants();
153 void invalidate_live_intervals();
154 void calculate_live_intervals();
155 void calculate_register_pressure();
156 bool opt_algebraic();
157 bool opt_redundant_discard_jumps();
158 bool opt_cse();
159 bool opt_cse_local(bblock_t *block);
160 bool opt_copy_propagate();
161 bool try_copy_propagate(fs_inst *inst, int arg, acp_entry *entry);
162 bool try_constant_propagate(fs_inst *inst, acp_entry *entry);
163 bool opt_copy_propagate_local(void *mem_ctx, bblock_t *block,
164 exec_list *acp);
165 bool opt_register_renaming();
166 bool register_coalesce();
167 bool compute_to_mrf();
168 bool eliminate_find_live_channel();
169 bool dead_code_eliminate();
170 bool remove_duplicate_mrf_writes();
171
172 bool opt_sampler_eot();
173 bool virtual_grf_interferes(int a, int b);
174 void schedule_instructions(instruction_scheduler_mode mode);
175 void insert_gen4_send_dependency_workarounds();
176 void insert_gen4_pre_send_dependency_workarounds(bblock_t *block,
177 fs_inst *inst);
178 void insert_gen4_post_send_dependency_workarounds(bblock_t *block,
179 fs_inst *inst);
180 void vfail(const char *msg, va_list args);
181 void fail(const char *msg, ...);
182 void no16(const char *msg);
183 void lower_uniform_pull_constant_loads();
184 bool lower_load_payload();
185 bool lower_logical_sends();
186 bool lower_integer_multiplication();
187 bool lower_simd_width();
188 bool opt_combine_constants();
189
190 void emit_dummy_fs();
191 void emit_repclear_shader();
192 fs_reg *emit_fragcoord_interpolation(bool pixel_center_integer,
193 bool origin_upper_left);
194 fs_inst *emit_linterp(const fs_reg &attr, const fs_reg &interp,
195 glsl_interp_qualifier interpolation_mode,
196 bool is_centroid, bool is_sample);
197 fs_reg *emit_frontfacing_interpolation();
198 fs_reg *emit_samplepos_setup();
199 fs_reg *emit_sampleid_setup();
200 void emit_general_interpolation(fs_reg attr, const char *name,
201 const glsl_type *type,
202 glsl_interp_qualifier interpolation_mode,
203 int location, bool mod_centroid,
204 bool mod_sample);
205 fs_reg *emit_vs_system_value(int location);
206 void emit_interpolation_setup_gen4();
207 void emit_interpolation_setup_gen6();
208 void compute_sample_position(fs_reg dst, fs_reg int_sample_pos);
209 fs_reg rescale_texcoord(fs_reg coordinate, int coord_components,
210 bool is_rect, uint32_t sampler, int texunit);
211 void emit_texture(ir_texture_opcode op,
212 const glsl_type *dest_type,
213 fs_reg coordinate, int components,
214 fs_reg shadow_c,
215 fs_reg lod, fs_reg dpdy, int grad_components,
216 fs_reg sample_index,
217 fs_reg offset,
218 fs_reg mcs,
219 int gather_component,
220 bool is_cube_array,
221 bool is_rect,
222 uint32_t sampler,
223 fs_reg sampler_reg,
224 int texunit);
225 fs_reg emit_mcs_fetch(const fs_reg &coordinate, unsigned components,
226 const fs_reg &sampler);
227 void emit_gen6_gather_wa(uint8_t wa, fs_reg dst);
228 void resolve_source_modifiers(fs_reg *src);
229 void emit_discard_jump();
230 bool try_replace_with_sel();
231 bool opt_peephole_sel();
232 bool opt_peephole_predicated_break();
233 bool opt_saturate_propagation();
234 bool opt_cmod_propagation();
235 bool opt_zero_samples();
236 void emit_unspill(bblock_t *block, fs_inst *inst, fs_reg reg,
237 uint32_t spill_offset, int count);
238 void emit_spill(bblock_t *block, fs_inst *inst, fs_reg reg,
239 uint32_t spill_offset, int count);
240
241 void emit_nir_code();
242 void nir_setup_inputs(nir_shader *shader);
243 void nir_setup_outputs(nir_shader *shader);
244 void nir_setup_uniforms(nir_shader *shader);
245 void nir_setup_uniform(nir_variable *var);
246 void nir_setup_builtin_uniform(nir_variable *var);
247 void nir_emit_system_values(nir_shader *shader);
248 void nir_emit_impl(nir_function_impl *impl);
249 void nir_emit_cf_list(exec_list *list);
250 void nir_emit_if(nir_if *if_stmt);
251 void nir_emit_loop(nir_loop *loop);
252 void nir_emit_block(nir_block *block);
253 void nir_emit_instr(nir_instr *instr);
254 void nir_emit_alu(const brw::fs_builder &bld, nir_alu_instr *instr);
255 void nir_emit_load_const(const brw::fs_builder &bld,
256 nir_load_const_instr *instr);
257 void nir_emit_undef(const brw::fs_builder &bld,
258 nir_ssa_undef_instr *instr);
259 void nir_emit_intrinsic(const brw::fs_builder &bld,
260 nir_intrinsic_instr *instr);
261 void nir_emit_texture(const brw::fs_builder &bld,
262 nir_tex_instr *instr);
263 void nir_emit_jump(const brw::fs_builder &bld,
264 nir_jump_instr *instr);
265 fs_reg get_nir_src(nir_src src);
266 fs_reg get_nir_dest(nir_dest dest);
267 void emit_percomp(const brw::fs_builder &bld, const fs_inst &inst,
268 unsigned wr_mask);
269
270 bool optimize_frontfacing_ternary(nir_alu_instr *instr,
271 const fs_reg &result);
272
273 void emit_alpha_test();
274 fs_inst *emit_single_fb_write(const brw::fs_builder &bld,
275 fs_reg color1, fs_reg color2,
276 fs_reg src0_alpha, unsigned components);
277 void emit_fb_writes();
278 void emit_urb_writes();
279 void emit_cs_terminate();
280
281 void emit_barrier();
282
283 void emit_shader_time_begin();
284 void emit_shader_time_end();
285 void SHADER_TIME_ADD(const brw::fs_builder &bld,
286 int shader_time_subindex,
287 fs_reg value);
288
289 void emit_untyped_atomic(unsigned atomic_op, unsigned surf_index,
290 fs_reg dst, fs_reg offset, fs_reg src0,
291 fs_reg src1);
292
293 void emit_untyped_surface_read(unsigned surf_index, fs_reg dst,
294 fs_reg offset);
295
296 fs_reg get_timestamp(const brw::fs_builder &bld);
297
298 struct brw_reg interp_reg(int location, int channel);
299
300 virtual void setup_vector_uniform_values(const gl_constant_value *values,
301 unsigned n);
302
303 int implied_mrf_writes(fs_inst *inst);
304
305 virtual void dump_instructions();
306 virtual void dump_instructions(const char *name);
307 void dump_instruction(backend_instruction *inst);
308 void dump_instruction(backend_instruction *inst, FILE *file);
309
310 const void *const key;
311 const struct brw_sampler_prog_key_data *key_tex;
312
313 struct brw_stage_prog_data *prog_data;
314 unsigned int sanity_param_count;
315
316 int *param_size;
317
318 int *virtual_grf_start;
319 int *virtual_grf_end;
320 brw::fs_live_variables *live_intervals;
321
322 int *regs_live_at_ip;
323
324 /** Number of uniform variable components visited. */
325 unsigned uniforms;
326
327 /** Total number of direct uniforms we can get from NIR */
328 unsigned num_direct_uniforms;
329
330 /** Byte-offset for the next available spot in the scratch space buffer. */
331 unsigned last_scratch;
332
333 /**
334 * Array mapping UNIFORM register numbers to the pull parameter index,
335 * or -1 if this uniform register isn't being uploaded as a pull constant.
336 */
337 int *pull_constant_loc;
338
339 /**
340 * Array mapping UNIFORM register numbers to the push parameter index,
341 * or -1 if this uniform register isn't being uploaded as a push constant.
342 */
343 int *push_constant_loc;
344
345 fs_reg frag_depth;
346 fs_reg sample_mask;
347 fs_reg outputs[VARYING_SLOT_MAX];
348 unsigned output_components[VARYING_SLOT_MAX];
349 fs_reg dual_src_output;
350 bool do_dual_src;
351 int first_non_payload_grf;
352 /** Either BRW_MAX_GRF or GEN7_MRF_HACK_START */
353 unsigned max_grf;
354
355 fs_reg *nir_locals;
356 fs_reg *nir_ssa_values;
357 fs_reg nir_inputs;
358 fs_reg nir_outputs;
359 fs_reg *nir_system_values;
360
361 bool failed;
362 char *fail_msg;
363 bool simd16_unsupported;
364 char *no16_msg;
365
366 /* Result of last visit() method. Still used by emit_texture() */
367 fs_reg result;
368
369 /** Register numbers for thread payload fields. */
370 struct thread_payload {
371 uint8_t source_depth_reg;
372 uint8_t source_w_reg;
373 uint8_t aa_dest_stencil_reg;
374 uint8_t dest_depth_reg;
375 uint8_t sample_pos_reg;
376 uint8_t sample_mask_in_reg;
377 uint8_t barycentric_coord_reg[BRW_WM_BARYCENTRIC_INTERP_MODE_COUNT];
378
379 /** The number of thread payload registers the hardware will supply. */
380 uint8_t num_regs;
381 } payload;
382
383 bool source_depth_to_render_target;
384 bool runtime_check_aads_emit;
385
386 fs_reg pixel_x;
387 fs_reg pixel_y;
388 fs_reg wpos_w;
389 fs_reg pixel_w;
390 fs_reg delta_xy[BRW_WM_BARYCENTRIC_INTERP_MODE_COUNT];
391 fs_reg shader_start_time;
392 fs_reg userplane[MAX_CLIP_PLANES];
393
394 unsigned grf_used;
395 bool spilled_any_registers;
396
397 const unsigned dispatch_width; /**< 8 or 16 */
398
399 int shader_time_index;
400
401 unsigned promoted_constants;
402 brw::fs_builder bld;
403 };
404
405 /**
406 * The fragment shader code generator.
407 *
408 * Translates FS IR to actual i965 assembly code.
409 */
410 class fs_generator
411 {
412 public:
413 fs_generator(const struct brw_compiler *compiler, void *log_data,
414 void *mem_ctx,
415 const void *key,
416 struct brw_stage_prog_data *prog_data,
417 struct gl_program *fp,
418 unsigned promoted_constants,
419 bool runtime_check_aads_emit,
420 const char *stage_abbrev);
421 ~fs_generator();
422
423 void enable_debug(const char *shader_name);
424 int generate_code(const cfg_t *cfg, int dispatch_width);
425 const unsigned *get_assembly(unsigned int *assembly_size);
426
427 private:
428 void fire_fb_write(fs_inst *inst,
429 struct brw_reg payload,
430 struct brw_reg implied_header,
431 GLuint nr);
432 void generate_fb_write(fs_inst *inst, struct brw_reg payload);
433 void generate_urb_write(fs_inst *inst, struct brw_reg payload);
434 void generate_cs_terminate(fs_inst *inst, struct brw_reg payload);
435 void generate_barrier(fs_inst *inst, struct brw_reg src);
436 void generate_blorp_fb_write(fs_inst *inst);
437 void generate_linterp(fs_inst *inst, struct brw_reg dst,
438 struct brw_reg *src);
439 void generate_tex(fs_inst *inst, struct brw_reg dst, struct brw_reg src,
440 struct brw_reg sampler_index);
441 void generate_math_gen6(fs_inst *inst,
442 struct brw_reg dst,
443 struct brw_reg src0,
444 struct brw_reg src1);
445 void generate_math_gen4(fs_inst *inst,
446 struct brw_reg dst,
447 struct brw_reg src);
448 void generate_math_g45(fs_inst *inst,
449 struct brw_reg dst,
450 struct brw_reg src);
451 void generate_ddx(enum opcode op, struct brw_reg dst, struct brw_reg src);
452 void generate_ddy(enum opcode op, struct brw_reg dst, struct brw_reg src,
453 bool negate_value);
454 void generate_scratch_write(fs_inst *inst, struct brw_reg src);
455 void generate_scratch_read(fs_inst *inst, struct brw_reg dst);
456 void generate_scratch_read_gen7(fs_inst *inst, struct brw_reg dst);
457 void generate_uniform_pull_constant_load(fs_inst *inst, struct brw_reg dst,
458 struct brw_reg index,
459 struct brw_reg offset);
460 void generate_uniform_pull_constant_load_gen7(fs_inst *inst,
461 struct brw_reg dst,
462 struct brw_reg surf_index,
463 struct brw_reg offset);
464 void generate_varying_pull_constant_load(fs_inst *inst, struct brw_reg dst,
465 struct brw_reg index,
466 struct brw_reg offset);
467 void generate_varying_pull_constant_load_gen7(fs_inst *inst,
468 struct brw_reg dst,
469 struct brw_reg index,
470 struct brw_reg offset);
471 void generate_mov_dispatch_to_flags(fs_inst *inst);
472
473 void generate_pixel_interpolator_query(fs_inst *inst,
474 struct brw_reg dst,
475 struct brw_reg src,
476 struct brw_reg msg_data,
477 unsigned msg_type);
478
479 void generate_set_sample_id(fs_inst *inst,
480 struct brw_reg dst,
481 struct brw_reg src0,
482 struct brw_reg src1);
483
484 void generate_set_simd4x2_offset(fs_inst *inst,
485 struct brw_reg dst,
486 struct brw_reg offset);
487 void generate_discard_jump(fs_inst *inst);
488
489 void generate_pack_half_2x16_split(fs_inst *inst,
490 struct brw_reg dst,
491 struct brw_reg x,
492 struct brw_reg y);
493 void generate_unpack_half_2x16_split(fs_inst *inst,
494 struct brw_reg dst,
495 struct brw_reg src);
496
497 void generate_shader_time_add(fs_inst *inst,
498 struct brw_reg payload,
499 struct brw_reg offset,
500 struct brw_reg value);
501
502 bool patch_discard_jumps_to_fb_writes();
503
504 const struct brw_compiler *compiler;
505 void *log_data; /* Passed to compiler->*_log functions */
506
507 const struct brw_device_info *devinfo;
508
509 struct brw_codegen *p;
510 const void * const key;
511 struct brw_stage_prog_data * const prog_data;
512
513 const struct gl_program *prog;
514
515 unsigned dispatch_width; /**< 8 or 16 */
516
517 exec_list discard_halt_patches;
518 unsigned promoted_constants;
519 bool runtime_check_aads_emit;
520 bool debug_flag;
521 const char *shader_name;
522 const char *stage_abbrev;
523 void *mem_ctx;
524 };
525
526 bool brw_do_channel_expressions(struct exec_list *instructions);
527 bool brw_do_vector_splitting(struct exec_list *instructions);
528 void brw_setup_tex_for_precompile(struct brw_context *brw,
529 struct brw_sampler_prog_key_data *tex,
530 struct gl_program *prog);