Mesa (main): nir: Support upper bound of unsigned bit size conversions.

[GitLab Mirror]

Module: Mesa
Branch: main
Commit: e905e0938a7f027e3ccda99ac0dd3cb7c4d46901
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=e905e0938a7f027e3ccda99ac0dd3cb7c4d46901

Author: Timur Kristóf <timur.kristof at gmail.com>
Date:   Fri Apr 23 08:48:46 2021 +0200

nir: Support upper bound of unsigned bit size conversions.

These allow us to generate slightly better code in some cases,
eg. multiplications in ACO.

Signed-off-by: Timur Kristóf <timur.kristof at gmail.com>
Reviewed-by: Daniel Schürmann <daniel at schuermann.dev>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/10740>

---

 src/compiler/nir/nir_range_analysis.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/compiler/nir/nir_range_analysis.c b/src/compiler/nir/nir_range_analysis.c
index 75632783c10..1d7eb99283e 100644
--- a/src/compiler/nir/nir_range_analysis.c
+++ b/src/compiler/nir/nir_range_analysis.c
@@ -1457,6 +1457,15 @@ nir_unsigned_upper_bound(nir_shader *shader, struct hash_table *range_ht,
       case nir_op_f2u32:
       case nir_op_fmul:
          break;
+      case nir_op_u2u1:
+      case nir_op_u2u8:
+      case nir_op_u2u16:
+      case nir_op_u2u32:
+         if (nir_ssa_scalar_chase_alu_src(scalar, 0).def->bit_size > 32) {
+            /* If src is >32 bits, return max */
+            return max;
+         }
+         break;
       default:
          return max;
       }
@@ -1570,6 +1579,12 @@ nir_unsigned_upper_bound(nir_shader *shader, struct hash_table *range_ht,
             memcpy(&res, &max_f, 4);
          }
          break;
+      case nir_op_u2u1:
+      case nir_op_u2u8:
+      case nir_op_u2u16:
+      case nir_op_u2u32:
+         res = MIN2(src0, max);
+         break;
       default:
          res = max;
          break;