From: Samuel Iglesias Gonsálvez <[email protected]> This function returns the nir_op corresponding to the conversion between the given nir_alu_type arguments.
This function lacks support for integer-based types with bit_size != 32 and for float16 conversion ops. Signed-off-by: Samuel Iglesias Gonsálvez <[email protected]> --- src/compiler/nir/nir.c | 83 ++++++++++++++++++++++++++++++++++++++++++++++++++ src/compiler/nir/nir.h | 2 ++ 2 files changed, 85 insertions(+) diff --git a/src/compiler/nir/nir.c b/src/compiler/nir/nir.c index 2d882f7..3e00452 100644 --- a/src/compiler/nir/nir.c +++ b/src/compiler/nir/nir.c @@ -1953,3 +1953,86 @@ nir_system_value_from_intrinsic(nir_intrinsic_op intrin) unreachable("intrinsic doesn't produce a system value"); } } + +nir_op +nir_type_conversion_op(nir_alu_type src, nir_alu_type dst) +{ + nir_alu_type src_base_type = (nir_alu_type) nir_alu_type_get_base_type(src); + nir_alu_type dst_base_type = (nir_alu_type) nir_alu_type_get_base_type(dst); + unsigned src_bitsize = nir_alu_type_get_type_size(src); + unsigned dst_bitsize = nir_alu_type_get_type_size(dst); + + if (src_base_type == dst_base_type) { + if (src_bitsize == dst_bitsize) + return (src_base_type == nir_type_float) ? nir_op_fmov : nir_op_imov; + + switch (src_base_type) { + case nir_type_bool: + case nir_type_uint: + case nir_type_int: + return nir_op_imov; + case nir_type_float: + /* TODO: implement support for float16 */ + assert(src_bitsize == 64 || dst_bitsize == 64); + return (src_bitsize == 64) ? nir_op_d2f : nir_op_f2d; + default: + assert(!"Invalid conversion"); + }; + } + + /* Different base type but same bit_size */ + if (src_bitsize == dst_bitsize) { + /* TODO: This does not include specific conversions between + * signed or unsigned integer types of bit size different of 32 yet. + */ + assert(src_bitsize == 32); + switch (src_base_type) { + case nir_type_uint: + return (dst_base_type == nir_type_float) ? nir_op_u2f : nir_op_imov; + case nir_type_int: + return (dst_base_type == nir_type_float) ? nir_op_i2f : nir_op_imov; + case nir_type_bool: + return (dst_base_type == nir_type_float) ? nir_op_b2f : nir_op_imov; + case nir_type_float: + return (dst_base_type == nir_type_uint) ? nir_op_f2u : + (dst_base_type == nir_type_bool) ? nir_op_f2b : nir_op_f2i; + default: + assert(!"Invalid conversion"); + }; + } + + /* Different bit_size and different base type */ + /* TODO: Implement integer support for types with bit_size != 32 */ + switch (src_base_type) { + case nir_type_uint: + assert(dst == nir_type_float64); + return nir_op_u2d; + case nir_type_int: + assert(dst == nir_type_float64); + return nir_op_i2d; + case nir_type_bool: + assert(dst == nir_type_float64); + return nir_op_u2d; + case nir_type_float: + assert(src_bitsize == 32 || src_bitsize == 64); + if (src_bitsize != 64) { + assert(dst == nir_type_float64); + return nir_op_f2d; + } + assert(dst_bitsize == 32); + switch (dst_base_type) { + case nir_type_uint: + return nir_op_d2u; + case nir_type_int: + return nir_op_d2i; + case nir_type_bool: + return nir_op_d2b; + case nir_type_float: + return nir_op_d2f; + default: + assert(!"Invalid conversion"); + }; + default: + assert(!"Invalid conversion"); + }; +} diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h index 9310dab..9f3abb7 100644 --- a/src/compiler/nir/nir.h +++ b/src/compiler/nir/nir.h @@ -689,6 +689,8 @@ nir_get_nir_type_for_glsl_type(const struct glsl_type *type) unreachable("unknown type"); } +nir_op nir_type_conversion_op(nir_alu_type src, nir_alu_type dst); + typedef enum { NIR_OP_IS_COMMUTATIVE = (1 << 0), NIR_OP_IS_ASSOCIATIVE = (1 << 1), -- 2.9.3 _______________________________________________ mesa-dev mailing list [email protected] https://lists.freedesktop.org/mailman/listinfo/mesa-dev
