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

nir: extract out convert_to_bitsize() helper

Browse Source 
Signed-off-by: Rob Clark <robdclark@chromium.org>
Reviewed-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Reviewed-by: Eric Anholt <eric@anholt.net>
Reviewed-by: Kristian H. Kristensen <hoegsberg@google.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/5318>
This commit is contained in:
Rob Clark
2020-06-03 12:09:59 -07:00
committed by Marge Bot
parent 924bfb6560
commit 866618c5c8
2 changed files with 17 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1384,4 +1384,19 @@ nir_scoped_memory_barrier(nir_builder *b,
nir_scoped_barrier(b, NIR_SCOPE_NONE, scope, semantics, modes);
}
static inline nir_ssa_def *
nir_convert_to_bit_size(nir_builder *b,
nir_ssa_def *src,
nir_alu_type type,
unsigned bit_size)
{
nir_alu_type base_type = nir_alu_type_get_base_type(type);
nir_alu_type dst_type = (nir_alu_type)(bit_size | base_type);
nir_op opcode =
nir_type_conversion_op(type, dst_type, nir_rounding_mode_undef);
return nir_build_alu(b, opcode, src, NULL, NULL, NULL);
}
#endif /* NIR_BUILDER_H */

18
src/compiler/nir/nir_lower_bit_size.c
View File

@@ -29,20 +29,6 @@
* a bit-size that is supported natively and then converts the result back to
* the original bit-size.
*/
static nir_ssa_def *
convert_to_bit_size(nir_builder *bld,
nir_ssa_def *src,
nir_alu_type type,
unsigned bit_size)
{
nir_alu_type base_type = nir_alu_type_get_base_type(type);
nir_alu_type lowered_type = bit_size | base_type;
nir_op opcode =
nir_type_conversion_op(type, lowered_type, nir_rounding_mode_undef);
return nir_build_alu(bld, opcode, src, NULL, NULL, NULL);
}
static void
lower_instr(nir_builder *bld, nir_alu_instr *alu, unsigned bit_size)
@@ -59,7 +45,7 @@ lower_instr(nir_builder *bld, nir_alu_instr *alu, unsigned bit_size)
nir_alu_type type = nir_op_infos[op].input_types[i];
if (nir_alu_type_get_type_size(type) == 0)
src = convert_to_bit_size(bld, src, type, bit_size);
src = nir_convert_to_bit_size(bld, src, type, bit_size);
if (i == 1 && (op == nir_op_ishl || op == nir_op_ishr || op == nir_op_ushr)) {
assert(util_is_power_of_two_nonzero(dst_bit_size));
@@ -85,7 +71,7 @@ lower_instr(nir_builder *bld, nir_alu_instr *alu, unsigned bit_size)
/* Convert result back to the original bit-size */
nir_alu_type type = nir_op_infos[op].output_type;
nir_ssa_def *dst = convert_to_bit_size(bld, lowered_dst, type, dst_bit_size);
nir_ssa_def *dst = nir_convert_to_bit_size(bld, lowered_dst, type, dst_bit_size);
nir_ssa_def_rewrite_uses(&alu->dest.dest.ssa, nir_src_for_ssa(dst));
}