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

nir: Add a nir_src_as_intrinsic() helper

Browse Source 
Reviewed-by: Eric Anholt <eric@anholt.net>
Reviewed-by: Caio Marcelo de Oliveira Filho <caio.oliveira@intel.com>
This commit is contained in:
Jason Ekstrand
2019-04-17 17:18:19 -05:00
parent 85c35885b3
commit c6463f8ac2
5 changed files with 19 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/compiler/nir/nir.h
View File

@@ -2748,12 +2748,6 @@ bool nir_foreach_src(nir_instr *instr, nir_foreach_src_cb cb, void *state);
nir_const_value *nir_src_as_const_value(nir_src src); nir_const_value *nir_src_as_const_value(nir_src src);
static inline struct nir_instr *
nir_src_instr(const struct nir_src *src)
{
return src->is_ssa ? src->ssa->parent_instr : NULL;
}
#define NIR_SRC_AS_(name, c_type, type_enum, cast_macro) \ #define NIR_SRC_AS_(name, c_type, type_enum, cast_macro) \
static inline c_type * \ static inline c_type * \
nir_src_as_ ## name (nir_src src) \ nir_src_as_ ## name (nir_src src) \
@@ -2763,6 +2757,8 @@ nir_src_as_ ## name (nir_src src) \
} }
NIR_SRC_AS_(alu_instr, nir_alu_instr, nir_instr_type_alu, nir_instr_as_alu) NIR_SRC_AS_(alu_instr, nir_alu_instr, nir_instr_type_alu, nir_instr_as_alu)
NIR_SRC_AS_(intrinsic, nir_intrinsic_instr,
nir_instr_type_intrinsic, nir_instr_as_intrinsic)
bool nir_src_is_dynamically_uniform(nir_src src); bool nir_src_is_dynamically_uniform(nir_src src);
bool nir_srcs_equal(nir_src src1, nir_src src2); bool nir_srcs_equal(nir_src src1, nir_src src2);

10
src/compiler/nir/nir_opt_find_array_copies.c
View File

@@ -37,14 +37,8 @@ index_ssa_def_cb(nir_ssa_def *def, void *state)
static nir_deref_instr * static nir_deref_instr *
get_deref_for_load_src(nir_src src, unsigned first_valid_load) get_deref_for_load_src(nir_src src, unsigned first_valid_load)
{ {
if (!src.is_ssa) nir_intrinsic_instr *load = nir_src_as_intrinsic(src);
return NULL; if (load == NULL || load->intrinsic != nir_intrinsic_load_deref)
if (src.ssa->parent_instr->type != nir_instr_type_intrinsic)
return NULL;
nir_intrinsic_instr *load = nir_instr_as_intrinsic(src.ssa->parent_instr);
if (load->intrinsic != nir_intrinsic_load_deref)
return NULL; return NULL;
if (load->dest.ssa.index < first_valid_load) if (load->dest.ssa.index < first_valid_load)

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

@@ -1030,14 +1030,8 @@ mark_deref_used(nir_deref_instr *deref,
static bool static bool
src_is_load_deref(nir_src src, nir_src deref_src) src_is_load_deref(nir_src src, nir_src deref_src)
{ {
assert(src.is_ssa); nir_intrinsic_instr *load = nir_src_as_intrinsic(src);
assert(deref_src.is_ssa); if (load == NULL || load->intrinsic != nir_intrinsic_load_deref)
if (src.ssa->parent_instr->type != nir_instr_type_intrinsic)
return false;
nir_intrinsic_instr *load = nir_instr_as_intrinsic(src.ssa->parent_instr);
if (load->intrinsic != nir_intrinsic_load_deref)
return false; return false;
assert(load->src[0].is_ssa); assert(load->src[0].is_ssa);

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

@@ -993,8 +993,7 @@ TEST_F(nir_combine_stores_test, non_overlapping_stores)
nir_alu_instr *vec = nir_src_as_alu_instr(combined->src[1]); nir_alu_instr *vec = nir_src_as_alu_instr(combined->src[1]);
ASSERT_TRUE(vec); ASSERT_TRUE(vec);
for (int i = 0; i < 4; i++) { for (int i = 0; i < 4; i++) {
nir_intrinsic_instr *load = nir_intrinsic_instr *load = nir_src_as_intrinsic(vec->src[i].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[i].src));
ASSERT_EQ(load->intrinsic, nir_intrinsic_load_deref); ASSERT_EQ(load->intrinsic, nir_intrinsic_load_deref);
ASSERT_EQ(nir_intrinsic_get_var(load, 0), v[i]) ASSERT_EQ(nir_intrinsic_get_var(load, 0), v[i])
<< "Source value for component " << i << " of store is wrong"; << "Source value for component " << i << " of store is wrong";
@@ -1034,22 +1033,18 @@ TEST_F(nir_combine_stores_test, overlapping_stores)
ASSERT_TRUE(vec); ASSERT_TRUE(vec);
/* Component x comes from v[0]. */ /* Component x comes from v[0]. */
nir_intrinsic_instr *load_for_x = nir_intrinsic_instr *load_for_x = nir_src_as_intrinsic(vec->src[0].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[0].src));
ASSERT_EQ(nir_intrinsic_get_var(load_for_x, 0), v[0]); ASSERT_EQ(nir_intrinsic_get_var(load_for_x, 0), v[0]);
ASSERT_EQ(vec->src[0].swizzle[0], 0); ASSERT_EQ(vec->src[0].swizzle[0], 0);
/* Component y comes from v[1]. */ /* Component y comes from v[1]. */
nir_intrinsic_instr *load_for_y = nir_intrinsic_instr *load_for_y = nir_src_as_intrinsic(vec->src[1].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[1].src));
ASSERT_EQ(nir_intrinsic_get_var(load_for_y, 0), v[1]); ASSERT_EQ(nir_intrinsic_get_var(load_for_y, 0), v[1]);
ASSERT_EQ(vec->src[1].swizzle[0], 1); ASSERT_EQ(vec->src[1].swizzle[0], 1);
/* Components z and w come from v[2]. */ /* Components z and w come from v[2]. */
nir_intrinsic_instr *load_for_z = nir_intrinsic_instr *load_for_z = nir_src_as_intrinsic(vec->src[2].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[2].src)); nir_intrinsic_instr *load_for_w = nir_src_as_intrinsic(vec->src[3].src);
nir_intrinsic_instr *load_for_w =
nir_instr_as_intrinsic(nir_src_instr(&vec->src[3].src));
ASSERT_EQ(load_for_z, load_for_w); ASSERT_EQ(load_for_z, load_for_w);
ASSERT_EQ(nir_intrinsic_get_var(load_for_z, 0), v[2]); ASSERT_EQ(nir_intrinsic_get_var(load_for_z, 0), v[2]);
ASSERT_EQ(vec->src[2].swizzle[0], 2); ASSERT_EQ(vec->src[2].swizzle[0], 2);
@@ -1104,26 +1099,22 @@ TEST_F(nir_combine_stores_test, direct_array_derefs)
ASSERT_TRUE(vec); ASSERT_TRUE(vec);
/* Component x comes from v[0]. */ /* Component x comes from v[0]. */
nir_intrinsic_instr *load_for_x = nir_intrinsic_instr *load_for_x = nir_src_as_intrinsic(vec->src[0].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[0].src));
ASSERT_EQ(nir_intrinsic_get_var(load_for_x, 0), v[0]); ASSERT_EQ(nir_intrinsic_get_var(load_for_x, 0), v[0]);
ASSERT_EQ(vec->src[0].swizzle[0], 0); ASSERT_EQ(vec->src[0].swizzle[0], 0);
/* Component y comes from v[1]. */ /* Component y comes from v[1]. */
nir_intrinsic_instr *load_for_y = nir_intrinsic_instr *load_for_y = nir_src_as_intrinsic(vec->src[1].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[1].src));
ASSERT_EQ(nir_intrinsic_get_var(load_for_y, 0), v[1]); ASSERT_EQ(nir_intrinsic_get_var(load_for_y, 0), v[1]);
ASSERT_EQ(vec->src[1].swizzle[0], 1); ASSERT_EQ(vec->src[1].swizzle[0], 1);
/* Components z comes from s[0]. */ /* Components z comes from s[0]. */
nir_intrinsic_instr *load_for_z = nir_intrinsic_instr *load_for_z = nir_src_as_intrinsic(vec->src[2].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[2].src));
ASSERT_EQ(nir_intrinsic_get_var(load_for_z, 0), s[0]); ASSERT_EQ(nir_intrinsic_get_var(load_for_z, 0), s[0]);
ASSERT_EQ(vec->src[2].swizzle[0], 0); ASSERT_EQ(vec->src[2].swizzle[0], 0);
/* Component w comes from s[1]. */ /* Component w comes from s[1]. */
nir_intrinsic_instr *load_for_w = nir_intrinsic_instr *load_for_w = nir_src_as_intrinsic(vec->src[3].src);
nir_instr_as_intrinsic(nir_src_instr(&vec->src[3].src));
ASSERT_EQ(nir_intrinsic_get_var(load_for_w, 0), s[1]); ASSERT_EQ(nir_intrinsic_get_var(load_for_w, 0), s[1]);
ASSERT_EQ(vec->src[3].swizzle[0], 0); ASSERT_EQ(vec->src[3].swizzle[0], 0);
} }

15
src/intel/compiler/brw_fs_nir.cpp
View File

@@ -558,14 +558,8 @@ bool
fs_visitor::optimize_frontfacing_ternary(nir_alu_instr *instr, fs_visitor::optimize_frontfacing_ternary(nir_alu_instr *instr,
const fs_reg &result) const fs_reg &result)
{ {
if (!instr->src[0].src.is_ssa || nir_intrinsic_instr *src0 = nir_src_as_intrinsic(instr->src[0].src);
instr->src[0].src.ssa->parent_instr->type != nir_instr_type_intrinsic) if (src0 == NULL || src0->intrinsic != nir_intrinsic_load_front_face)
return false;
nir_intrinsic_instr *src0 =
nir_instr_as_intrinsic(instr->src[0].src.ssa->parent_instr);
if (src0->intrinsic != nir_intrinsic_load_front_face)
return false; return false;
if (!nir_src_is_const(instr->src[1].src) || if (!nir_src_is_const(instr->src[1].src) ||
@@ -2527,9 +2521,8 @@ fs_visitor::nir_emit_tcs_intrinsic(const fs_builder &bld,
retype(brw_vec1_grf(1 + (vertex >> 3), vertex & 7), retype(brw_vec1_grf(1 + (vertex >> 3), vertex & 7),
BRW_REGISTER_TYPE_UD)); BRW_REGISTER_TYPE_UD));
} else if (tcs_prog_data->instances == 1 && } else if (tcs_prog_data->instances == 1 &&
vertex_src.is_ssa && nir_src_as_intrinsic(vertex_src) != NULL &&
vertex_src.ssa->parent_instr->type == nir_instr_type_intrinsic && nir_src_as_intrinsic(vertex_src)->intrinsic == nir_intrinsic_load_invocation_id) {
nir_instr_as_intrinsic(vertex_src.ssa->parent_instr)->intrinsic == nir_intrinsic_load_invocation_id) {
/* For the common case of only 1 instance, an array index of /* For the common case of only 1 instance, an array index of
* gl_InvocationID means reading g1. Skip all the indirect work. * gl_InvocationID means reading g1. Skip all the indirect work.
*/ */