From: Mihai Preda <mhpreda@gmail.com>
Date: Wed, 29 Jun 2022 12:18:41 +0000 (+0300)
Subject: nir: update nir->num_inputs, num_outputs in nir_recompute_io_bases()
X-Git-Tag: upstream/23.3.3~9777
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=2a9ef6b1f84b9b07c12150545fe2111f7e793e7a;p=platform%2Fupstream%2Fmesa.git

nir: update nir->num_inputs, num_outputs in nir_recompute_io_bases()

Since nir_recompute_io_bases is going to be used instead of
nir_assign_io_var_locations, we need to update num_inputs, num_outputs.

Reviewed-by: Qiang Yu <yuq825@gmail.com>
Reviewed-by: Timur Kristóf <timur.kristof@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/21861>
---

diff --git a/src/compiler/nir/nir_lower_mediump.c b/src/compiler/nir/nir_lower_mediump.c
index 3da34ea..8698b0d 100644
--- a/src/compiler/nir/nir_lower_mediump.c
+++ b/src/compiler/nir/nir_lower_mediump.c
@@ -130,6 +130,11 @@ nir_recompute_io_bases(nir_shader *nir, nir_variable_mode modes)
       nir_metadata_preserve(impl, nir_metadata_all);
    }
 
+   if (modes & nir_var_shader_in)
+      nir->num_inputs = BITSET_COUNT(inputs);
+   if (modes & nir_var_shader_out)
+      nir->num_outputs = BITSET_COUNT(outputs);
+
    return changed;
 }