nir/phi_builder: Internal users should use nir_phi_builder_value_set_block_def too

Signed-off-by: Ian Romanick <[email protected]>
Reviewed-by: Jason Ekstrand <[email protected]>

1 files changed, 2 insertions, 2 deletions

diff --git a/src/compiler/nir/nir_phi_builder.c b/src/compiler/nir/nir_phi_builder.c
index cc5ce81d120..add3efd2dfc 100644
--- a/src/compiler/nir/nir_phi_builder.c
+++ b/src/compiler/nir/nir_phi_builder.c
@@ -147,7 +147,7 @@ nir_phi_builder_add_value(struct nir_phi_builder *pb, unsigned num_components,
              * value to the magic value NEEDS_PHI.  Later, we create phi nodes
              * on demand in nir_phi_builder_value_get_block_def().
              */
-            val->defs[next->index] = NEEDS_PHI;
+            nir_phi_builder_value_set_block_def(val, next, NEEDS_PHI);
 
             if (pb->work[next->index] < pb->iter_count) {
                pb->work[next->index] = pb->iter_count;
@@ -232,7 +232,7 @@ nir_phi_builder_value_get_block_def(struct nir_phi_builder_value *val,
     *  2) To avoid unneeded recreation of phi nodes and undefs.
     */
    for (dom = block; dom && val->defs[dom->index] == NULL; dom = dom->imm_dom)
-      val->defs[dom->index] = def;
+      nir_phi_builder_value_set_block_def(val, dom, def);
 
    return def;
 }