Mesa (main): zink: use nir_shader_instructions_pass in nir_lower_dynamic_bo_access

[GitLab Mirror]

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

Author: Marcin Ślusarz <marcin.slusarz at intel.com>
Date:   Tue Aug 10 13:53:40 2021 +0200

zink: use nir_shader_instructions_pass in nir_lower_dynamic_bo_access

Changes:
- nir_metadata_preserve(..., nir_metadata_dominance)
  is called only when pass makes progress
- nir_metadata_preserve(..., nir_metadata_all) is called when pass doesn't
  make progress

Signed-off-by: Marcin Ślusarz <marcin.slusarz at intel.com>
Reviewed-by: Erik Faye-Lund <erik.faye-lund at collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/12324>

---

 .../drivers/zink/nir_lower_dynamic_bo_access.c     | 33 ++++++++--------------
 1 file changed, 12 insertions(+), 21 deletions(-)

diff --git a/src/gallium/drivers/zink/nir_lower_dynamic_bo_access.c b/src/gallium/drivers/zink/nir_lower_dynamic_bo_access.c
index 864030ad8d8..157b67bb2cf 100644
--- a/src/gallium/drivers/zink/nir_lower_dynamic_bo_access.c
+++ b/src/gallium/drivers/zink/nir_lower_dynamic_bo_access.c
@@ -91,8 +91,15 @@ generate_store_ssbo_ssa_def(nir_builder *b, nir_intrinsic_instr *instr, nir_ssa_
 }
 
 static bool
-lower_dynamic_bo_access_instr(nir_intrinsic_instr *instr, nir_builder *b)
+lower_dynamic_bo_access_instr(nir_builder *b,
+                              nir_instr *instr_,
+                              UNUSED void *cb_data)
 {
+   if (instr_->type != nir_instr_type_intrinsic)
+      return false;
+
+   nir_intrinsic_instr *instr = nir_instr_as_intrinsic(instr_);
+
    if (instr->intrinsic != nir_intrinsic_load_ubo &&
        instr->intrinsic != nir_intrinsic_load_ubo_vec4 &&
        instr->intrinsic != nir_intrinsic_get_ssbo_size &&
@@ -137,24 +144,8 @@ lower_dynamic_bo_access_instr(nir_intrinsic_instr *instr, nir_builder *b)
 bool
 nir_lower_dynamic_bo_access(nir_shader *shader)
 {
-   bool progress = false;
-
-   nir_foreach_function(function, shader) {
-      if (function->impl) {
-         nir_builder builder;
-         nir_builder_init(&builder, function->impl);
-         nir_foreach_block(block, function->impl) {
-            nir_foreach_instr_safe(instr, block) {
-               if (instr->type == nir_instr_type_intrinsic)
-                  progress |= lower_dynamic_bo_access_instr(
-                                                  nir_instr_as_intrinsic(instr),
-                                                  &builder);
-            }
-         }
-
-         nir_metadata_preserve(function->impl, nir_metadata_dominance);
-      }
-   }
-
-   return progress;
+   return nir_shader_instructions_pass(shader,
+                                       lower_dynamic_bo_access_instr,
+                                       nir_metadata_dominance,
+                                       NULL);
 }