OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nir/builder: Add a build_deref_array_imm helper

Browse Source 
Unlike most of the cases in which we do this by hand, the new helper
properly handles non-32-bit pointers.

Reviewed-by: Karol Herbst <kherbst@redhat.com>
Reviewed-by: Lionel Landwerlin <lionel.g.landwerlin@intel.com>
This commit is contained in:
Jason Ekstrand
2019-03-07 11:45:13 -06:00
committed by Jason Ekstrand
parent fcf2a0122e
commit 1664de5924
8 changed files with 26 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/compiler/nir/nir_builder.h
View File

@@ -843,6 +843,17 @@ nir_build_deref_array(nir_builder *build, nir_deref_instr *parent,
return deref;
}
static inline nir_deref_instr *
nir_build_deref_array_imm(nir_builder *build, nir_deref_instr *parent,
int64_t index)
{
assert(parent->dest.is_ssa);
nir_ssa_def *idx_ssa = nir_imm_intN_t(build, index,
parent->dest.ssa.bit_size);
return nir_build_deref_array(build, parent, idx_ssa);
}
static inline nir_deref_instr *
nir_build_deref_ptr_as_array(nir_builder *build, nir_deref_instr *parent,
nir_ssa_def *index)

4
src/compiler/nir/nir_lower_constant_initializers.c
View File

@@ -44,7 +44,7 @@ build_constant_load(nir_builder *b, nir_deref_instr *deref, nir_constant *c)
nir_load_const_instr_create(b->shader, rows, bit_size);
load->value = c->values[i];
nir_builder_instr_insert(b, &load->instr);
nir_store_deref(b, nir_build_deref_array(b, deref, nir_imm_int(b, i)),
nir_store_deref(b, nir_build_deref_array_imm(b, deref, i),
&load->def, ~0);
}
} else if (glsl_type_is_struct_or_ifc(deref->type)) {
@@ -58,7 +58,7 @@ build_constant_load(nir_builder *b, nir_deref_instr *deref, nir_constant *c)
unsigned len = glsl_get_length(deref->type);
for (unsigned i = 0; i < len; i++) {
build_constant_load(b,
nir_build_deref_array(b, deref, nir_imm_int(b, i)),
nir_build_deref_array_imm(b, deref, i),
c->elements[i]);
}
}

3
src/compiler/nir/nir_lower_indirect_derefs.c
View File

@@ -40,9 +40,8 @@ emit_indirect_load_store_deref(nir_builder *b, nir_intrinsic_instr *orig_instr,
{
assert(start < end);
if (start == end - 1) {
nir_ssa_def *index = nir_imm_int(b, start);
emit_load_store_deref(b, orig_instr,
nir_build_deref_array(b, parent, index),
nir_build_deref_array_imm(b, parent, start),
deref_arr + 1, dest, src);
} else {
int mid = start + (end - start) / 2;

5
src/compiler/nir/nir_lower_var_copies.c
View File

@@ -75,11 +75,10 @@ emit_deref_copy_load_store(nir_builder *b,
assert(length > 0);
for (unsigned i = 0; i < length; i++) {
nir_ssa_def *index = nir_imm_int(b, i);
emit_deref_copy_load_store(b,
nir_build_deref_array(b, dst_deref, index),
nir_build_deref_array_imm(b, dst_deref, i),
dst_deref_arr + 1,
nir_build_deref_array(b, src_deref, index),
nir_build_deref_array_imm(b, src_deref, i),
src_deref_arr + 1);
}
} else {

5
src/compiler/nir/nir_split_vars.c
View File

@@ -616,11 +616,10 @@ emit_split_copies(nir_builder *b,
glsl_get_length(src_path->path[src_level]->type));
unsigned len = glsl_get_length(dst_path->path[dst_level]->type);
for (unsigned i = 0; i < len; i++) {
nir_ssa_def *idx = nir_imm_int(b, i);
emit_split_copies(b, dst_info, dst_path, dst_level + 1,
nir_build_deref_array(b, dst, idx),
nir_build_deref_array_imm(b, dst, i),
src_info, src_path, src_level + 1,
nir_build_deref_array(b, src, idx));
nir_build_deref_array_imm(b, src, i));
}
} else {
/* Neither side is being split so we just keep going */

12
src/compiler/nir/tests/vars_tests.cpp
View File

@@ -476,7 +476,7 @@ TEST_F(nir_copy_prop_vars_test, load_direct_array_deref_on_vector_reuses_previou
/* This load will be dropped, as vec.y (or vec[1]) is already known. */
nir_deref_instr *deref =
nir_build_deref_array(b, nir_build_deref_var(b, vec), nir_imm_int(b, 1));
nir_build_deref_array_imm(b, nir_build_deref_var(b, vec), 1);
nir_ssa_def *loaded_from_deref = nir_load_deref(b, deref);
/* This store should use the value loaded from in1. */
@@ -509,7 +509,7 @@ TEST_F(nir_copy_prop_vars_test, load_direct_array_deref_on_vector_reuses_previou
/* This load will be replaced with one from in0. */
nir_deref_instr *deref =
nir_build_deref_array(b, nir_build_deref_var(b, vec), nir_imm_int(b, 1));
nir_build_deref_array_imm(b, nir_build_deref_var(b, vec), 1);
nir_load_deref(b, deref);
nir_validate_shader(b->shader, NULL);
@@ -532,7 +532,7 @@ TEST_F(nir_copy_prop_vars_test, load_direct_array_deref_on_vector_gets_reused)
/* Loading "vec[1]" deref will save the information about vec.y. */
nir_deref_instr *deref =
nir_build_deref_array(b, nir_build_deref_var(b, vec), nir_imm_int(b, 1));
nir_build_deref_array_imm(b, nir_build_deref_var(b, vec), 1);
nir_load_deref(b, deref);
/* Store to vec.x. */
@@ -566,16 +566,16 @@ TEST_F(nir_copy_prop_vars_test, store_load_direct_array_deref_on_vector)
/* Store to "vec[1]" and "vec[0]". */
nir_deref_instr *store_deref_y =
nir_build_deref_array(b, nir_build_deref_var(b, vec), nir_imm_int(b, 1));
nir_build_deref_array_imm(b, nir_build_deref_var(b, vec), 1);
nir_store_deref(b, store_deref_y, nir_imm_int(b, 20), 1);
nir_deref_instr *store_deref_x =
nir_build_deref_array(b, nir_build_deref_var(b, vec), nir_imm_int(b, 0));
nir_build_deref_array_imm(b, nir_build_deref_var(b, vec), 0);
nir_store_deref(b, store_deref_x, nir_imm_int(b, 10), 1);
/* Both loads below will be dropped, because the values are already known. */
nir_deref_instr *load_deref_y =
nir_build_deref_array(b, nir_build_deref_var(b, vec), nir_imm_int(b, 1));
nir_build_deref_array_imm(b, nir_build_deref_var(b, vec), 1);
nir_store_var(b, out0, nir_load_deref(b, load_deref_y), 1);
nir_store_var(b, out1, nir_load_var(b, vec), 1);

2
src/compiler/spirv/vtn_variables.c
View File

@@ -602,7 +602,7 @@ _vtn_local_load_store(struct vtn_builder *b, bool load, nir_deref_instr *deref,
unsigned elems = glsl_get_length(deref->type);
for (unsigned i = 0; i < elems; i++) {
nir_deref_instr *child =
nir_build_deref_array(&b->nb, deref, nir_imm_int(&b->nb, i));
nir_build_deref_array_imm(&b->nb, deref, i);
_vtn_local_load_store(b, load, child, inout->elems[i]);
}
} else {

2
src/intel/vulkan/anv_nir_lower_multiview.c
View File

@@ -136,7 +136,7 @@ build_view_index(struct lower_multiview_state *state)
nir_deref_instr *deref = nir_build_deref_var(b, idx_var);
if (glsl_type_is_array(type))
deref = nir_build_deref_array(b, deref, nir_imm_int(b, 0));
deref = nir_build_deref_array_imm(b, deref, 0);
state->view_index = nir_load_deref(b, deref);
}