[Mesa-dev] [PATCH 022/133] i965/fs: add a NIR frontend
Jason Ekstrand
jason at jlekstrand.net
Mon Dec 15 22:04:32 PST 2014
From: Connor Abbott <connor.abbott at intel.com>
This is similar to the GLSL IR frontend, except consuming NIR. This lets
us test NIR as part of an actual compiler.
v2: Jason Ekstrand <jason.ekstrand at intel.com>:
Make brw_fs_nir build again
Only use NIR of INTEL_USE_NIR is set
whitespace fixes
---
src/mesa/drivers/dri/i965/Makefile.sources | 1 +
src/mesa/drivers/dri/i965/brw_fs.cpp | 12 +-
src/mesa/drivers/dri/i965/brw_fs.h | 45 +
src/mesa/drivers/dri/i965/brw_fs_nir.cpp | 1702 ++++++++++++++++++++++++++
src/mesa/drivers/dri/i965/brw_fs_visitor.cpp | 2 +
5 files changed, 1758 insertions(+), 4 deletions(-)
create mode 100644 src/mesa/drivers/dri/i965/brw_fs_nir.cpp
diff --git a/src/mesa/drivers/dri/i965/Makefile.sources b/src/mesa/drivers/dri/i965/Makefile.sources
index 989a165..3b72955 100644
--- a/src/mesa/drivers/dri/i965/Makefile.sources
+++ b/src/mesa/drivers/dri/i965/Makefile.sources
@@ -48,6 +48,7 @@ i965_FILES = \
brw_fs.h \
brw_fs_live_variables.cpp \
brw_fs_live_variables.h \
+ brw_fs_nir.cpp \
brw_fs_peephole_predicated_break.cpp \
brw_fs_reg_allocate.cpp \
brw_fs_register_coalesce.cpp \
diff --git a/src/mesa/drivers/dri/i965/brw_fs.cpp b/src/mesa/drivers/dri/i965/brw_fs.cpp
index 5de862c..a059052 100644
--- a/src/mesa/drivers/dri/i965/brw_fs.cpp
+++ b/src/mesa/drivers/dri/i965/brw_fs.cpp
@@ -3710,10 +3710,14 @@ fs_visitor::run_fs()
* functions called "main").
*/
if (shader) {
- foreach_in_list(ir_instruction, ir, shader->base.ir) {
- base_ir = ir;
- this->result = reg_undef;
- ir->accept(this);
+ if (getenv("INTEL_USE_NIR") != NULL) {
+ emit_nir_code();
+ } else {
+ foreach_in_list(ir_instruction, ir, shader->base.ir) {
+ base_ir = ir;
+ this->result = reg_undef;
+ ir->accept(this);
+ }
}
} else {
emit_fragment_program_code();
diff --git a/src/mesa/drivers/dri/i965/brw_fs.h b/src/mesa/drivers/dri/i965/brw_fs.h
index 0559d00..918008d 100644
--- a/src/mesa/drivers/dri/i965/brw_fs.h
+++ b/src/mesa/drivers/dri/i965/brw_fs.h
@@ -50,6 +50,7 @@ extern "C" {
}
#include "glsl/glsl_types.h"
#include "glsl/ir.h"
+#include "glsl/nir/nir.h"
#define MAX_SAMPLER_MESSAGE_SIZE 11
#define MAX_VGRF_SIZE 16
@@ -563,6 +564,45 @@ public:
const struct prog_instruction *fpi,
fs_reg dst, fs_reg src0, fs_reg src1, fs_reg one);
+ void emit_nir_code();
+ void nir_setup_inputs(nir_shader *shader);
+ void nir_setup_outputs(nir_shader *shader);
+ void nir_setup_uniforms(nir_shader *shader);
+ void nir_setup_registers(exec_list *regs);
+ void nir_emit_interpolation(nir_variable *var, fs_reg *reg);
+ void nir_setup_uniform(nir_variable *var);
+ void nir_setup_builtin_uniform(nir_variable *var);
+ void nir_emit_impl(nir_function_impl *impl);
+ void nir_emit_cf_list(exec_list *list);
+ void nir_emit_if(nir_if *if_stmt);
+ void nir_emit_loop(nir_loop *loop);
+ void nir_emit_block(nir_block *block);
+ void nir_emit_instr(nir_instr *instr);
+ void nir_emit_alu(nir_alu_instr *instr);
+ void nir_emit_intrinsic(nir_intrinsic_instr *instr);
+ void nir_emit_texture(nir_tex_instr *instr);
+ void nir_emit_load_const(nir_load_const_instr *instr);
+ void nir_emit_jump(nir_jump_instr *instr);
+ fs_reg get_nir_src(nir_src src);
+ fs_reg get_nir_alu_src(nir_alu_instr *instr, unsigned src);
+ fs_reg get_nir_dest(nir_dest dest);
+ void emit_percomp(fs_inst *inst, unsigned wr_mask);
+ void emit_percomp(enum opcode op, fs_reg dest, fs_reg src0,
+ unsigned wr_mask, bool saturate = false,
+ enum brw_predicate predicate = BRW_PREDICATE_NONE,
+ enum brw_conditional_mod mod = BRW_CONDITIONAL_NONE);
+ void emit_percomp(enum opcode op, fs_reg dest, fs_reg src0, fs_reg src1,
+ unsigned wr_mask, bool saturate = false,
+ enum brw_predicate predicate = BRW_PREDICATE_NONE,
+ enum brw_conditional_mod mod = BRW_CONDITIONAL_NONE);
+ void emit_math_percomp(enum opcode op, fs_reg dest, fs_reg src0,
+ unsigned wr_mask, bool saturate = false);
+ void emit_math_percomp(enum opcode op, fs_reg dest, fs_reg src0,
+ fs_reg src1, unsigned wr_mask,
+ bool saturate = false);
+ void emit_reduction(enum opcode op, fs_reg dest, fs_reg src,
+ unsigned num_components);
+
int setup_color_payload(fs_reg *dst, fs_reg color, unsigned components);
void emit_alpha_test();
fs_inst *emit_single_fb_write(fs_reg color1, fs_reg color2,
@@ -655,6 +695,11 @@ public:
fs_reg *fp_temp_regs;
fs_reg *fp_input_regs;
+ struct hash_table *nir_reg_ht;
+ fs_reg nir_inputs;
+ fs_reg nir_outputs;
+ fs_reg nir_uniforms;
+
/** @{ debug annotation info */
const char *current_annotation;
const void *base_ir;
diff --git a/src/mesa/drivers/dri/i965/brw_fs_nir.cpp b/src/mesa/drivers/dri/i965/brw_fs_nir.cpp
new file mode 100644
index 0000000..ac79064
--- /dev/null
+++ b/src/mesa/drivers/dri/i965/brw_fs_nir.cpp
@@ -0,0 +1,1702 @@
+/*
+ * Copyright © 2010 Intel Corporation
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+ * IN THE SOFTWARE.
+ */
+
+#include "glsl/nir/glsl_to_nir.h"
+#include "brw_fs.h"
+
+static glsl_interp_qualifier
+determine_interpolation_mode(nir_variable *var, bool flat_shade)
+{
+ if (var->data.interpolation != INTERP_QUALIFIER_NONE)
+ return (glsl_interp_qualifier) var->data.interpolation;
+ int location = var->data.location;
+ bool is_gl_Color =
+ location == VARYING_SLOT_COL0 || location == VARYING_SLOT_COL1;
+ if (flat_shade && is_gl_Color)
+ return INTERP_QUALIFIER_FLAT;
+ else
+ return INTERP_QUALIFIER_SMOOTH;
+}
+
+void
+fs_visitor::emit_nir_code()
+{
+ /* first, lower the GLSL IR shader to NIR */
+ nir_shader *nir = glsl_to_nir(shader->base.ir, NULL, true);
+ nir_validate_shader(nir);
+
+ /* lower some of the GLSL-isms into NIR-isms - after this point, we no
+ * longer have to deal with variables inside the shader
+ */
+
+ nir_lower_variables_scalar(nir, true, true, true, true);
+ nir_validate_shader(nir);
+
+ nir_lower_samplers(nir, shader_prog, shader->base.Program);
+ nir_validate_shader(nir);
+
+ nir_lower_system_values(nir);
+ nir_validate_shader(nir);
+
+ nir_lower_atomics(nir);
+ nir_validate_shader(nir);
+
+ nir_remove_dead_variables(nir);
+ nir_opt_global_to_local(nir);
+ nir_validate_shader(nir);
+
+ if (1)
+ nir_print_shader(nir, stderr);
+
+ /* emit the arrays used for inputs and outputs - load/store intrinsics will
+ * be converted to reads/writes of these arrays
+ */
+
+ if (nir->num_inputs > 0) {
+ nir_inputs = fs_reg(GRF, virtual_grf_alloc(nir->num_inputs));
+ nir_setup_inputs(nir);
+ }
+
+ if (nir->num_outputs > 0) {
+ nir_outputs = fs_reg(GRF, virtual_grf_alloc(nir->num_outputs));
+ nir_setup_outputs(nir);
+ }
+
+ if (nir->num_uniforms > 0) {
+ nir_uniforms = fs_reg(UNIFORM, 0);
+ nir_setup_uniforms(nir);
+ }
+
+ nir_setup_registers(&nir->registers);
+
+ /* get the main function and emit it */
+ nir_foreach_overload(nir, overload) {
+ assert(strcmp(overload->function->name, "main") == 0);
+ assert(overload->impl);
+ nir_emit_impl(overload->impl);
+ }
+
+ ralloc_free(nir);
+}
+
+void
+fs_visitor::nir_setup_inputs(nir_shader *shader)
+{
+ fs_reg varying = nir_inputs;
+
+ struct hash_entry *entry;
+ hash_table_foreach(shader->inputs, entry) {
+ nir_variable *var = (nir_variable *) entry->data;
+ varying.reg_offset = var->data.driver_location;
+
+ fs_reg reg;
+ if (!strcmp(var->name, "gl_FragCoord")) {
+ reg = *emit_fragcoord_interpolation(var->data.pixel_center_integer,
+ var->data.origin_upper_left);
+ emit_percomp(MOV(varying, reg), 0xF);
+ } else if (!strcmp(var->name, "gl_FrontFacing")) {
+ reg = *emit_frontfacing_interpolation();
+ emit(MOV(retype(varying, BRW_REGISTER_TYPE_UD), reg));
+ } else {
+ nir_emit_interpolation(var, &varying);
+ }
+ }
+}
+
+void
+fs_visitor::nir_emit_interpolation(nir_variable *var, fs_reg *varying)
+{
+ brw_wm_prog_data *prog_data = (brw_wm_prog_data*) this->prog_data;
+ brw_wm_prog_key *key = (brw_wm_prog_key*) this->key;
+ fs_reg reg = *varying;
+ reg.type = brw_type_for_base_type(var->type->get_scalar_type());
+
+ unsigned int array_elements;
+ const glsl_type *type;
+
+ if (var->type->is_array()) {
+ array_elements = var->type->length;
+ if (array_elements == 0) {
+ fail("dereferenced array '%s' has length 0\n", var->name);
+ }
+ type = var->type->fields.array;
+ } else {
+ array_elements = 1;
+ type = var->type;
+ }
+
+ glsl_interp_qualifier interpolation_mode =
+ determine_interpolation_mode(var, key->flat_shade);
+
+ int location = var->data.location;
+ for (unsigned int i = 0; i < array_elements; i++) {
+ for (unsigned int j = 0; j < type->matrix_columns; j++) {
+ if (prog_data->urb_setup[location] == -1) {
+ /* If there's no incoming setup data for this slot, don't
+ * emit interpolation for it.
+ */
+ reg.reg_offset += type->vector_elements;
+ location++;
+ continue;
+ }
+
+ if (interpolation_mode == INTERP_QUALIFIER_FLAT) {
+ /* Constant interpolation (flat shading) case. The SF has
+ * handed us defined values in only the constant offset
+ * field of the setup reg.
+ */
+ for (unsigned int k = 0; k < type->vector_elements; k++) {
+ struct brw_reg interp = interp_reg(location, k);
+ interp = suboffset(interp, 3);
+ interp.type = reg.type;
+ emit(FS_OPCODE_CINTERP, reg, fs_reg(interp));
+ reg.reg_offset++;
+ }
+ } else {
+ /* Smooth/noperspective interpolation case. */
+ for (unsigned int k = 0; k < type->vector_elements; k++) {
+ struct brw_reg interp = interp_reg(location, k);
+ if (brw->needs_unlit_centroid_workaround && var->data.centroid) {
+ /* Get the pixel/sample mask into f0 so that we know
+ * which pixels are lit. Then, for each channel that is
+ * unlit, replace the centroid data with non-centroid
+ * data.
+ */
+ emit(FS_OPCODE_MOV_DISPATCH_TO_FLAGS);
+
+ fs_inst *inst;
+ inst = emit_linterp(reg, fs_reg(interp), interpolation_mode,
+ false, false);
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ inst->predicate_inverse = true;
+ if (brw->has_pln)
+ inst->no_dd_clear = true;
+
+ inst = emit_linterp(reg, fs_reg(interp), interpolation_mode,
+ var->data.centroid && !key->persample_shading,
+ var->data.sample || key->persample_shading);
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ inst->predicate_inverse = false;
+ if (brw->has_pln)
+ inst->no_dd_check = true;
+
+ } else {
+ emit_linterp(reg, fs_reg(interp), interpolation_mode,
+ var->data.centroid && !key->persample_shading,
+ var->data.sample || key->persample_shading);
+ }
+ if (brw->gen < 6 && interpolation_mode == INTERP_QUALIFIER_SMOOTH) {
+ emit(BRW_OPCODE_MUL, reg, reg, this->pixel_w);
+ }
+ reg.reg_offset++;
+ }
+
+ }
+ location++;
+ }
+ }
+}
+
+void
+fs_visitor::nir_setup_outputs(nir_shader *shader)
+{
+ brw_wm_prog_key *key = (brw_wm_prog_key*) this->key;
+ fs_reg reg = nir_outputs;
+
+ struct hash_entry *entry;
+ hash_table_foreach(shader->outputs, entry) {
+ nir_variable *var = (nir_variable *) entry->data;
+ reg.reg_offset = var->data.driver_location;
+
+ if (var->data.index > 0) {
+ assert(var->data.location == FRAG_RESULT_DATA0);
+ assert(var->data.index == 1);
+ this->dual_src_output = reg;
+ this->do_dual_src = true;
+ } else if (var->data.location == FRAG_RESULT_COLOR) {
+ /* Writing gl_FragColor outputs to all color regions. */
+ for (unsigned int i = 0; i < MAX2(key->nr_color_regions, 1); i++) {
+ this->outputs[i] = reg;
+ this->output_components[i] = 4;
+ }
+ } else if (var->data.location == FRAG_RESULT_DEPTH) {
+ this->frag_depth = reg;
+ } else if (var->data.location == FRAG_RESULT_SAMPLE_MASK) {
+ this->sample_mask = reg;
+ } else {
+ /* gl_FragData or a user-defined FS output */
+ assert(var->data.location >= FRAG_RESULT_DATA0 &&
+ var->data.location < FRAG_RESULT_DATA0 + BRW_MAX_DRAW_BUFFERS);
+
+ int vector_elements =
+ var->type->is_array() ? var->type->fields.array->vector_elements
+ : var->type->vector_elements;
+
+ /* General color output. */
+ for (unsigned int i = 0; i < MAX2(1, var->type->length); i++) {
+ int output = var->data.location - FRAG_RESULT_DATA0 + i;
+ this->outputs[output] = reg;
+ this->outputs[output].reg_offset += vector_elements * i;
+ this->output_components[output] = vector_elements;
+ }
+ }
+ }
+}
+
+void
+fs_visitor::nir_setup_uniforms(nir_shader *shader)
+{
+ uniforms = shader->num_uniforms;
+ param_size[0] = shader->num_uniforms;
+
+ if (dispatch_width != 8)
+ return;
+
+ struct hash_entry *entry;
+ hash_table_foreach(shader->uniforms, entry) {
+ nir_variable *var = (nir_variable *) entry->data;
+
+ /* UBO's and atomics don't take up space in the uniform file */
+
+ if (var->interface_type != NULL || var->type->contains_atomic())
+ continue;
+
+ if (strncmp(var->name, "gl_", 3) == 0)
+ nir_setup_builtin_uniform(var);
+ else
+ nir_setup_uniform(var);
+ }
+}
+
+void
+fs_visitor::nir_setup_uniform(nir_variable *var)
+{
+ int namelen = strlen(var->name);
+
+ /* The data for our (non-builtin) uniforms is stored in a series of
+ * gl_uniform_driver_storage structs for each subcomponent that
+ * glGetUniformLocation() could name. We know it's been set up in the
+ * same order we'd walk the type, so walk the list of storage and find
+ * anything with our name, or the prefix of a component that starts with
+ * our name.
+ */
+ unsigned index = var->data.driver_location;
+ for (unsigned u = 0; u < shader_prog->NumUserUniformStorage; u++) {
+ struct gl_uniform_storage *storage = &shader_prog->UniformStorage[u];
+
+ if (strncmp(var->name, storage->name, namelen) != 0 ||
+ (storage->name[namelen] != 0 &&
+ storage->name[namelen] != '.' &&
+ storage->name[namelen] != '[')) {
+ continue;
+ }
+
+ unsigned slots = storage->type->component_slots();
+ if (storage->array_elements)
+ slots *= storage->array_elements;
+
+ for (unsigned i = 0; i < slots; i++) {
+ stage_prog_data->param[index++] = &storage->storage[i];
+ }
+ }
+
+ /* Make sure we actually initialized the right amount of stuff here. */
+ assert(var->data.driver_location + var->type->component_slots() == index);
+}
+
+void
+fs_visitor::nir_setup_builtin_uniform(nir_variable *var)
+{
+ const nir_state_slot *const slots = var->state_slots;
+ assert(var->state_slots != NULL);
+
+ unsigned uniform_index = var->data.driver_location;
+ for (unsigned int i = 0; i < var->num_state_slots; i++) {
+ /* This state reference has already been setup by ir_to_mesa, but we'll
+ * get the same index back here.
+ */
+ int index = _mesa_add_state_reference(this->prog->Parameters,
+ (gl_state_index *)slots[i].tokens);
+
+ /* Add each of the unique swizzles of the element as a parameter.
+ * This'll end up matching the expected layout of the
+ * array/matrix/structure we're trying to fill in.
+ */
+ int last_swiz = -1;
+ for (unsigned int j = 0; j < 4; j++) {
+ int swiz = GET_SWZ(slots[i].swizzle, j);
+ if (swiz == last_swiz)
+ break;
+ last_swiz = swiz;
+
+ stage_prog_data->param[uniform_index++] =
+ &prog->Parameters->ParameterValues[index][swiz];
+ }
+ }
+}
+
+void
+fs_visitor::nir_setup_registers(exec_list *list)
+{
+ foreach_list_typed(nir_register, nir_reg, node, list) {
+ unsigned array_elems =
+ nir_reg->num_array_elems == 0 ? 1 : nir_reg->num_array_elems;
+ unsigned size = array_elems * nir_reg->num_components;
+ fs_reg *reg = new(mem_ctx) fs_reg(GRF, virtual_grf_alloc(size));
+ _mesa_hash_table_insert(this->nir_reg_ht, _mesa_hash_pointer(nir_reg),
+ nir_reg, reg);
+ }
+}
+
+void
+fs_visitor::nir_emit_impl(nir_function_impl *impl)
+{
+ nir_setup_registers(&impl->registers);
+ nir_emit_cf_list(&impl->body);
+}
+
+void
+fs_visitor::nir_emit_cf_list(exec_list *list)
+{
+ foreach_list_typed(nir_cf_node, node, node, list) {
+ switch (node->type) {
+ case nir_cf_node_if:
+ nir_emit_if(nir_cf_node_as_if(node));
+ break;
+
+ case nir_cf_node_loop:
+ nir_emit_loop(nir_cf_node_as_loop(node));
+ break;
+
+ case nir_cf_node_block:
+ nir_emit_block(nir_cf_node_as_block(node));
+ break;
+
+ default:
+ unreachable("Invalid CFG node block");
+ }
+ }
+}
+
+void
+fs_visitor::nir_emit_if(nir_if *if_stmt)
+{
+ if (brw->gen < 6) {
+ no16("Can't support (non-uniform) control flow on SIMD16\n");
+ }
+
+ /* first, put the condition into f0 */
+ fs_inst *inst = emit(MOV(reg_null_d,
+ retype(get_nir_src(if_stmt->condition),
+ BRW_REGISTER_TYPE_UD)));
+ inst->conditional_mod = BRW_CONDITIONAL_NZ;
+
+ emit(IF(BRW_PREDICATE_NORMAL));
+
+ nir_emit_cf_list(&if_stmt->then_list);
+
+ /* note: if the else is empty, dead CF elimination will remove it */
+ emit(BRW_OPCODE_ELSE);
+
+ nir_emit_cf_list(&if_stmt->else_list);
+
+ emit(BRW_OPCODE_ENDIF);
+
+ try_replace_with_sel();
+}
+
+void
+fs_visitor::nir_emit_loop(nir_loop *loop)
+{
+ if (brw->gen < 6) {
+ no16("Can't support (non-uniform) control flow on SIMD16\n");
+ }
+
+ emit(BRW_OPCODE_DO);
+
+ nir_emit_cf_list(&loop->body);
+
+ emit(BRW_OPCODE_WHILE);
+}
+
+void
+fs_visitor::nir_emit_block(nir_block *block)
+{
+ nir_foreach_instr(block, instr) {
+ nir_emit_instr(instr);
+ }
+}
+
+void
+fs_visitor::nir_emit_instr(nir_instr *instr)
+{
+ switch (instr->type) {
+ case nir_instr_type_alu:
+ nir_emit_alu(nir_instr_as_alu(instr));
+ break;
+
+ case nir_instr_type_intrinsic:
+ nir_emit_intrinsic(nir_instr_as_intrinsic(instr));
+ break;
+
+ case nir_instr_type_texture:
+ nir_emit_texture(nir_instr_as_texture(instr));
+ break;
+
+ case nir_instr_type_load_const:
+ nir_emit_load_const(nir_instr_as_load_const(instr));
+ break;
+
+ case nir_instr_type_jump:
+ nir_emit_jump(nir_instr_as_jump(instr));
+ break;
+
+ default:
+ unreachable("unknown instruction type");
+ }
+}
+
+static brw_reg_type
+brw_type_for_nir_type(nir_alu_type type)
+{
+ switch (type) {
+ case nir_type_bool:
+ case nir_type_unsigned:
+ return BRW_REGISTER_TYPE_UD;
+ case nir_type_int:
+ return BRW_REGISTER_TYPE_D;
+ case nir_type_float:
+ return BRW_REGISTER_TYPE_F;
+ default:
+ unreachable("unknown type");
+ }
+
+ return BRW_REGISTER_TYPE_F;
+}
+
+void
+fs_visitor::nir_emit_alu(nir_alu_instr *instr)
+{
+ struct brw_wm_prog_key *fs_key = (struct brw_wm_prog_key *) this->key;
+
+ fs_reg op[3];
+ fs_reg dest = retype(get_nir_dest(instr->dest.dest),
+ brw_type_for_nir_type(nir_op_infos[instr->op].output_type));
+
+ fs_reg result;
+ if (instr->has_predicate) {
+ result = fs_reg(GRF, virtual_grf_alloc(4));
+ result.type = dest.type;
+ } else {
+ result = dest;
+ }
+
+
+ for (unsigned i = 0; i < nir_op_infos[instr->op].num_inputs; i++) {
+ op[i] = retype(get_nir_alu_src(instr, i),
+ brw_type_for_nir_type(nir_op_infos[instr->op].input_types[i]));
+ }
+
+ switch (instr->op) {
+ case nir_op_fmov:
+ case nir_op_i2f:
+ case nir_op_u2f: {
+ fs_inst *inst = MOV(result, op[0]);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(inst, instr->dest.write_mask);
+ }
+ break;
+
+ case nir_op_imov:
+ case nir_op_f2i:
+ case nir_op_f2u:
+ emit_percomp(MOV(result, op[0]), instr->dest.write_mask);
+ break;
+
+ case nir_op_fsign: {
+ /* AND(val, 0x80000000) gives the sign bit.
+ *
+ * Predicated OR ORs 1.0 (0x3f800000) with the sign bit if val is not
+ * zero.
+ */
+ emit_percomp(CMP(reg_null_f, op[0], fs_reg(0.0f), BRW_CONDITIONAL_NZ),
+ instr->dest.write_mask);
+
+ fs_reg result_int = retype(result, BRW_REGISTER_TYPE_UD);
+ op[0].type = BRW_REGISTER_TYPE_UD;
+ result.type = BRW_REGISTER_TYPE_UD;
+ emit_percomp(AND(result_int, op[0], fs_reg(0x80000000u)),
+ instr->dest.write_mask);
+
+ fs_inst *inst = OR(result_int, result_int, fs_reg(0x3f800000u));
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit_percomp(inst, instr->dest.write_mask);
+ if (instr->dest.saturate) {
+ fs_inst *inst = MOV(result, result);
+ inst->saturate = true;
+ emit_percomp(inst, instr->dest.write_mask);
+ }
+ break;
+ }
+
+ case nir_op_isign: {
+ /* ASR(val, 31) -> negative val generates 0xffffffff (signed -1).
+ * -> non-negative val generates 0x00000000.
+ * Predicated OR sets 1 if val is positive.
+ */
+ emit_percomp(CMP(reg_null_d, op[0], fs_reg(0), BRW_CONDITIONAL_G),
+ instr->dest.write_mask);
+
+ emit_percomp(ASR(result, op[0], fs_reg(31)), instr->dest.write_mask);
+
+ fs_inst *inst = OR(result, result, fs_reg(1));
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_frcp:
+ emit_math_percomp(SHADER_OPCODE_RCP, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_fexp2:
+ emit_math_percomp(SHADER_OPCODE_EXP2, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_flog2:
+ emit_math_percomp(SHADER_OPCODE_LOG2, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_fexp:
+ case nir_op_flog:
+ unreachable("not reached: should be handled by ir_explog_to_explog2");
+
+ case nir_op_fsin:
+ case nir_op_fsin_reduced:
+ emit_math_percomp(SHADER_OPCODE_SIN, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_fcos:
+ case nir_op_fcos_reduced:
+ emit_math_percomp(SHADER_OPCODE_COS, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_fddx:
+ if (fs_key->high_quality_derivatives)
+ emit_percomp(FS_OPCODE_DDX_FINE, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ else
+ emit_percomp(FS_OPCODE_DDX_COARSE, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+ case nir_op_fddy:
+ if (fs_key->high_quality_derivatives)
+ emit_percomp(FS_OPCODE_DDY_FINE, result, op[0],
+ fs_reg(fs_key->render_to_fbo),
+ instr->dest.write_mask, instr->dest.saturate);
+ else
+ emit_percomp(FS_OPCODE_DDY_COARSE, result, op[0],
+ fs_reg(fs_key->render_to_fbo),
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_fadd:
+ case nir_op_iadd: {
+ fs_inst *inst = ADD(result, op[0], op[1]);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_fmul: {
+ fs_inst *inst = MUL(result, op[0], op[1]);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(MUL(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_imul: {
+ /* TODO put in the 16-bit constant optimization once we have SSA */
+
+ if (brw->gen >= 7)
+ no16("SIMD16 explicit accumulator operands unsupported\n");
+
+ struct brw_reg acc = retype(brw_acc_reg(dispatch_width), result.type);
+
+ emit_percomp(MUL(acc, op[0], op[1]), instr->dest.write_mask);
+ emit_percomp(MACH(reg_null_d, op[0], op[1]), instr->dest.write_mask);
+ emit_percomp(MOV(result, fs_reg(acc)), instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_imul_high:
+ case nir_op_umul_high: {
+ if (brw->gen >= 7)
+ no16("SIMD16 explicit accumulator operands unsupported\n");
+
+ struct brw_reg acc = retype(brw_acc_reg(dispatch_width), result.type);
+
+ emit_percomp(MUL(acc, op[0], op[1]), instr->dest.write_mask);
+ emit_percomp(MACH(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_idiv:
+ case nir_op_udiv:
+ emit_math_percomp(SHADER_OPCODE_INT_QUOTIENT, result, op[0], op[1],
+ instr->dest.write_mask);
+ break;
+
+ case nir_op_uadd_carry: {
+ if (brw->gen >= 7)
+ no16("SIMD16 explicit accumulator operands unsupported\n");
+
+ struct brw_reg acc = retype(brw_acc_reg(dispatch_width),
+ BRW_REGISTER_TYPE_UD);
+
+ emit_percomp(ADDC(reg_null_ud, op[0], op[1]), instr->dest.write_mask);
+ emit_percomp(MOV(result, fs_reg(acc)), instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_usub_borrow: {
+ if (brw->gen >= 7)
+ no16("SIMD16 explicit accumulator operands unsupported\n");
+
+ struct brw_reg acc = retype(brw_acc_reg(dispatch_width),
+ BRW_REGISTER_TYPE_UD);
+
+ emit_percomp(SUBB(reg_null_ud, op[0], op[1]), instr->dest.write_mask);
+ emit_percomp(MOV(result, fs_reg(acc)), instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_umod:
+ emit_math_percomp(SHADER_OPCODE_INT_REMAINDER, result, op[0],
+ op[1], instr->dest.write_mask);
+ break;
+
+ case nir_op_flt:
+ case nir_op_ilt:
+ case nir_op_ult:
+ emit_percomp(CMP(result, op[0], op[1], BRW_CONDITIONAL_L),
+ instr->dest.write_mask);
+ break;
+
+ case nir_op_fge:
+ case nir_op_ige:
+ case nir_op_uge:
+ emit_percomp(CMP(result, op[0], op[1], BRW_CONDITIONAL_GE),
+ instr->dest.write_mask);
+ break;
+
+ case nir_op_feq:
+ case nir_op_ieq:
+ emit_percomp(CMP(result, op[0], op[1], BRW_CONDITIONAL_Z),
+ instr->dest.write_mask);
+ break;
+
+ case nir_op_fne:
+ case nir_op_ine:
+ emit_percomp(CMP(result, op[0], op[1], BRW_CONDITIONAL_NZ),
+ instr->dest.write_mask);
+ break;
+
+ case nir_op_ball_fequal2:
+ case nir_op_ball_iequal2:
+ case nir_op_ball_fequal3:
+ case nir_op_ball_iequal3:
+ case nir_op_ball_fequal4:
+ case nir_op_ball_iequal4: {
+ unsigned num_components = nir_op_infos[instr->op].input_sizes[0];
+ fs_reg temp = fs_reg(GRF, virtual_grf_alloc(num_components));
+ emit_percomp(CMP(temp, op[0], op[1], BRW_CONDITIONAL_Z),
+ (1 << num_components) - 1);
+ emit_reduction(BRW_OPCODE_AND, result, temp, num_components);
+ break;
+ }
+
+ case nir_op_bany_fnequal2:
+ case nir_op_bany_inequal2:
+ case nir_op_bany_fnequal3:
+ case nir_op_bany_inequal3:
+ case nir_op_bany_fnequal4:
+ case nir_op_bany_inequal4: {
+ unsigned num_components = nir_op_infos[instr->op].input_sizes[0];
+ fs_reg temp = fs_reg(GRF, virtual_grf_alloc(num_components));
+ temp.type = BRW_REGISTER_TYPE_UD;
+ emit_percomp(CMP(temp, op[0], op[1], BRW_CONDITIONAL_NZ),
+ (1 << num_components) - 1);
+ emit_reduction(BRW_OPCODE_OR, result, temp, num_components);
+ break;
+ }
+
+ case nir_op_inot:
+ emit_percomp(NOT(result, op[0]), instr->dest.write_mask);
+ break;
+ case nir_op_ixor:
+ emit_percomp(XOR(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+ case nir_op_ior:
+ emit_percomp(OR(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+ case nir_op_iand:
+ emit_percomp(AND(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+
+ case nir_op_fdot2:
+ case nir_op_fdot3:
+ case nir_op_fdot4: {
+ unsigned num_components = nir_op_infos[instr->op].input_sizes[0];
+ fs_reg temp = fs_reg(GRF, virtual_grf_alloc(num_components));
+ emit_percomp(MUL(temp, op[0], op[1]), (1 << num_components) - 1);
+ emit_reduction(BRW_OPCODE_ADD, result, temp, num_components);
+ if (instr->dest.saturate) {
+ fs_inst *inst = emit(MOV(result, result));
+ inst->saturate = true;
+ }
+ break;
+ }
+
+ case nir_op_bany2:
+ case nir_op_bany3:
+ case nir_op_bany4: {
+ unsigned num_components = nir_op_infos[instr->op].input_sizes[0];
+ emit_reduction(BRW_OPCODE_OR, result, op[0], num_components);
+ break;
+ }
+
+ case nir_op_ball2:
+ case nir_op_ball3:
+ case nir_op_ball4: {
+ unsigned num_components = nir_op_infos[instr->op].input_sizes[0];
+ emit_reduction(BRW_OPCODE_AND, result, op[0], num_components);
+ break;
+ }
+
+ case nir_op_fnoise1_1:
+ case nir_op_fnoise1_2:
+ case nir_op_fnoise1_3:
+ case nir_op_fnoise1_4:
+ case nir_op_fnoise2_1:
+ case nir_op_fnoise2_2:
+ case nir_op_fnoise2_3:
+ case nir_op_fnoise2_4:
+ case nir_op_fnoise3_1:
+ case nir_op_fnoise3_2:
+ case nir_op_fnoise3_3:
+ case nir_op_fnoise3_4:
+ case nir_op_fnoise4_1:
+ case nir_op_fnoise4_2:
+ case nir_op_fnoise4_3:
+ case nir_op_fnoise4_4:
+ unreachable("not reached: should be handled by lower_noise");
+
+ case nir_op_vec2:
+ case nir_op_vec3:
+ case nir_op_vec4:
+ unreachable("not reached: should be handled by lower_quadop_vector");
+
+ case nir_op_ldexp:
+ unreachable("not reached: should be handled by ldexp_to_arith()");
+
+ case nir_op_fsqrt:
+ emit_math_percomp(SHADER_OPCODE_SQRT, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_frsq:
+ emit_math_percomp(SHADER_OPCODE_RSQ, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_b2i:
+ emit_percomp(AND(result, op[0], fs_reg(1)), instr->dest.write_mask);
+ break;
+ case nir_op_b2f: {
+ emit_percomp(AND(retype(result, BRW_REGISTER_TYPE_UD), op[0],
+ fs_reg(0x3f800000u)),
+ instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_f2b:
+ emit_percomp(CMP(result, op[0], fs_reg(0.0f), BRW_CONDITIONAL_NZ),
+ instr->dest.write_mask);
+ break;
+ case nir_op_i2b:
+ emit_percomp(CMP(result, op[0], fs_reg(0), BRW_CONDITIONAL_NZ),
+ instr->dest.write_mask);
+ break;
+
+ case nir_op_ftrunc: {
+ fs_inst *inst = RNDZ(result, op[0]);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+ case nir_op_fceil: {
+ op[0].negate = !op[0].negate;
+ fs_reg temp = fs_reg(this, glsl_type::vec4_type);
+ emit_percomp(RNDD(temp, op[0]), instr->dest.write_mask);
+ temp.negate = true;
+ fs_inst *inst = MOV(result, temp);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+ case nir_op_ffloor: {
+ fs_inst *inst = RNDD(result, op[0]);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+ case nir_op_ffract: {
+ fs_inst *inst = FRC(result, op[0]);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+ case nir_op_fround_even: {
+ fs_inst *inst = RNDE(result, op[0]);
+ inst->saturate = instr->dest.saturate;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_fmin:
+ case nir_op_imin:
+ case nir_op_umin:
+ if (brw->gen >= 6) {
+ emit_percomp(BRW_OPCODE_SEL, result, op[0], op[1],
+ instr->dest.write_mask, instr->dest.saturate,
+ BRW_PREDICATE_NONE, BRW_CONDITIONAL_L);
+ } else {
+ emit_percomp(CMP(reg_null_d, op[0], op[1], BRW_CONDITIONAL_L),
+ instr->dest.write_mask);
+
+ emit_percomp(BRW_OPCODE_SEL, result, op[0], op[1],
+ instr->dest.write_mask, instr->dest.saturate,
+ BRW_PREDICATE_NORMAL);
+ }
+ break;
+
+ case nir_op_fmax:
+ case nir_op_imax:
+ case nir_op_umax:
+ if (brw->gen >= 6) {
+ emit_percomp(BRW_OPCODE_SEL, result, op[0], op[1],
+ instr->dest.write_mask, instr->dest.saturate,
+ BRW_PREDICATE_NONE, BRW_CONDITIONAL_GE);
+ } else {
+ emit_percomp(CMP(reg_null_d, op[0], op[1], BRW_CONDITIONAL_GE),
+ instr->dest.write_mask);
+
+ emit_percomp(BRW_OPCODE_SEL, result, op[0], op[1],
+ instr->dest.write_mask, instr->dest.saturate,
+ BRW_PREDICATE_NORMAL);
+ }
+ break;
+
+ case nir_op_pack_snorm_2x16:
+ case nir_op_pack_snorm_4x8:
+ case nir_op_pack_unorm_2x16:
+ case nir_op_pack_unorm_4x8:
+ case nir_op_unpack_snorm_2x16:
+ case nir_op_unpack_snorm_4x8:
+ case nir_op_unpack_unorm_2x16:
+ case nir_op_unpack_unorm_4x8:
+ case nir_op_unpack_half_2x16:
+ case nir_op_pack_half_2x16:
+ unreachable("not reached: should be handled by lower_packing_builtins");
+
+ case nir_op_unpack_half_2x16_split_x:
+ emit_percomp(FS_OPCODE_UNPACK_HALF_2x16_SPLIT_X, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+ case nir_op_unpack_half_2x16_split_y:
+ emit_percomp(FS_OPCODE_UNPACK_HALF_2x16_SPLIT_Y, result, op[0],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_fpow:
+ emit_percomp(SHADER_OPCODE_POW, result, op[0], op[1],
+ instr->dest.write_mask, instr->dest.saturate);
+ break;
+
+ case nir_op_bitfield_reverse:
+ emit_percomp(BFREV(result, op[0]), instr->dest.write_mask);
+ break;
+
+ case nir_op_bit_count:
+ emit_percomp(CBIT(result, op[0]), instr->dest.write_mask);
+ break;
+
+ case nir_op_find_msb: {
+ fs_reg temp = fs_reg(this, glsl_type::uvec4_type);
+ emit_percomp(FBH(temp, op[0]), instr->dest.write_mask);
+
+ /* FBH counts from the MSB side, while GLSL's findMSB() wants the count
+ * from the LSB side. If FBH didn't return an error (0xFFFFFFFF), then
+ * subtract the result from 31 to convert the MSB count into an LSB count.
+ */
+
+ emit_percomp(CMP(reg_null_d, temp, fs_reg(~0), BRW_CONDITIONAL_NZ),
+ instr->dest.write_mask);
+ temp.negate = true;
+ fs_inst *inst = ADD(result, temp, fs_reg(31));
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit_percomp(inst, instr->dest.write_mask);
+ break;
+ }
+
+ case nir_op_find_lsb:
+ emit_percomp(FBL(result, op[0]), instr->dest.write_mask);
+ break;
+
+ case nir_op_ubitfield_extract:
+ case nir_op_ibitfield_extract:
+ emit_percomp(BFE(result, op[2], op[1], op[0]), instr->dest.write_mask);
+ break;
+ case nir_op_bfm:
+ emit_percomp(BFI1(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+ case nir_op_bfi:
+ emit_percomp(BFI2(result, op[0], op[1], op[2]), instr->dest.write_mask);
+ break;
+
+ case nir_op_bitfield_insert:
+ unreachable("not reached: should be handled by "
+ "lower_instructions::bitfield_insert_to_bfm_bfi");
+
+ case nir_op_ishl:
+ emit_percomp(SHL(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+ case nir_op_ishr:
+ emit_percomp(ASR(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+ case nir_op_ushr:
+ emit_percomp(SHR(result, op[0], op[1]), instr->dest.write_mask);
+ break;
+
+ case nir_op_pack_half_2x16_split:
+ emit_percomp(FS_OPCODE_PACK_HALF_2x16_SPLIT, result, op[0], op[1],
+ instr->dest.write_mask);
+ break;
+
+ case nir_op_ffma:
+ emit_percomp(MAD(result, op[2], op[1], op[0]), instr->dest.write_mask);
+ break;
+
+ case nir_op_flrp:
+ /* TODO emulate for gen < 6 */
+ emit_percomp(LRP(result, op[2], op[1], op[0]), instr->dest.write_mask);
+ break;
+
+ case nir_op_bcsel:
+ emit(CMP(reg_null_d, op[0], fs_reg(0), BRW_CONDITIONAL_NZ));
+ emit_percomp(BRW_OPCODE_SEL, result, op[1], op[2],
+ instr->dest.write_mask, false, BRW_PREDICATE_NORMAL);
+ break;
+
+ default:
+ unreachable("unhandled instruction");
+ }
+
+ /* emit a predicated move if there was predication */
+ if (instr->has_predicate) {
+ fs_inst *inst = emit(MOV(reg_null_d,
+ retype(get_nir_src(instr->predicate),
+ BRW_REGISTER_TYPE_UD)));
+ inst->conditional_mod = BRW_CONDITIONAL_NZ;
+ inst = MOV(dest, result);
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit_percomp(inst, instr->dest.write_mask);
+ }
+}
+
+fs_reg
+fs_visitor::get_nir_src(nir_src src)
+{
+ struct hash_entry *entry =
+ _mesa_hash_table_search(this->nir_reg_ht, _mesa_hash_pointer(src.reg.reg),
+ src.reg.reg);
+ fs_reg reg = *((fs_reg *) entry->data);
+ /* to avoid floating-point denorm flushing problems, set the type by
+ * default to D - instructions that need floating point semantics will set
+ * this to F if they need to
+ */
+ reg.type = BRW_REGISTER_TYPE_D;
+ reg.reg_offset = src.reg.base_offset;
+ if (src.reg.indirect) {
+ reg.reladdr = new(mem_ctx) fs_reg();
+ *reg.reladdr = retype(get_nir_src(*src.reg.indirect),
+ BRW_REGISTER_TYPE_D);
+ }
+
+ return reg;
+}
+
+fs_reg
+fs_visitor::get_nir_alu_src(nir_alu_instr *instr, unsigned src)
+{
+ fs_reg reg = get_nir_src(instr->src[src].src);
+
+ reg.abs = instr->src[src].abs;
+ reg.negate = instr->src[src].negate;
+
+ bool needs_swizzle = false;
+ unsigned num_components = 0;
+ for (unsigned i = 0; i < 4; i++) {
+ if (!nir_alu_instr_channel_used(instr, src, i))
+ continue;
+
+ if (instr->src[src].swizzle[i] != i)
+ needs_swizzle = true;
+
+ num_components = i + 1;
+ }
+
+ if (needs_swizzle) {
+ /* resolve the swizzle through MOV's */
+ fs_reg new_reg = fs_reg(GRF, virtual_grf_alloc(num_components));
+
+ for (unsigned i = 0; i < 4; i++) {
+ if (!nir_alu_instr_channel_used(instr, src, i))
+ continue;
+
+ fs_reg dest = new_reg;
+ dest.type = reg.type;
+ dest.reg_offset = i;
+
+ fs_reg src0 = reg;
+ src0.reg_offset += instr->src[src].swizzle[i];
+
+ emit(MOV(dest, src0));
+ }
+
+ return new_reg;
+ }
+
+ return reg;
+}
+
+fs_reg
+fs_visitor::get_nir_dest(nir_dest dest)
+{
+ struct hash_entry *entry =
+ _mesa_hash_table_search(this->nir_reg_ht,
+ _mesa_hash_pointer(dest.reg.reg), dest.reg.reg);
+ fs_reg reg = *((fs_reg *) entry->data);
+ reg.reg_offset = dest.reg.base_offset;
+ if (dest.reg.indirect) {
+ reg.reladdr = new(mem_ctx) fs_reg();
+ *reg.reladdr = retype(get_nir_src(*dest.reg.indirect),
+ BRW_REGISTER_TYPE_D);
+ }
+
+ return reg;
+}
+
+void
+fs_visitor::emit_percomp(fs_inst *inst, unsigned wr_mask)
+{
+ for (unsigned i = 0; i < 4; i++) {
+ if (!((wr_mask >> i) & 1))
+ continue;
+
+ fs_inst *new_inst = new(mem_ctx) fs_inst(*inst);
+ new_inst->dst.reg_offset += i;
+ for (unsigned j = 0; j < new_inst->sources; j++)
+ if (inst->src[j].file == GRF)
+ new_inst->src[j].reg_offset += i;
+
+ emit(new_inst);
+ }
+}
+
+void
+fs_visitor::emit_percomp(enum opcode op, fs_reg dest, fs_reg src0,
+ unsigned wr_mask, bool saturate,
+ enum brw_predicate predicate,
+ enum brw_conditional_mod mod)
+{
+ for (unsigned i = 0; i < 4; i++) {
+ if (!((wr_mask >> i) & 1))
+ continue;
+
+ fs_inst *new_inst = new(mem_ctx) fs_inst(op, dest, src0);
+ new_inst->dst.reg_offset += i;
+ for (unsigned j = 0; j < new_inst->sources; j++)
+ if (new_inst->src[j].file == GRF)
+ new_inst->src[j].reg_offset += i;
+
+ new_inst->predicate = predicate;
+ new_inst->conditional_mod = mod;
+ new_inst->saturate = saturate;
+ emit(new_inst);
+ }
+}
+
+void
+fs_visitor::emit_percomp(enum opcode op, fs_reg dest, fs_reg src0, fs_reg src1,
+ unsigned wr_mask, bool saturate,
+ enum brw_predicate predicate,
+ enum brw_conditional_mod mod)
+{
+ for (unsigned i = 0; i < 4; i++) {
+ if (!((wr_mask >> i) & 1))
+ continue;
+
+ fs_inst *new_inst = new(mem_ctx) fs_inst(op, dest, src0, src1);
+ new_inst->dst.reg_offset += i;
+ for (unsigned j = 0; j < new_inst->sources; j++)
+ if (new_inst->src[j].file == GRF)
+ new_inst->src[j].reg_offset += i;
+
+ new_inst->predicate = predicate;
+ new_inst->conditional_mod = mod;
+ new_inst->saturate = saturate;
+ emit(new_inst);
+ }
+}
+
+void
+fs_visitor::emit_math_percomp(enum opcode op, fs_reg dest, fs_reg src0,
+ unsigned wr_mask, bool saturate)
+{
+ for (unsigned i = 0; i < 4; i++) {
+ if (!((wr_mask >> i) & 1))
+ continue;
+
+ fs_reg new_dest = dest;
+ new_dest.reg_offset += i;
+ fs_reg new_src0 = src0;
+ if (src0.file == GRF)
+ new_src0.reg_offset += i;
+
+ fs_inst *new_inst = emit_math(op, new_dest, new_src0);
+ new_inst->saturate = saturate;
+ }
+}
+
+void
+fs_visitor::emit_math_percomp(enum opcode op, fs_reg dest, fs_reg src0,
+ fs_reg src1, unsigned wr_mask,
+ bool saturate)
+{
+ for (unsigned i = 0; i < 4; i++) {
+ if (!((wr_mask >> i) & 1))
+ continue;
+
+ fs_reg new_dest = dest;
+ new_dest.reg_offset += i;
+ fs_reg new_src0 = src0;
+ if (src0.file == GRF)
+ new_src0.reg_offset += i;
+ fs_reg new_src1 = src1;
+ if (src1.file == GRF)
+ new_src1.reg_offset += i;
+
+ fs_inst *new_inst = emit_math(op, new_dest, new_src0, new_src1);
+ new_inst->saturate = saturate;
+ }
+}
+
+void
+fs_visitor::emit_reduction(enum opcode op, fs_reg dest, fs_reg src,
+ unsigned num_components)
+{
+ fs_reg src0 = src;
+ fs_reg src1 = src;
+ src1.reg_offset++;
+
+ if (num_components == 2) {
+ emit(op, dest, src0, src1);
+ return;
+ }
+
+ fs_reg temp1 = fs_reg(GRF, virtual_grf_alloc(1));
+ temp1.type = src.type;
+ emit(op, temp1, src0, src1);
+
+ fs_reg src2 = src;
+ src2.reg_offset += 2;
+
+ if (num_components == 3) {
+ emit(op, dest, temp1, src2);
+ return;
+ }
+
+ assert(num_components == 4);
+
+ fs_reg src3 = src;
+ src3.reg_offset += 3;
+ fs_reg temp2 = fs_reg(GRF, virtual_grf_alloc(1));
+ temp2.type = src.type;
+
+ emit(op, temp2, src2, src3);
+ emit(op, dest, temp1, temp2);
+}
+
+void
+fs_visitor::nir_emit_intrinsic(nir_intrinsic_instr *instr)
+{
+ fs_reg dest;
+ if (nir_intrinsic_infos[instr->intrinsic].has_dest)
+ dest = get_nir_dest(instr->dest);
+ if (instr->has_predicate) {
+ fs_inst *inst = emit(MOV(reg_null_d,
+ retype(get_nir_src(instr->predicate),
+ BRW_REGISTER_TYPE_UD)));
+ inst->conditional_mod = BRW_CONDITIONAL_NZ;
+ }
+
+ switch (instr->intrinsic) {
+ case nir_intrinsic_discard: {
+ /* We track our discarded pixels in f0.1. By predicating on it, we can
+ * update just the flag bits that aren't yet discarded. By emitting a
+ * CMP of g0 != g0, all our currently executing channels will get turned
+ * off.
+ */
+ fs_reg some_reg = fs_reg(retype(brw_vec8_grf(0, 0),
+ BRW_REGISTER_TYPE_UW));
+ fs_inst *cmp = emit(CMP(reg_null_f, some_reg, some_reg,
+ BRW_CONDITIONAL_NZ));
+ cmp->predicate = BRW_PREDICATE_NORMAL;
+ cmp->flag_subreg = 1;
+
+ if (brw->gen >= 6) {
+ /* For performance, after a discard, jump to the end of the shader.
+ * Only jump if all relevant channels have been discarded.
+ */
+ fs_inst *discard_jump = emit(FS_OPCODE_DISCARD_JUMP);
+ discard_jump->flag_subreg = 1;
+
+ discard_jump->predicate = (dispatch_width == 8)
+ ? BRW_PREDICATE_ALIGN1_ANY8H
+ : BRW_PREDICATE_ALIGN1_ANY16H;
+ discard_jump->predicate_inverse = true;
+ }
+
+ break;
+ }
+
+ case nir_intrinsic_atomic_counter_inc:
+ case nir_intrinsic_atomic_counter_dec:
+ case nir_intrinsic_atomic_counter_read:
+ assert(!"TODO");
+
+
+ case nir_intrinsic_load_front_face:
+ assert(!"TODO");
+
+ case nir_intrinsic_load_sample_mask_in: {
+ assert(brw->gen >= 7);
+ fs_reg reg = fs_reg(retype(brw_vec8_grf(payload.sample_mask_in_reg, 0),
+ BRW_REGISTER_TYPE_D));
+ dest.type = reg.type;
+ fs_inst *inst = MOV(dest, reg);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(inst);
+ break;
+ }
+
+ case nir_intrinsic_load_sample_pos:
+ case nir_intrinsic_load_sample_id:
+ assert(!"TODO");
+
+ case nir_intrinsic_load_uniform_vec1:
+ case nir_intrinsic_load_uniform_vec2:
+ case nir_intrinsic_load_uniform_vec3:
+ case nir_intrinsic_load_uniform_vec4: {
+ unsigned index = 0;
+ for (int i = 0; i < instr->const_index[1]; i++) {
+ for (unsigned j = 0;
+ j < nir_intrinsic_infos[instr->intrinsic].dest_components; j++) {
+ fs_reg src = nir_uniforms;
+ src.reg_offset = instr->const_index[0] + index;
+ src.type = dest.type;
+ index++;
+
+ fs_inst *inst = MOV(dest, src);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(inst);
+ dest.reg_offset++;
+ }
+ }
+ break;
+ }
+
+ case nir_intrinsic_load_uniform_vec1_indirect:
+ case nir_intrinsic_load_uniform_vec2_indirect:
+ case nir_intrinsic_load_uniform_vec3_indirect:
+ case nir_intrinsic_load_uniform_vec4_indirect: {
+ unsigned index = 0;
+ for (int i = 0; i < instr->const_index[1]; i++) {
+ for (unsigned j = 0;
+ j < nir_intrinsic_infos[instr->intrinsic].dest_components; j++) {
+ fs_reg src = nir_uniforms;
+ src.reg_offset = instr->const_index[0] + index;
+ src.reladdr = new(mem_ctx) fs_reg(get_nir_src(instr->src[0]));
+ src.reladdr->type = BRW_REGISTER_TYPE_D;
+ src.type = dest.type;
+ index++;
+
+ fs_inst *inst = MOV(dest, src);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(inst);
+ dest.reg_offset++;
+ }
+ }
+ break;
+ }
+
+ case nir_intrinsic_load_ubo_vec1:
+ case nir_intrinsic_load_ubo_vec2:
+ case nir_intrinsic_load_ubo_vec3:
+ case nir_intrinsic_load_ubo_vec4: {
+ fs_reg surf_index = fs_reg(prog_data->binding_table.ubo_start +
+ (unsigned) instr->const_index[0]);
+ fs_reg packed_consts = fs_reg(this, glsl_type::float_type);
+ packed_consts.type = dest.type;
+
+ fs_reg const_offset_reg = fs_reg((unsigned) instr->const_index[1] & ~15);
+ emit(new(mem_ctx) fs_inst(FS_OPCODE_UNIFORM_PULL_CONSTANT_LOAD,
+ packed_consts, surf_index, const_offset_reg));
+
+ for (unsigned i = 0;
+ i < nir_intrinsic_infos[instr->intrinsic].dest_components; i++) {
+ packed_consts.set_smear(instr->const_index[1] % 16 / 4 + i);
+
+ /* The std140 packing rules don't allow vectors to cross 16-byte
+ * boundaries, and a reg is 32 bytes.
+ */
+ assert(packed_consts.subreg_offset < 32);
+
+ fs_inst *inst = MOV(dest, packed_consts);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(inst);
+
+ dest.reg_offset++;
+ }
+ break;
+ }
+
+ case nir_intrinsic_load_ubo_vec1_indirect:
+ case nir_intrinsic_load_ubo_vec2_indirect:
+ case nir_intrinsic_load_ubo_vec3_indirect:
+ case nir_intrinsic_load_ubo_vec4_indirect: {
+ fs_reg surf_index = fs_reg(prog_data->binding_table.ubo_start +
+ instr->const_index[0]);
+ /* Turn the byte offset into a dword offset. */
+ unsigned base_offset = instr->const_index[1] / 4;
+ fs_reg offset = fs_reg(this, glsl_type::int_type);
+ emit(SHR(offset, retype(get_nir_src(instr->src[0]), BRW_REGISTER_TYPE_D),
+ fs_reg(2)));
+
+ for (unsigned i = 0;
+ i < nir_intrinsic_infos[instr->intrinsic].dest_components; i++) {
+ exec_list list = VARYING_PULL_CONSTANT_LOAD(dest, surf_index,
+ dest, base_offset + i);
+ fs_inst *last_inst = (fs_inst *) list.get_tail();
+ if (instr->has_predicate)
+ last_inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(list);
+
+ dest.reg_offset++;
+ }
+ break;
+ }
+
+ case nir_intrinsic_load_input_vec1:
+ case nir_intrinsic_load_input_vec2:
+ case nir_intrinsic_load_input_vec3:
+ case nir_intrinsic_load_input_vec4: {
+ unsigned index = 0;
+ for (int i = 0; i < instr->const_index[1]; i++) {
+ for (unsigned j = 0;
+ j < nir_intrinsic_infos[instr->intrinsic].dest_components; j++) {
+ fs_reg src = nir_inputs;
+ src.reg_offset = instr->const_index[0] + index;
+ src.type = dest.type;
+ index++;
+
+ fs_inst *inst = MOV(dest, src);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(inst);
+ dest.reg_offset++;
+ }
+ }
+ break;
+ }
+
+ case nir_intrinsic_load_input_vec1_indirect:
+ case nir_intrinsic_load_input_vec2_indirect:
+ case nir_intrinsic_load_input_vec3_indirect:
+ case nir_intrinsic_load_input_vec4_indirect: {
+ unsigned index = 0;
+ for (int i = 0; i < instr->const_index[1]; i++) {
+ for (unsigned j = 0;
+ j < nir_intrinsic_infos[instr->intrinsic].dest_components; j++) {
+ fs_reg src = nir_inputs;
+ src.reg_offset = instr->const_index[0] + index;
+ src.reladdr = new(mem_ctx) fs_reg(get_nir_src(instr->src[0]));
+ src.reladdr->type = BRW_REGISTER_TYPE_D;
+ src.type = dest.type;
+ index++;
+
+ fs_inst *inst = MOV(dest, src);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(inst);
+ dest.reg_offset++;
+ }
+ }
+ break;
+ }
+
+ case nir_intrinsic_store_output_vec1:
+ case nir_intrinsic_store_output_vec2:
+ case nir_intrinsic_store_output_vec3:
+ case nir_intrinsic_store_output_vec4: {
+ fs_reg src = get_nir_src(instr->src[0]);
+ unsigned index = 0;
+ for (int i = 0; i < instr->const_index[1]; i++) {
+ for (unsigned j = 0;
+ j < nir_intrinsic_infos[instr->intrinsic].src_components[0]; j++) {
+ fs_reg new_dest = nir_outputs;
+ new_dest.reg_offset = instr->const_index[0] + index;
+ new_dest.type = src.type;
+ index++;
+ fs_inst *inst = MOV(new_dest, src);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(inst);
+ src.reg_offset++;
+ }
+ }
+ break;
+ }
+
+ case nir_intrinsic_store_output_vec1_indirect:
+ case nir_intrinsic_store_output_vec2_indirect:
+ case nir_intrinsic_store_output_vec3_indirect:
+ case nir_intrinsic_store_output_vec4_indirect: {
+ fs_reg src = get_nir_src(instr->src[0]);
+ fs_reg indirect = get_nir_src(instr->src[1]);
+ unsigned index = 0;
+ for (int i = 0; i < instr->const_index[1]; i++) {
+ for (unsigned j = 0;
+ j < nir_intrinsic_infos[instr->intrinsic].src_components[0]; j++) {
+ fs_reg new_dest = nir_outputs;
+ new_dest.reg_offset = instr->const_index[0] + index;
+ new_dest.reladdr = new(mem_ctx) fs_reg(indirect);
+ new_dest.type = src.type;
+ index++;
+ fs_inst *inst = MOV(new_dest, src);
+ if (instr->has_predicate)
+ inst->predicate = BRW_PREDICATE_NORMAL;
+ emit(MOV(new_dest, src));
+ src.reg_offset++;
+ }
+ }
+ break;
+ }
+
+ default:
+ unreachable("unknown intrinsic");
+ }
+}
+
+void
+fs_visitor::nir_emit_texture(nir_tex_instr *instr)
+{
+ brw_wm_prog_key *key = (brw_wm_prog_key*) this->key;
+ int sampler = instr->sampler_index;
+
+ /* FINISHME: We're failing to recompile our programs when the sampler is
+ * updated. This only matters for the texture rectangle scale parameters
+ * (pre-gen6, or gen6+ with GL_CLAMP).
+ */
+ int texunit = prog->SamplerUnits[sampler];
+
+ int gather_component = instr->component;
+
+ bool is_rect = instr->sampler_dim == GLSL_SAMPLER_DIM_RECT;
+
+ bool is_cube_array = instr->sampler_dim == GLSL_SAMPLER_DIM_CUBE &&
+ instr->is_array;
+
+ int lod_components, offset_components = 0;
+
+ fs_reg coordinate, shadow_comparitor, lod, lod2, sample_index, mcs, offset;
+
+ for (unsigned i = 0; i < instr->num_srcs; i++) {
+ fs_reg src = get_nir_src(instr->src[i]);
+ switch (instr->src_type[i]) {
+ case nir_tex_src_bias:
+ lod = src;
+ break;
+ case nir_tex_src_comparitor:
+ shadow_comparitor = src;
+ break;
+ case nir_tex_src_coord:
+ coordinate = src;
+ break;
+ case nir_tex_src_ddx:
+ lod = src;
+ lod_components = nir_tex_instr_src_size(instr, i);
+ break;
+ case nir_tex_src_ddy:
+ lod2 = src;
+ break;
+ case nir_tex_src_lod:
+ lod = src;
+ break;
+ case nir_tex_src_ms_index:
+ sample_index = src;
+ break;
+ case nir_tex_src_offset:
+ offset = src;
+ if (instr->is_array)
+ offset_components = instr->coord_components - 1;
+ else
+ offset_components = instr->coord_components;
+ break;
+ case nir_tex_src_projector:
+ unreachable("should be lowered");
+ case nir_tex_src_sampler_index:
+ unreachable("not yet supported");
+ default:
+ unreachable("unknown texture source");
+ }
+ }
+
+ if (instr->op == nir_texop_txf_ms) {
+ if (brw->gen >= 7 && key->tex.compressed_multisample_layout_mask & (1<<sampler))
+ mcs = emit_mcs_fetch(coordinate, instr->coord_components, fs_reg(sampler));
+ else
+ mcs = fs_reg(0u);
+ }
+
+ for (unsigned i = 0; i < 4; i++) {
+ if (instr->const_offset[i] != 0) {
+ assert(offset_components == 0);
+ offset = fs_reg(instr->const_offset[i]);
+ offset_components = 1;
+ break;
+ }
+ }
+
+ enum glsl_base_type dest_base_type;
+ switch (instr->dest_type) {
+ case nir_type_float:
+ dest_base_type = GLSL_TYPE_FLOAT;
+ break;
+ case nir_type_int:
+ dest_base_type = GLSL_TYPE_INT;
+ break;
+ case nir_type_unsigned:
+ dest_base_type = GLSL_TYPE_UINT;
+ break;
+ default:
+ unreachable("bad type");
+ }
+
+ const glsl_type *dest_type =
+ glsl_type::get_instance(dest_base_type, nir_tex_instr_dest_size(instr),
+ 1);
+
+ ir_texture_opcode op;
+ switch (instr->op) {
+ case nir_texop_lod: op = ir_lod; break;
+ case nir_texop_query_levels: op = ir_query_levels; break;
+ case nir_texop_tex: op = ir_tex; break;
+ case nir_texop_tg4: op = ir_tg4; break;
+ case nir_texop_txb: op = ir_txb; break;
+ case nir_texop_txd: op = ir_txd; break;
+ case nir_texop_txf: op = ir_txf; break;
+ case nir_texop_txf_ms: op = ir_txf_ms; break;
+ case nir_texop_txl: op = ir_txl; break;
+ case nir_texop_txs: op = ir_txs; break;
+ default:
+ unreachable("unknown texture opcode");
+ }
+
+ emit_texture(op, dest_type, coordinate, instr->coord_components,
+ shadow_comparitor, lod, lod2, lod_components, sample_index,
+ offset,offset_components, mcs, gather_component,
+ is_cube_array, is_rect, sampler, fs_reg(sampler), texunit);
+
+ fs_reg dest = get_nir_dest(instr->dest);
+ dest.type = this->result.type;
+ unsigned num_components = nir_tex_instr_dest_size(instr);
+ emit_percomp(MOV(dest, this->result), (1 << num_components) - 1);
+}
+
+void
+fs_visitor::nir_emit_load_const(nir_load_const_instr *instr)
+{
+ fs_reg dest = get_nir_dest(instr->dest);
+ dest.type = BRW_REGISTER_TYPE_UD;
+ if (instr->array_elems == 0) {
+ for (unsigned i = 0; i < instr->num_components; i++) {
+ emit(MOV(dest, fs_reg(instr->value.u[i])));
+ dest.reg_offset++;
+ }
+ } else {
+ for (unsigned i = 0; i < instr->array_elems; i++) {
+ for (unsigned j = 0; j < instr->num_components; j++) {
+ emit(MOV(dest, fs_reg(instr->array[i].u[j])));
+ dest.reg_offset++;
+ }
+ }
+ }
+}
+
+void
+fs_visitor::nir_emit_jump(nir_jump_instr *instr)
+{
+ switch (instr->type) {
+ case nir_jump_break:
+ emit(BRW_OPCODE_BREAK);
+ break;
+ case nir_jump_continue:
+ emit(BRW_OPCODE_CONTINUE);
+ break;
+ case nir_jump_return:
+ default:
+ unreachable("unknown jump");
+ }
+}
diff --git a/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp b/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp
index 025cac5..3e447b8 100644
--- a/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp
+++ b/src/mesa/drivers/dri/i965/brw_fs_visitor.cpp
@@ -3816,6 +3816,7 @@ fs_visitor::init()
this->variable_ht = hash_table_ctor(0,
hash_table_pointer_hash,
hash_table_pointer_compare);
+ this->nir_reg_ht = _mesa_hash_table_create(NULL, _mesa_key_pointer_equal);
memset(&this->payload, 0, sizeof(this->payload));
memset(this->outputs, 0, sizeof(this->outputs));
@@ -3851,4 +3852,5 @@ fs_visitor::init()
fs_visitor::~fs_visitor()
{
hash_table_dtor(this->variable_ht);
+ _mesa_hash_table_destroy(this->nir_reg_ht, NULL);
}
--
2.2.0
More information about the mesa-dev
mailing list