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

nir: Add a read_mask helper for ALU instructions

Browse Source 
Reviewed-by: Kenneth Graunke <kenneth@whitecape.org>
This commit is contained in:
Jason Ekstrand
2018-11-07 17:47:45 -06:00
parent 344cfe6980
commit 61e15348c4
2 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1530,13 +1530,7 @@ nir_ssa_def_components_read(const nir_ssa_def *def)
nir_alu_src *alu_src = exec_node_data(nir_alu_src, use, src); nir_alu_src *alu_src = exec_node_data(nir_alu_src, use, src);
int src_idx = alu_src - &alu->src[0]; int src_idx = alu_src - &alu->src[0];
assert(src_idx >= 0 && src_idx < nir_op_infos[alu->op].num_inputs); assert(src_idx >= 0 && src_idx < nir_op_infos[alu->op].num_inputs);
read_mask |= nir_alu_instr_src_read_mask(alu, src_idx);
for (unsigned c = 0; c < NIR_MAX_VEC_COMPONENTS; c++) {
if (!nir_alu_instr_channel_used(alu, src_idx, c))
continue;
read_mask |= (1 << alu_src->swizzle[c]);
}
} else { } else {
return (1 << def->num_components) - 1; return (1 << def->num_components) - 1;
} }

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

@@ -952,6 +952,19 @@ nir_alu_instr_channel_used(const nir_alu_instr *instr, unsigned src,
return (instr->dest.write_mask >> channel) & 1; return (instr->dest.write_mask >> channel) & 1;
} }
static inline nir_component_mask_t
nir_alu_instr_src_read_mask(const nir_alu_instr *instr, unsigned src)
{
nir_component_mask_t read_mask = 0;
for (unsigned c = 0; c < NIR_MAX_VEC_COMPONENTS; c++) {
if (!nir_alu_instr_channel_used(instr, src, c))
continue;
read_mask |= (1 << instr->src[src].swizzle[c]);
}
return read_mask;
}
/* /*
* For instructions whose destinations are SSA, get the number of channels * For instructions whose destinations are SSA, get the number of channels
* used for a source * used for a source